├── .gitignore ├── ReadME.md ├── data ├── ImagenetDataset.py └── __init__.py ├── model └── CNN │ ├── __init__.py │ ├── resnet.py │ └── resnet_features.py ├── test.py ├── test.sh ├── train_lanuch.py ├── train_lanuch_amp.sh └── utils ├── __init__.py ├── calculate_acc.py ├── precise_bn.py └── utils.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FlyEgle/imageclassification/HEAD/.gitignore -------------------------------------------------------------------------------- /ReadME.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FlyEgle/imageclassification/HEAD/ReadME.md -------------------------------------------------------------------------------- /data/ImagenetDataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FlyEgle/imageclassification/HEAD/data/ImagenetDataset.py -------------------------------------------------------------------------------- /data/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /model/CNN/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /model/CNN/resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FlyEgle/imageclassification/HEAD/model/CNN/resnet.py -------------------------------------------------------------------------------- /model/CNN/resnet_features.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FlyEgle/imageclassification/HEAD/model/CNN/resnet_features.py -------------------------------------------------------------------------------- /test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FlyEgle/imageclassification/HEAD/test.py -------------------------------------------------------------------------------- /test.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FlyEgle/imageclassification/HEAD/test.sh -------------------------------------------------------------------------------- /train_lanuch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FlyEgle/imageclassification/HEAD/train_lanuch.py -------------------------------------------------------------------------------- /train_lanuch_amp.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FlyEgle/imageclassification/HEAD/train_lanuch_amp.sh -------------------------------------------------------------------------------- /utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /utils/calculate_acc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FlyEgle/imageclassification/HEAD/utils/calculate_acc.py -------------------------------------------------------------------------------- /utils/precise_bn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/FlyEgle/imageclassification/HEAD/utils/precise_bn.py -------------------------------------------------------------------------------- /utils/utils.py: -------------------------------------------------------------------------------- 1 | --------------------------------------------------------------------------------