├── .gitignore ├── LICENSE.txt ├── README.md ├── _config.yml ├── docs ├── _config.yml └── index.md ├── pretrainedmodels ├── __init__.py ├── models │ ├── __init__.py │ ├── bninception.py │ ├── cafferesnet.py │ ├── dpn.py │ ├── fbresnet.py │ ├── fbresnet │ │ ├── resnet152_dump.lua │ │ └── resnet152_load.py │ ├── googlenet.py │ ├── inceptionresnetv2.py │ ├── inceptionv4.py │ ├── mobilenet.py │ ├── mobilenetv2.py │ ├── nasnet.py │ ├── nasnet_mobile.py │ ├── resnext.py │ ├── resnext_features │ │ ├── __init__.py │ │ ├── resnext101_32x4d_features.py │ │ └── resnext101_64x4d_features.py │ ├── senet.py │ ├── shufflenet.py │ ├── state_dict_surgery.py │ ├── torchvision_models.py │ ├── utils.py │ ├── vggm.py │ ├── wideresnet.py │ └── xception.py ├── utils.py └── version.py ├── requirements.txt └── scripts ├── README.md ├── compute_accuracy_rate.py ├── compute_computational_complexity.py ├── compute_inference_time.py └── compute_memory_usage.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CeLuigi/models-comparison.pytorch/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CeLuigi/models-comparison.pytorch/HEAD/LICENSE.txt -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CeLuigi/models-comparison.pytorch/HEAD/README.md -------------------------------------------------------------------------------- /_config.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CeLuigi/models-comparison.pytorch/HEAD/_config.yml -------------------------------------------------------------------------------- /docs/_config.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CeLuigi/models-comparison.pytorch/HEAD/docs/_config.yml -------------------------------------------------------------------------------- /docs/index.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CeLuigi/models-comparison.pytorch/HEAD/docs/index.md -------------------------------------------------------------------------------- /pretrainedmodels/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CeLuigi/models-comparison.pytorch/HEAD/pretrainedmodels/__init__.py -------------------------------------------------------------------------------- /pretrainedmodels/models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CeLuigi/models-comparison.pytorch/HEAD/pretrainedmodels/models/__init__.py -------------------------------------------------------------------------------- /pretrainedmodels/models/bninception.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CeLuigi/models-comparison.pytorch/HEAD/pretrainedmodels/models/bninception.py -------------------------------------------------------------------------------- /pretrainedmodels/models/cafferesnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CeLuigi/models-comparison.pytorch/HEAD/pretrainedmodels/models/cafferesnet.py -------------------------------------------------------------------------------- /pretrainedmodels/models/dpn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CeLuigi/models-comparison.pytorch/HEAD/pretrainedmodels/models/dpn.py -------------------------------------------------------------------------------- /pretrainedmodels/models/fbresnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CeLuigi/models-comparison.pytorch/HEAD/pretrainedmodels/models/fbresnet.py -------------------------------------------------------------------------------- /pretrainedmodels/models/fbresnet/resnet152_dump.lua: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CeLuigi/models-comparison.pytorch/HEAD/pretrainedmodels/models/fbresnet/resnet152_dump.lua -------------------------------------------------------------------------------- /pretrainedmodels/models/fbresnet/resnet152_load.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CeLuigi/models-comparison.pytorch/HEAD/pretrainedmodels/models/fbresnet/resnet152_load.py -------------------------------------------------------------------------------- /pretrainedmodels/models/googlenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CeLuigi/models-comparison.pytorch/HEAD/pretrainedmodels/models/googlenet.py -------------------------------------------------------------------------------- /pretrainedmodels/models/inceptionresnetv2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CeLuigi/models-comparison.pytorch/HEAD/pretrainedmodels/models/inceptionresnetv2.py -------------------------------------------------------------------------------- /pretrainedmodels/models/inceptionv4.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CeLuigi/models-comparison.pytorch/HEAD/pretrainedmodels/models/inceptionv4.py -------------------------------------------------------------------------------- /pretrainedmodels/models/mobilenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CeLuigi/models-comparison.pytorch/HEAD/pretrainedmodels/models/mobilenet.py -------------------------------------------------------------------------------- /pretrainedmodels/models/mobilenetv2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CeLuigi/models-comparison.pytorch/HEAD/pretrainedmodels/models/mobilenetv2.py -------------------------------------------------------------------------------- /pretrainedmodels/models/nasnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CeLuigi/models-comparison.pytorch/HEAD/pretrainedmodels/models/nasnet.py -------------------------------------------------------------------------------- /pretrainedmodels/models/nasnet_mobile.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CeLuigi/models-comparison.pytorch/HEAD/pretrainedmodels/models/nasnet_mobile.py -------------------------------------------------------------------------------- /pretrainedmodels/models/resnext.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CeLuigi/models-comparison.pytorch/HEAD/pretrainedmodels/models/resnext.py -------------------------------------------------------------------------------- /pretrainedmodels/models/resnext_features/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CeLuigi/models-comparison.pytorch/HEAD/pretrainedmodels/models/resnext_features/__init__.py -------------------------------------------------------------------------------- /pretrainedmodels/models/resnext_features/resnext101_32x4d_features.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CeLuigi/models-comparison.pytorch/HEAD/pretrainedmodels/models/resnext_features/resnext101_32x4d_features.py -------------------------------------------------------------------------------- /pretrainedmodels/models/resnext_features/resnext101_64x4d_features.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CeLuigi/models-comparison.pytorch/HEAD/pretrainedmodels/models/resnext_features/resnext101_64x4d_features.py -------------------------------------------------------------------------------- /pretrainedmodels/models/senet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CeLuigi/models-comparison.pytorch/HEAD/pretrainedmodels/models/senet.py -------------------------------------------------------------------------------- /pretrainedmodels/models/shufflenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CeLuigi/models-comparison.pytorch/HEAD/pretrainedmodels/models/shufflenet.py -------------------------------------------------------------------------------- /pretrainedmodels/models/state_dict_surgery.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CeLuigi/models-comparison.pytorch/HEAD/pretrainedmodels/models/state_dict_surgery.py -------------------------------------------------------------------------------- /pretrainedmodels/models/torchvision_models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CeLuigi/models-comparison.pytorch/HEAD/pretrainedmodels/models/torchvision_models.py -------------------------------------------------------------------------------- /pretrainedmodels/models/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CeLuigi/models-comparison.pytorch/HEAD/pretrainedmodels/models/utils.py -------------------------------------------------------------------------------- /pretrainedmodels/models/vggm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CeLuigi/models-comparison.pytorch/HEAD/pretrainedmodels/models/vggm.py -------------------------------------------------------------------------------- /pretrainedmodels/models/wideresnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CeLuigi/models-comparison.pytorch/HEAD/pretrainedmodels/models/wideresnet.py -------------------------------------------------------------------------------- /pretrainedmodels/models/xception.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CeLuigi/models-comparison.pytorch/HEAD/pretrainedmodels/models/xception.py -------------------------------------------------------------------------------- /pretrainedmodels/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CeLuigi/models-comparison.pytorch/HEAD/pretrainedmodels/utils.py -------------------------------------------------------------------------------- /pretrainedmodels/version.py: -------------------------------------------------------------------------------- 1 | __version__ = '0.6.0' 2 | -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CeLuigi/models-comparison.pytorch/HEAD/requirements.txt -------------------------------------------------------------------------------- /scripts/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CeLuigi/models-comparison.pytorch/HEAD/scripts/README.md -------------------------------------------------------------------------------- /scripts/compute_accuracy_rate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CeLuigi/models-comparison.pytorch/HEAD/scripts/compute_accuracy_rate.py -------------------------------------------------------------------------------- /scripts/compute_computational_complexity.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CeLuigi/models-comparison.pytorch/HEAD/scripts/compute_computational_complexity.py -------------------------------------------------------------------------------- /scripts/compute_inference_time.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CeLuigi/models-comparison.pytorch/HEAD/scripts/compute_inference_time.py -------------------------------------------------------------------------------- /scripts/compute_memory_usage.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CeLuigi/models-comparison.pytorch/HEAD/scripts/compute_memory_usage.py --------------------------------------------------------------------------------