【文件属性】:
文件名称:PyTorch-SE-ResNet:SE-ResNet PyTorch版本
文件大小:9KB
文件格式:ZIP
更新时间:2021-05-16 15:01:58
Python
PyTorch-SE-ResNet
SE-ResNet PyTorch版本
我将其用于Place365和Challenge AI场景分类。
实作网
SE-ResNet(50、101、152)
SE-ResNeXt(50、101、152)
其他框架可以很容易地实现来修改model/model.py 。
Caffe源代码在
引文
如果您在研究中使用挤压和激发网络,请引用以下文章:
@article{hu2017,
title={Squeeze-and-Excitation Networks},
author={Jie Hu and Li Shen and Gang Sun},
journal={arXiv preprint arXiv:1709.01507},
year={2017}
}
【文件预览】:
PyTorch-SE-ResNet-master
----train.py(8KB)
----utils()
--------utils.py(939B)
--------dataset.py(3KB)
--------__init__.py(78B)
----model()
--------model.py(9KB)
--------__init__.py(47B)
----train.sh(257B)
----README.md(618B)
----.gitignore(1KB)