PyTorch-SE-ResNet:SE-ResNet PyTorch版本

时间:2024-06-05 08:48:38
【文件属性】:

文件名称:PyTorch-SE-ResNet:SE-ResNet PyTorch版本

文件大小:9KB

文件格式:ZIP

更新时间:2024-06-05 08:48:38

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)

网友评论