├── .github ├── CONTRIBUTING.md └── workflows │ ├── build.yml │ └── deploy.yml ├── .gitignore ├── .pre-commit-config.yaml ├── .readthedocs.yml ├── LICENSE ├── README.md ├── ckpts ├── shufflenet_1x_b256x4_imagenet │ ├── shufflenet_1x_b256x4_imagenet.json │ ├── shufflenet_1x_b256x4_imagenet.log │ └── shufflenet_1x_b256x4_imagenet.pth └── weightnet_1x_b256x4_imagenet │ ├── weightnet_1x_b256x4_imagenet.json │ ├── weightnet_1x_b256x4_imagenet.log │ └── weightnet_1x_b256x4_imagenet.pth ├── configs ├── _base_ │ ├── datasets │ │ ├── cifar10_bs16.py │ │ ├── imagenet_bs32.py │ │ ├── imagenet_bs32_pil_resize.py │ │ ├── imagenet_bs64.py │ │ ├── imagenet_bs64_pil_resize.py │ │ └── voc_bs16.py │ ├── default_runtime.py │ ├── models │ │ ├── mobilenet_v2_1x.py │ │ ├── resnest101.py │ │ ├── resnest200.py │ │ ├── resnest269.py │ │ ├── resnest50.py │ │ ├── resnet101.py │ │ ├── resnet101_cifar.py │ │ ├── resnet152.py │ │ ├── resnet152_cifar.py │ │ ├── resnet18.py │ │ ├── resnet18_cifar.py │ │ ├── resnet34.py │ │ ├── resnet34_cifar.py │ │ ├── resnet50.py │ │ ├── resnet50_cifar.py │ │ ├── resnetv1d101.py │ │ ├── resnetv1d152.py │ │ ├── resnetv1d50.py │ │ ├── resnext101_32x4d.py │ │ ├── resnext101_32x8d.py │ │ ├── resnext152_32x4d.py │ │ ├── resnext50_32x4d.py │ │ ├── seresnet101.py │ │ ├── seresnet50.py │ │ ├── seresnext101_32x4d.py │ │ ├── seresnext50_32x4d.py │ │ ├── shufflenet_v1_1x.py │ │ ├── shufflenet_v2_1x.py │ │ ├── vgg11.py │ │ ├── vgg11bn.py │ │ ├── vgg13.py │ │ ├── vgg13bn.py │ │ ├── vgg16.py │ │ ├── vgg16bn.py │ │ ├── vgg19.py │ │ ├── vgg19bn.py │ │ └── weightnet_1x.py │ └── schedules │ │ ├── cifar10_bs128.py │ │ ├── imagenet_bs1024_linearlr_bn_nowd.py │ │ ├── imagenet_bs2048.py │ │ ├── imagenet_bs2048_coslr.py │ │ ├── imagenet_bs256.py │ │ ├── imagenet_bs256_140e.py │ │ ├── imagenet_bs256_coslr.py │ │ └── imagenet_bs256_epochstep.py ├── lenet │ ├── README.md │ └── lenet5_mnist.py ├── mobilenet_v2 │ ├── README.md │ └── mobilenet_v2_b32x8_imagenet.py ├── resnet │ ├── README.md │ ├── resnet101_b16x8_cifar10.py │ ├── resnet101_b32x8_imagenet.py │ ├── resnet152_b16x8_cifar10.py │ ├── resnet152_b32x8_imagenet.py │ ├── resnet18_b16x8_cifar10.py │ ├── resnet18_b32x8_imagenet.py │ ├── resnet34_b16x8_cifar10.py │ ├── resnet34_b32x8_imagenet.py │ ├── resnet50_b16x8_cifar10.py │ ├── resnet50_b32x8_coslr_imagenet.py │ ├── resnet50_b32x8_imagenet.py │ ├── resnet50_b32x8_label_smooth_imagenet.py │ ├── resnet50_b64x32_warmup_coslr_imagenet.py │ ├── resnet50_b64x32_warmup_imagenet.py │ ├── resnet50_b64x32_warmup_label_smooth_imagenet.py │ ├── resnetv1d101_b32x8_imagenet.py │ ├── resnetv1d152_b32x8_imagenet.py │ └── resnetv1d50_b32x8_imagenet.py ├── resnext │ ├── README.md │ ├── resnext101_32x4d_b32x8_imagenet.py │ ├── resnext101_32x8d_b32x8_imagenet.py │ ├── resnext152_32x4d_b32x8_imagenet.py │ └── resnext50_32x4d_b32x8_imagenet.py ├── seresnet │ ├── README.md │ ├── seresnet101_b32x8_imagenet.py │ └── seresnet50_b32x8_imagenet.py ├── seresnext │ ├── README.md │ ├── seresnext101_32x4d_b32x8_imagenet.py │ └── seresnext50_32x4d_b32x8_imagenet.py ├── shufflenet_v1 │ ├── README.md │ └── shufflenet_v1_1x_b64x16_linearlr_bn_nowd_imagenet.py ├── shufflenet_v2 │ ├── README.md │ ├── shufflenet_1x_b256x4_imagenet.py │ ├── shufflenet_v2_1x_b64x16_linearlr_bn_nowd_imagenet.py │ └── weightnet_1x_b256x4_imagenet.py └── vgg │ ├── README.md │ ├── vgg11_imagenet.py │ ├── vgg11bn_imagenet.py │ ├── vgg13_imagenet.py │ ├── vgg13bn_imagenet.py │ ├── vgg16_imagenet.py │ ├── vgg16_voc.py │ ├── vgg16bn_imagenet.py │ ├── vgg19_imagenet.py │ └── vgg19bn_imagenet.py ├── demo ├── demo.JPEG └── image_demo.py ├── docs ├── Makefile ├── changelog.md ├── conf.py ├── getting_started.md ├── index.rst ├── install.md ├── model_zoo.md ├── stat.py └── tutorials │ ├── data_pipeline.md │ ├── finetune.md │ ├── new_dataset.md │ └── new_modules.md ├── mmcls ├── __init__.py ├── apis │ ├── __init__.py │ ├── inference.py │ ├── test.py │ └── train.py ├── core │ ├── __init__.py │ ├── evaluation │ │ ├── __init__.py │ │ ├── eval_hooks.py │ │ ├── eval_metrics.py │ │ ├── mean_ap.py │ │ └── multilabel_eval_metrics.py │ ├── fp16 │ │ ├── __init__.py │ │ ├── decorators.py │ │ ├── hooks.py │ │ └── utils.py │ └── utils │ │ ├── __init__.py │ │ ├── dist_utils.py │ │ └── misc.py ├── datasets │ ├── __init__.py │ ├── base_dataset.py │ ├── builder.py │ ├── cifar.py │ ├── dataset_wrappers.py │ ├── imagenet.py │ ├── mnist.py │ ├── multi_label.py │ ├── pipelines │ │ ├── __init__.py │ │ ├── compose.py │ │ ├── formating.py │ │ ├── loading.py │ │ └── transforms.py │ ├── samplers │ │ ├── __init__.py │ │ └── distributed_sampler.py │ ├── utils.py │ └── voc.py ├── models │ ├── __init__.py │ ├── backbones │ │ ├── __init__.py │ │ ├── alexnet.py │ │ ├── base_backbone.py │ │ ├── lenet.py │ │ ├── mobilenet_v2.py │ │ ├── mobilenet_v3.py │ │ ├── regnet.py │ │ ├── resnest.py │ │ ├── resnet.py │ │ ├── resnet_cifar.py │ │ ├── resnext.py │ │ ├── seresnet.py │ │ ├── seresnext.py │ │ ├── shufflenet_v1.py │ │ ├── shufflenet_v2.py │ │ ├── vgg.py │ │ └── weightnet.py │ ├── builder.py │ ├── classifiers │ │ ├── __init__.py │ │ ├── base.py │ │ └── image.py │ ├── heads │ │ ├── __init__.py │ │ ├── base_head.py │ │ ├── cls_head.py │ │ ├── linear_head.py │ │ ├── multi_label_head.py │ │ └── multi_label_linear_head.py │ ├── losses │ │ ├── __init__.py │ │ ├── accuracy.py │ │ ├── asymmetric_loss.py │ │ ├── cross_entropy_loss.py │ │ ├── focal_loss.py │ │ ├── label_smooth_loss.py │ │ └── utils.py │ ├── necks │ │ ├── __init__.py │ │ └── gap.py │ └── utils │ │ ├── __init__.py │ │ ├── channel_shuffle.py │ │ ├── inverted_residual.py │ │ ├── make_divisible.py │ │ └── se_layer.py ├── utils │ ├── __init__.py │ ├── collect_env.py │ └── logger.py └── version.py ├── requirements.txt ├── requirements ├── docs.txt ├── optional.txt ├── readthedocs.txt ├── runtime.txt └── tests.txt ├── resources └── mmcls-logo.png ├── setup.cfg ├── setup.py ├── tests ├── data │ ├── color.jpg │ └── gray.jpg ├── test_backbones │ ├── test_mobilenet_v2.py │ ├── test_mobilenet_v3.py │ ├── test_regnet.py │ ├── test_resnest.py │ ├── test_resnet.py │ ├── test_resnet_cifar.py │ ├── test_resnext.py │ ├── test_seresnet.py │ ├── test_seresnext.py │ ├── test_shufflenet_v1.py │ ├── test_shufflenet_v2.py │ ├── test_utils.py │ └── test_vgg.py ├── test_dataset.py ├── test_eval_hook.py ├── test_heads.py ├── test_losses.py ├── test_metrics.py └── test_pipelines │ ├── test_loading.py │ └── test_transform.py └── tools ├── analyze_results.py ├── convert_models ├── mobilenetv2_to_mmcls.py ├── shufflenetv2_to_mmcls.py └── vgg_to_mmcls.py ├── dist_test.sh ├── dist_train.sh ├── get_flops.py ├── onnx2tensorrt.py ├── publish_model.py ├── pytorch2onnx.py ├── slurm_test.sh ├── slurm_train.sh ├── test.py └── train.py /.github/CONTRIBUTING.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DequanWang/weightnet.pytorch/HEAD/.github/CONTRIBUTING.md -------------------------------------------------------------------------------- /.github/workflows/build.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DequanWang/weightnet.pytorch/HEAD/.github/workflows/build.yml -------------------------------------------------------------------------------- /.github/workflows/deploy.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DequanWang/weightnet.pytorch/HEAD/.github/workflows/deploy.yml -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DequanWang/weightnet.pytorch/HEAD/.gitignore -------------------------------------------------------------------------------- /.pre-commit-config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DequanWang/weightnet.pytorch/HEAD/.pre-commit-config.yaml -------------------------------------------------------------------------------- /.readthedocs.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DequanWang/weightnet.pytorch/HEAD/.readthedocs.yml -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DequanWang/weightnet.pytorch/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DequanWang/weightnet.pytorch/HEAD/README.md -------------------------------------------------------------------------------- /ckpts/shufflenet_1x_b256x4_imagenet/shufflenet_1x_b256x4_imagenet.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DequanWang/weightnet.pytorch/HEAD/ckpts/shufflenet_1x_b256x4_imagenet/shufflenet_1x_b256x4_imagenet.json -------------------------------------------------------------------------------- /ckpts/shufflenet_1x_b256x4_imagenet/shufflenet_1x_b256x4_imagenet.log: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DequanWang/weightnet.pytorch/HEAD/ckpts/shufflenet_1x_b256x4_imagenet/shufflenet_1x_b256x4_imagenet.log -------------------------------------------------------------------------------- /ckpts/shufflenet_1x_b256x4_imagenet/shufflenet_1x_b256x4_imagenet.pth: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DequanWang/weightnet.pytorch/HEAD/ckpts/shufflenet_1x_b256x4_imagenet/shufflenet_1x_b256x4_imagenet.pth -------------------------------------------------------------------------------- /ckpts/weightnet_1x_b256x4_imagenet/weightnet_1x_b256x4_imagenet.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DequanWang/weightnet.pytorch/HEAD/ckpts/weightnet_1x_b256x4_imagenet/weightnet_1x_b256x4_imagenet.json -------------------------------------------------------------------------------- /ckpts/weightnet_1x_b256x4_imagenet/weightnet_1x_b256x4_imagenet.log: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DequanWang/weightnet.pytorch/HEAD/ckpts/weightnet_1x_b256x4_imagenet/weightnet_1x_b256x4_imagenet.log -------------------------------------------------------------------------------- /ckpts/weightnet_1x_b256x4_imagenet/weightnet_1x_b256x4_imagenet.pth: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DequanWang/weightnet.pytorch/HEAD/ckpts/weightnet_1x_b256x4_imagenet/weightnet_1x_b256x4_imagenet.pth -------------------------------------------------------------------------------- /configs/_base_/datasets/cifar10_bs16.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DequanWang/weightnet.pytorch/HEAD/configs/_base_/datasets/cifar10_bs16.py -------------------------------------------------------------------------------- /configs/_base_/datasets/imagenet_bs32.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DequanWang/weightnet.pytorch/HEAD/configs/_base_/datasets/imagenet_bs32.py -------------------------------------------------------------------------------- /configs/_base_/datasets/imagenet_bs32_pil_resize.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DequanWang/weightnet.pytorch/HEAD/configs/_base_/datasets/imagenet_bs32_pil_resize.py -------------------------------------------------------------------------------- /configs/_base_/datasets/imagenet_bs64.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DequanWang/weightnet.pytorch/HEAD/configs/_base_/datasets/imagenet_bs64.py -------------------------------------------------------------------------------- /configs/_base_/datasets/imagenet_bs64_pil_resize.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DequanWang/weightnet.pytorch/HEAD/configs/_base_/datasets/imagenet_bs64_pil_resize.py -------------------------------------------------------------------------------- /configs/_base_/datasets/voc_bs16.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DequanWang/weightnet.pytorch/HEAD/configs/_base_/datasets/voc_bs16.py -------------------------------------------------------------------------------- /configs/_base_/default_runtime.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DequanWang/weightnet.pytorch/HEAD/configs/_base_/default_runtime.py -------------------------------------------------------------------------------- /configs/_base_/models/mobilenet_v2_1x.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DequanWang/weightnet.pytorch/HEAD/configs/_base_/models/mobilenet_v2_1x.py -------------------------------------------------------------------------------- /configs/_base_/models/resnest101.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DequanWang/weightnet.pytorch/HEAD/configs/_base_/models/resnest101.py -------------------------------------------------------------------------------- /configs/_base_/models/resnest200.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DequanWang/weightnet.pytorch/HEAD/configs/_base_/models/resnest200.py -------------------------------------------------------------------------------- /configs/_base_/models/resnest269.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DequanWang/weightnet.pytorch/HEAD/configs/_base_/models/resnest269.py -------------------------------------------------------------------------------- /configs/_base_/models/resnest50.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DequanWang/weightnet.pytorch/HEAD/configs/_base_/models/resnest50.py -------------------------------------------------------------------------------- /configs/_base_/models/resnet101.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DequanWang/weightnet.pytorch/HEAD/configs/_base_/models/resnet101.py -------------------------------------------------------------------------------- /configs/_base_/models/resnet101_cifar.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DequanWang/weightnet.pytorch/HEAD/configs/_base_/models/resnet101_cifar.py -------------------------------------------------------------------------------- /configs/_base_/models/resnet152.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DequanWang/weightnet.pytorch/HEAD/configs/_base_/models/resnet152.py -------------------------------------------------------------------------------- /configs/_base_/models/resnet152_cifar.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DequanWang/weightnet.pytorch/HEAD/configs/_base_/models/resnet152_cifar.py -------------------------------------------------------------------------------- /configs/_base_/models/resnet18.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DequanWang/weightnet.pytorch/HEAD/configs/_base_/models/resnet18.py -------------------------------------------------------------------------------- /configs/_base_/models/resnet18_cifar.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DequanWang/weightnet.pytorch/HEAD/configs/_base_/models/resnet18_cifar.py -------------------------------------------------------------------------------- /configs/_base_/models/resnet34.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DequanWang/weightnet.pytorch/HEAD/configs/_base_/models/resnet34.py -------------------------------------------------------------------------------- /configs/_base_/models/resnet34_cifar.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DequanWang/weightnet.pytorch/HEAD/configs/_base_/models/resnet34_cifar.py -------------------------------------------------------------------------------- /configs/_base_/models/resnet50.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DequanWang/weightnet.pytorch/HEAD/configs/_base_/models/resnet50.py -------------------------------------------------------------------------------- /configs/_base_/models/resnet50_cifar.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DequanWang/weightnet.pytorch/HEAD/configs/_base_/models/resnet50_cifar.py -------------------------------------------------------------------------------- /configs/_base_/models/resnetv1d101.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DequanWang/weightnet.pytorch/HEAD/configs/_base_/models/resnetv1d101.py -------------------------------------------------------------------------------- /configs/_base_/models/resnetv1d152.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DequanWang/weightnet.pytorch/HEAD/configs/_base_/models/resnetv1d152.py -------------------------------------------------------------------------------- /configs/_base_/models/resnetv1d50.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DequanWang/weightnet.pytorch/HEAD/configs/_base_/models/resnetv1d50.py -------------------------------------------------------------------------------- /configs/_base_/models/resnext101_32x4d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DequanWang/weightnet.pytorch/HEAD/configs/_base_/models/resnext101_32x4d.py -------------------------------------------------------------------------------- /configs/_base_/models/resnext101_32x8d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DequanWang/weightnet.pytorch/HEAD/configs/_base_/models/resnext101_32x8d.py -------------------------------------------------------------------------------- /configs/_base_/models/resnext152_32x4d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DequanWang/weightnet.pytorch/HEAD/configs/_base_/models/resnext152_32x4d.py -------------------------------------------------------------------------------- /configs/_base_/models/resnext50_32x4d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DequanWang/weightnet.pytorch/HEAD/configs/_base_/models/resnext50_32x4d.py -------------------------------------------------------------------------------- /configs/_base_/models/seresnet101.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DequanWang/weightnet.pytorch/HEAD/configs/_base_/models/seresnet101.py -------------------------------------------------------------------------------- /configs/_base_/models/seresnet50.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DequanWang/weightnet.pytorch/HEAD/configs/_base_/models/seresnet50.py -------------------------------------------------------------------------------- /configs/_base_/models/seresnext101_32x4d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DequanWang/weightnet.pytorch/HEAD/configs/_base_/models/seresnext101_32x4d.py -------------------------------------------------------------------------------- /configs/_base_/models/seresnext50_32x4d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DequanWang/weightnet.pytorch/HEAD/configs/_base_/models/seresnext50_32x4d.py -------------------------------------------------------------------------------- /configs/_base_/models/shufflenet_v1_1x.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DequanWang/weightnet.pytorch/HEAD/configs/_base_/models/shufflenet_v1_1x.py -------------------------------------------------------------------------------- /configs/_base_/models/shufflenet_v2_1x.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DequanWang/weightnet.pytorch/HEAD/configs/_base_/models/shufflenet_v2_1x.py -------------------------------------------------------------------------------- /configs/_base_/models/vgg11.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DequanWang/weightnet.pytorch/HEAD/configs/_base_/models/vgg11.py -------------------------------------------------------------------------------- /configs/_base_/models/vgg11bn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DequanWang/weightnet.pytorch/HEAD/configs/_base_/models/vgg11bn.py -------------------------------------------------------------------------------- /configs/_base_/models/vgg13.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DequanWang/weightnet.pytorch/HEAD/configs/_base_/models/vgg13.py -------------------------------------------------------------------------------- /configs/_base_/models/vgg13bn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DequanWang/weightnet.pytorch/HEAD/configs/_base_/models/vgg13bn.py -------------------------------------------------------------------------------- /configs/_base_/models/vgg16.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DequanWang/weightnet.pytorch/HEAD/configs/_base_/models/vgg16.py -------------------------------------------------------------------------------- /configs/_base_/models/vgg16bn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DequanWang/weightnet.pytorch/HEAD/configs/_base_/models/vgg16bn.py -------------------------------------------------------------------------------- /configs/_base_/models/vgg19.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DequanWang/weightnet.pytorch/HEAD/configs/_base_/models/vgg19.py -------------------------------------------------------------------------------- /configs/_base_/models/vgg19bn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DequanWang/weightnet.pytorch/HEAD/configs/_base_/models/vgg19bn.py -------------------------------------------------------------------------------- /configs/_base_/models/weightnet_1x.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DequanWang/weightnet.pytorch/HEAD/configs/_base_/models/weightnet_1x.py -------------------------------------------------------------------------------- /configs/_base_/schedules/cifar10_bs128.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DequanWang/weightnet.pytorch/HEAD/configs/_base_/schedules/cifar10_bs128.py -------------------------------------------------------------------------------- /configs/_base_/schedules/imagenet_bs1024_linearlr_bn_nowd.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DequanWang/weightnet.pytorch/HEAD/configs/_base_/schedules/imagenet_bs1024_linearlr_bn_nowd.py -------------------------------------------------------------------------------- /configs/_base_/schedules/imagenet_bs2048.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DequanWang/weightnet.pytorch/HEAD/configs/_base_/schedules/imagenet_bs2048.py -------------------------------------------------------------------------------- /configs/_base_/schedules/imagenet_bs2048_coslr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DequanWang/weightnet.pytorch/HEAD/configs/_base_/schedules/imagenet_bs2048_coslr.py -------------------------------------------------------------------------------- /configs/_base_/schedules/imagenet_bs256.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DequanWang/weightnet.pytorch/HEAD/configs/_base_/schedules/imagenet_bs256.py -------------------------------------------------------------------------------- /configs/_base_/schedules/imagenet_bs256_140e.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DequanWang/weightnet.pytorch/HEAD/configs/_base_/schedules/imagenet_bs256_140e.py -------------------------------------------------------------------------------- /configs/_base_/schedules/imagenet_bs256_coslr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DequanWang/weightnet.pytorch/HEAD/configs/_base_/schedules/imagenet_bs256_coslr.py -------------------------------------------------------------------------------- /configs/_base_/schedules/imagenet_bs256_epochstep.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DequanWang/weightnet.pytorch/HEAD/configs/_base_/schedules/imagenet_bs256_epochstep.py -------------------------------------------------------------------------------- /configs/lenet/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DequanWang/weightnet.pytorch/HEAD/configs/lenet/README.md -------------------------------------------------------------------------------- /configs/lenet/lenet5_mnist.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DequanWang/weightnet.pytorch/HEAD/configs/lenet/lenet5_mnist.py -------------------------------------------------------------------------------- /configs/mobilenet_v2/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DequanWang/weightnet.pytorch/HEAD/configs/mobilenet_v2/README.md -------------------------------------------------------------------------------- /configs/mobilenet_v2/mobilenet_v2_b32x8_imagenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DequanWang/weightnet.pytorch/HEAD/configs/mobilenet_v2/mobilenet_v2_b32x8_imagenet.py -------------------------------------------------------------------------------- /configs/resnet/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DequanWang/weightnet.pytorch/HEAD/configs/resnet/README.md -------------------------------------------------------------------------------- /configs/resnet/resnet101_b16x8_cifar10.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DequanWang/weightnet.pytorch/HEAD/configs/resnet/resnet101_b16x8_cifar10.py -------------------------------------------------------------------------------- /configs/resnet/resnet101_b32x8_imagenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DequanWang/weightnet.pytorch/HEAD/configs/resnet/resnet101_b32x8_imagenet.py -------------------------------------------------------------------------------- /configs/resnet/resnet152_b16x8_cifar10.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DequanWang/weightnet.pytorch/HEAD/configs/resnet/resnet152_b16x8_cifar10.py -------------------------------------------------------------------------------- /configs/resnet/resnet152_b32x8_imagenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DequanWang/weightnet.pytorch/HEAD/configs/resnet/resnet152_b32x8_imagenet.py -------------------------------------------------------------------------------- /configs/resnet/resnet18_b16x8_cifar10.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DequanWang/weightnet.pytorch/HEAD/configs/resnet/resnet18_b16x8_cifar10.py -------------------------------------------------------------------------------- /configs/resnet/resnet18_b32x8_imagenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DequanWang/weightnet.pytorch/HEAD/configs/resnet/resnet18_b32x8_imagenet.py -------------------------------------------------------------------------------- /configs/resnet/resnet34_b16x8_cifar10.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DequanWang/weightnet.pytorch/HEAD/configs/resnet/resnet34_b16x8_cifar10.py -------------------------------------------------------------------------------- /configs/resnet/resnet34_b32x8_imagenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DequanWang/weightnet.pytorch/HEAD/configs/resnet/resnet34_b32x8_imagenet.py -------------------------------------------------------------------------------- /configs/resnet/resnet50_b16x8_cifar10.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DequanWang/weightnet.pytorch/HEAD/configs/resnet/resnet50_b16x8_cifar10.py -------------------------------------------------------------------------------- /configs/resnet/resnet50_b32x8_coslr_imagenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DequanWang/weightnet.pytorch/HEAD/configs/resnet/resnet50_b32x8_coslr_imagenet.py -------------------------------------------------------------------------------- /configs/resnet/resnet50_b32x8_imagenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DequanWang/weightnet.pytorch/HEAD/configs/resnet/resnet50_b32x8_imagenet.py -------------------------------------------------------------------------------- /configs/resnet/resnet50_b32x8_label_smooth_imagenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DequanWang/weightnet.pytorch/HEAD/configs/resnet/resnet50_b32x8_label_smooth_imagenet.py -------------------------------------------------------------------------------- /configs/resnet/resnet50_b64x32_warmup_coslr_imagenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DequanWang/weightnet.pytorch/HEAD/configs/resnet/resnet50_b64x32_warmup_coslr_imagenet.py -------------------------------------------------------------------------------- /configs/resnet/resnet50_b64x32_warmup_imagenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DequanWang/weightnet.pytorch/HEAD/configs/resnet/resnet50_b64x32_warmup_imagenet.py -------------------------------------------------------------------------------- /configs/resnet/resnet50_b64x32_warmup_label_smooth_imagenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DequanWang/weightnet.pytorch/HEAD/configs/resnet/resnet50_b64x32_warmup_label_smooth_imagenet.py -------------------------------------------------------------------------------- /configs/resnet/resnetv1d101_b32x8_imagenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DequanWang/weightnet.pytorch/HEAD/configs/resnet/resnetv1d101_b32x8_imagenet.py -------------------------------------------------------------------------------- /configs/resnet/resnetv1d152_b32x8_imagenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DequanWang/weightnet.pytorch/HEAD/configs/resnet/resnetv1d152_b32x8_imagenet.py -------------------------------------------------------------------------------- /configs/resnet/resnetv1d50_b32x8_imagenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DequanWang/weightnet.pytorch/HEAD/configs/resnet/resnetv1d50_b32x8_imagenet.py -------------------------------------------------------------------------------- /configs/resnext/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DequanWang/weightnet.pytorch/HEAD/configs/resnext/README.md -------------------------------------------------------------------------------- /configs/resnext/resnext101_32x4d_b32x8_imagenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DequanWang/weightnet.pytorch/HEAD/configs/resnext/resnext101_32x4d_b32x8_imagenet.py -------------------------------------------------------------------------------- /configs/resnext/resnext101_32x8d_b32x8_imagenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DequanWang/weightnet.pytorch/HEAD/configs/resnext/resnext101_32x8d_b32x8_imagenet.py -------------------------------------------------------------------------------- /configs/resnext/resnext152_32x4d_b32x8_imagenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DequanWang/weightnet.pytorch/HEAD/configs/resnext/resnext152_32x4d_b32x8_imagenet.py -------------------------------------------------------------------------------- /configs/resnext/resnext50_32x4d_b32x8_imagenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DequanWang/weightnet.pytorch/HEAD/configs/resnext/resnext50_32x4d_b32x8_imagenet.py -------------------------------------------------------------------------------- /configs/seresnet/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DequanWang/weightnet.pytorch/HEAD/configs/seresnet/README.md -------------------------------------------------------------------------------- /configs/seresnet/seresnet101_b32x8_imagenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DequanWang/weightnet.pytorch/HEAD/configs/seresnet/seresnet101_b32x8_imagenet.py -------------------------------------------------------------------------------- /configs/seresnet/seresnet50_b32x8_imagenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DequanWang/weightnet.pytorch/HEAD/configs/seresnet/seresnet50_b32x8_imagenet.py -------------------------------------------------------------------------------- /configs/seresnext/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DequanWang/weightnet.pytorch/HEAD/configs/seresnext/README.md -------------------------------------------------------------------------------- /configs/seresnext/seresnext101_32x4d_b32x8_imagenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DequanWang/weightnet.pytorch/HEAD/configs/seresnext/seresnext101_32x4d_b32x8_imagenet.py -------------------------------------------------------------------------------- /configs/seresnext/seresnext50_32x4d_b32x8_imagenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DequanWang/weightnet.pytorch/HEAD/configs/seresnext/seresnext50_32x4d_b32x8_imagenet.py -------------------------------------------------------------------------------- /configs/shufflenet_v1/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DequanWang/weightnet.pytorch/HEAD/configs/shufflenet_v1/README.md -------------------------------------------------------------------------------- /configs/shufflenet_v1/shufflenet_v1_1x_b64x16_linearlr_bn_nowd_imagenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DequanWang/weightnet.pytorch/HEAD/configs/shufflenet_v1/shufflenet_v1_1x_b64x16_linearlr_bn_nowd_imagenet.py -------------------------------------------------------------------------------- /configs/shufflenet_v2/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DequanWang/weightnet.pytorch/HEAD/configs/shufflenet_v2/README.md -------------------------------------------------------------------------------- /configs/shufflenet_v2/shufflenet_1x_b256x4_imagenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DequanWang/weightnet.pytorch/HEAD/configs/shufflenet_v2/shufflenet_1x_b256x4_imagenet.py -------------------------------------------------------------------------------- /configs/shufflenet_v2/shufflenet_v2_1x_b64x16_linearlr_bn_nowd_imagenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DequanWang/weightnet.pytorch/HEAD/configs/shufflenet_v2/shufflenet_v2_1x_b64x16_linearlr_bn_nowd_imagenet.py -------------------------------------------------------------------------------- /configs/shufflenet_v2/weightnet_1x_b256x4_imagenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DequanWang/weightnet.pytorch/HEAD/configs/shufflenet_v2/weightnet_1x_b256x4_imagenet.py -------------------------------------------------------------------------------- /configs/vgg/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DequanWang/weightnet.pytorch/HEAD/configs/vgg/README.md -------------------------------------------------------------------------------- /configs/vgg/vgg11_imagenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DequanWang/weightnet.pytorch/HEAD/configs/vgg/vgg11_imagenet.py -------------------------------------------------------------------------------- /configs/vgg/vgg11bn_imagenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DequanWang/weightnet.pytorch/HEAD/configs/vgg/vgg11bn_imagenet.py -------------------------------------------------------------------------------- /configs/vgg/vgg13_imagenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DequanWang/weightnet.pytorch/HEAD/configs/vgg/vgg13_imagenet.py -------------------------------------------------------------------------------- /configs/vgg/vgg13bn_imagenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DequanWang/weightnet.pytorch/HEAD/configs/vgg/vgg13bn_imagenet.py -------------------------------------------------------------------------------- /configs/vgg/vgg16_imagenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DequanWang/weightnet.pytorch/HEAD/configs/vgg/vgg16_imagenet.py -------------------------------------------------------------------------------- /configs/vgg/vgg16_voc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DequanWang/weightnet.pytorch/HEAD/configs/vgg/vgg16_voc.py -------------------------------------------------------------------------------- /configs/vgg/vgg16bn_imagenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DequanWang/weightnet.pytorch/HEAD/configs/vgg/vgg16bn_imagenet.py -------------------------------------------------------------------------------- /configs/vgg/vgg19_imagenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DequanWang/weightnet.pytorch/HEAD/configs/vgg/vgg19_imagenet.py -------------------------------------------------------------------------------- /configs/vgg/vgg19bn_imagenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DequanWang/weightnet.pytorch/HEAD/configs/vgg/vgg19bn_imagenet.py -------------------------------------------------------------------------------- /demo/demo.JPEG: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DequanWang/weightnet.pytorch/HEAD/demo/demo.JPEG -------------------------------------------------------------------------------- /demo/image_demo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DequanWang/weightnet.pytorch/HEAD/demo/image_demo.py -------------------------------------------------------------------------------- /docs/Makefile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DequanWang/weightnet.pytorch/HEAD/docs/Makefile -------------------------------------------------------------------------------- /docs/changelog.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DequanWang/weightnet.pytorch/HEAD/docs/changelog.md -------------------------------------------------------------------------------- /docs/conf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DequanWang/weightnet.pytorch/HEAD/docs/conf.py -------------------------------------------------------------------------------- /docs/getting_started.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DequanWang/weightnet.pytorch/HEAD/docs/getting_started.md -------------------------------------------------------------------------------- /docs/index.rst: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DequanWang/weightnet.pytorch/HEAD/docs/index.rst -------------------------------------------------------------------------------- /docs/install.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DequanWang/weightnet.pytorch/HEAD/docs/install.md -------------------------------------------------------------------------------- /docs/model_zoo.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DequanWang/weightnet.pytorch/HEAD/docs/model_zoo.md -------------------------------------------------------------------------------- /docs/stat.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DequanWang/weightnet.pytorch/HEAD/docs/stat.py -------------------------------------------------------------------------------- /docs/tutorials/data_pipeline.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DequanWang/weightnet.pytorch/HEAD/docs/tutorials/data_pipeline.md -------------------------------------------------------------------------------- /docs/tutorials/finetune.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DequanWang/weightnet.pytorch/HEAD/docs/tutorials/finetune.md -------------------------------------------------------------------------------- /docs/tutorials/new_dataset.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DequanWang/weightnet.pytorch/HEAD/docs/tutorials/new_dataset.md -------------------------------------------------------------------------------- /docs/tutorials/new_modules.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DequanWang/weightnet.pytorch/HEAD/docs/tutorials/new_modules.md -------------------------------------------------------------------------------- /mmcls/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DequanWang/weightnet.pytorch/HEAD/mmcls/__init__.py -------------------------------------------------------------------------------- /mmcls/apis/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DequanWang/weightnet.pytorch/HEAD/mmcls/apis/__init__.py -------------------------------------------------------------------------------- /mmcls/apis/inference.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DequanWang/weightnet.pytorch/HEAD/mmcls/apis/inference.py -------------------------------------------------------------------------------- /mmcls/apis/test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DequanWang/weightnet.pytorch/HEAD/mmcls/apis/test.py -------------------------------------------------------------------------------- /mmcls/apis/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DequanWang/weightnet.pytorch/HEAD/mmcls/apis/train.py -------------------------------------------------------------------------------- /mmcls/core/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DequanWang/weightnet.pytorch/HEAD/mmcls/core/__init__.py -------------------------------------------------------------------------------- /mmcls/core/evaluation/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DequanWang/weightnet.pytorch/HEAD/mmcls/core/evaluation/__init__.py -------------------------------------------------------------------------------- /mmcls/core/evaluation/eval_hooks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DequanWang/weightnet.pytorch/HEAD/mmcls/core/evaluation/eval_hooks.py -------------------------------------------------------------------------------- /mmcls/core/evaluation/eval_metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DequanWang/weightnet.pytorch/HEAD/mmcls/core/evaluation/eval_metrics.py -------------------------------------------------------------------------------- /mmcls/core/evaluation/mean_ap.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DequanWang/weightnet.pytorch/HEAD/mmcls/core/evaluation/mean_ap.py -------------------------------------------------------------------------------- /mmcls/core/evaluation/multilabel_eval_metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DequanWang/weightnet.pytorch/HEAD/mmcls/core/evaluation/multilabel_eval_metrics.py -------------------------------------------------------------------------------- /mmcls/core/fp16/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DequanWang/weightnet.pytorch/HEAD/mmcls/core/fp16/__init__.py -------------------------------------------------------------------------------- /mmcls/core/fp16/decorators.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DequanWang/weightnet.pytorch/HEAD/mmcls/core/fp16/decorators.py -------------------------------------------------------------------------------- /mmcls/core/fp16/hooks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DequanWang/weightnet.pytorch/HEAD/mmcls/core/fp16/hooks.py -------------------------------------------------------------------------------- /mmcls/core/fp16/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DequanWang/weightnet.pytorch/HEAD/mmcls/core/fp16/utils.py -------------------------------------------------------------------------------- /mmcls/core/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DequanWang/weightnet.pytorch/HEAD/mmcls/core/utils/__init__.py -------------------------------------------------------------------------------- /mmcls/core/utils/dist_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DequanWang/weightnet.pytorch/HEAD/mmcls/core/utils/dist_utils.py -------------------------------------------------------------------------------- /mmcls/core/utils/misc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DequanWang/weightnet.pytorch/HEAD/mmcls/core/utils/misc.py -------------------------------------------------------------------------------- /mmcls/datasets/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DequanWang/weightnet.pytorch/HEAD/mmcls/datasets/__init__.py -------------------------------------------------------------------------------- /mmcls/datasets/base_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DequanWang/weightnet.pytorch/HEAD/mmcls/datasets/base_dataset.py -------------------------------------------------------------------------------- /mmcls/datasets/builder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DequanWang/weightnet.pytorch/HEAD/mmcls/datasets/builder.py -------------------------------------------------------------------------------- /mmcls/datasets/cifar.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DequanWang/weightnet.pytorch/HEAD/mmcls/datasets/cifar.py -------------------------------------------------------------------------------- /mmcls/datasets/dataset_wrappers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DequanWang/weightnet.pytorch/HEAD/mmcls/datasets/dataset_wrappers.py -------------------------------------------------------------------------------- /mmcls/datasets/imagenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DequanWang/weightnet.pytorch/HEAD/mmcls/datasets/imagenet.py -------------------------------------------------------------------------------- /mmcls/datasets/mnist.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DequanWang/weightnet.pytorch/HEAD/mmcls/datasets/mnist.py -------------------------------------------------------------------------------- /mmcls/datasets/multi_label.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DequanWang/weightnet.pytorch/HEAD/mmcls/datasets/multi_label.py -------------------------------------------------------------------------------- /mmcls/datasets/pipelines/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DequanWang/weightnet.pytorch/HEAD/mmcls/datasets/pipelines/__init__.py -------------------------------------------------------------------------------- /mmcls/datasets/pipelines/compose.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DequanWang/weightnet.pytorch/HEAD/mmcls/datasets/pipelines/compose.py -------------------------------------------------------------------------------- /mmcls/datasets/pipelines/formating.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DequanWang/weightnet.pytorch/HEAD/mmcls/datasets/pipelines/formating.py -------------------------------------------------------------------------------- /mmcls/datasets/pipelines/loading.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DequanWang/weightnet.pytorch/HEAD/mmcls/datasets/pipelines/loading.py -------------------------------------------------------------------------------- /mmcls/datasets/pipelines/transforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DequanWang/weightnet.pytorch/HEAD/mmcls/datasets/pipelines/transforms.py -------------------------------------------------------------------------------- /mmcls/datasets/samplers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DequanWang/weightnet.pytorch/HEAD/mmcls/datasets/samplers/__init__.py -------------------------------------------------------------------------------- /mmcls/datasets/samplers/distributed_sampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DequanWang/weightnet.pytorch/HEAD/mmcls/datasets/samplers/distributed_sampler.py -------------------------------------------------------------------------------- /mmcls/datasets/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DequanWang/weightnet.pytorch/HEAD/mmcls/datasets/utils.py -------------------------------------------------------------------------------- /mmcls/datasets/voc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DequanWang/weightnet.pytorch/HEAD/mmcls/datasets/voc.py -------------------------------------------------------------------------------- /mmcls/models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DequanWang/weightnet.pytorch/HEAD/mmcls/models/__init__.py -------------------------------------------------------------------------------- /mmcls/models/backbones/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DequanWang/weightnet.pytorch/HEAD/mmcls/models/backbones/__init__.py -------------------------------------------------------------------------------- /mmcls/models/backbones/alexnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DequanWang/weightnet.pytorch/HEAD/mmcls/models/backbones/alexnet.py -------------------------------------------------------------------------------- /mmcls/models/backbones/base_backbone.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DequanWang/weightnet.pytorch/HEAD/mmcls/models/backbones/base_backbone.py -------------------------------------------------------------------------------- /mmcls/models/backbones/lenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DequanWang/weightnet.pytorch/HEAD/mmcls/models/backbones/lenet.py -------------------------------------------------------------------------------- /mmcls/models/backbones/mobilenet_v2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DequanWang/weightnet.pytorch/HEAD/mmcls/models/backbones/mobilenet_v2.py -------------------------------------------------------------------------------- /mmcls/models/backbones/mobilenet_v3.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DequanWang/weightnet.pytorch/HEAD/mmcls/models/backbones/mobilenet_v3.py -------------------------------------------------------------------------------- /mmcls/models/backbones/regnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DequanWang/weightnet.pytorch/HEAD/mmcls/models/backbones/regnet.py -------------------------------------------------------------------------------- /mmcls/models/backbones/resnest.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DequanWang/weightnet.pytorch/HEAD/mmcls/models/backbones/resnest.py -------------------------------------------------------------------------------- /mmcls/models/backbones/resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DequanWang/weightnet.pytorch/HEAD/mmcls/models/backbones/resnet.py -------------------------------------------------------------------------------- /mmcls/models/backbones/resnet_cifar.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DequanWang/weightnet.pytorch/HEAD/mmcls/models/backbones/resnet_cifar.py -------------------------------------------------------------------------------- /mmcls/models/backbones/resnext.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DequanWang/weightnet.pytorch/HEAD/mmcls/models/backbones/resnext.py -------------------------------------------------------------------------------- /mmcls/models/backbones/seresnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DequanWang/weightnet.pytorch/HEAD/mmcls/models/backbones/seresnet.py -------------------------------------------------------------------------------- /mmcls/models/backbones/seresnext.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DequanWang/weightnet.pytorch/HEAD/mmcls/models/backbones/seresnext.py -------------------------------------------------------------------------------- /mmcls/models/backbones/shufflenet_v1.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DequanWang/weightnet.pytorch/HEAD/mmcls/models/backbones/shufflenet_v1.py -------------------------------------------------------------------------------- /mmcls/models/backbones/shufflenet_v2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DequanWang/weightnet.pytorch/HEAD/mmcls/models/backbones/shufflenet_v2.py -------------------------------------------------------------------------------- /mmcls/models/backbones/vgg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DequanWang/weightnet.pytorch/HEAD/mmcls/models/backbones/vgg.py -------------------------------------------------------------------------------- /mmcls/models/backbones/weightnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DequanWang/weightnet.pytorch/HEAD/mmcls/models/backbones/weightnet.py -------------------------------------------------------------------------------- /mmcls/models/builder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DequanWang/weightnet.pytorch/HEAD/mmcls/models/builder.py -------------------------------------------------------------------------------- /mmcls/models/classifiers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DequanWang/weightnet.pytorch/HEAD/mmcls/models/classifiers/__init__.py -------------------------------------------------------------------------------- /mmcls/models/classifiers/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DequanWang/weightnet.pytorch/HEAD/mmcls/models/classifiers/base.py -------------------------------------------------------------------------------- /mmcls/models/classifiers/image.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DequanWang/weightnet.pytorch/HEAD/mmcls/models/classifiers/image.py -------------------------------------------------------------------------------- /mmcls/models/heads/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DequanWang/weightnet.pytorch/HEAD/mmcls/models/heads/__init__.py -------------------------------------------------------------------------------- /mmcls/models/heads/base_head.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DequanWang/weightnet.pytorch/HEAD/mmcls/models/heads/base_head.py -------------------------------------------------------------------------------- /mmcls/models/heads/cls_head.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DequanWang/weightnet.pytorch/HEAD/mmcls/models/heads/cls_head.py -------------------------------------------------------------------------------- /mmcls/models/heads/linear_head.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DequanWang/weightnet.pytorch/HEAD/mmcls/models/heads/linear_head.py -------------------------------------------------------------------------------- /mmcls/models/heads/multi_label_head.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DequanWang/weightnet.pytorch/HEAD/mmcls/models/heads/multi_label_head.py -------------------------------------------------------------------------------- /mmcls/models/heads/multi_label_linear_head.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DequanWang/weightnet.pytorch/HEAD/mmcls/models/heads/multi_label_linear_head.py -------------------------------------------------------------------------------- /mmcls/models/losses/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DequanWang/weightnet.pytorch/HEAD/mmcls/models/losses/__init__.py -------------------------------------------------------------------------------- /mmcls/models/losses/accuracy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DequanWang/weightnet.pytorch/HEAD/mmcls/models/losses/accuracy.py -------------------------------------------------------------------------------- /mmcls/models/losses/asymmetric_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DequanWang/weightnet.pytorch/HEAD/mmcls/models/losses/asymmetric_loss.py -------------------------------------------------------------------------------- /mmcls/models/losses/cross_entropy_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DequanWang/weightnet.pytorch/HEAD/mmcls/models/losses/cross_entropy_loss.py -------------------------------------------------------------------------------- /mmcls/models/losses/focal_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DequanWang/weightnet.pytorch/HEAD/mmcls/models/losses/focal_loss.py -------------------------------------------------------------------------------- /mmcls/models/losses/label_smooth_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DequanWang/weightnet.pytorch/HEAD/mmcls/models/losses/label_smooth_loss.py -------------------------------------------------------------------------------- /mmcls/models/losses/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DequanWang/weightnet.pytorch/HEAD/mmcls/models/losses/utils.py -------------------------------------------------------------------------------- /mmcls/models/necks/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DequanWang/weightnet.pytorch/HEAD/mmcls/models/necks/__init__.py -------------------------------------------------------------------------------- /mmcls/models/necks/gap.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DequanWang/weightnet.pytorch/HEAD/mmcls/models/necks/gap.py -------------------------------------------------------------------------------- /mmcls/models/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DequanWang/weightnet.pytorch/HEAD/mmcls/models/utils/__init__.py -------------------------------------------------------------------------------- /mmcls/models/utils/channel_shuffle.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DequanWang/weightnet.pytorch/HEAD/mmcls/models/utils/channel_shuffle.py -------------------------------------------------------------------------------- /mmcls/models/utils/inverted_residual.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DequanWang/weightnet.pytorch/HEAD/mmcls/models/utils/inverted_residual.py -------------------------------------------------------------------------------- /mmcls/models/utils/make_divisible.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DequanWang/weightnet.pytorch/HEAD/mmcls/models/utils/make_divisible.py -------------------------------------------------------------------------------- /mmcls/models/utils/se_layer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DequanWang/weightnet.pytorch/HEAD/mmcls/models/utils/se_layer.py -------------------------------------------------------------------------------- /mmcls/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DequanWang/weightnet.pytorch/HEAD/mmcls/utils/__init__.py -------------------------------------------------------------------------------- /mmcls/utils/collect_env.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DequanWang/weightnet.pytorch/HEAD/mmcls/utils/collect_env.py -------------------------------------------------------------------------------- /mmcls/utils/logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DequanWang/weightnet.pytorch/HEAD/mmcls/utils/logger.py -------------------------------------------------------------------------------- /mmcls/version.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DequanWang/weightnet.pytorch/HEAD/mmcls/version.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DequanWang/weightnet.pytorch/HEAD/requirements.txt -------------------------------------------------------------------------------- /requirements/docs.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DequanWang/weightnet.pytorch/HEAD/requirements/docs.txt -------------------------------------------------------------------------------- /requirements/optional.txt: -------------------------------------------------------------------------------- 1 | albumentations>=0.3.2 2 | -------------------------------------------------------------------------------- /requirements/readthedocs.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DequanWang/weightnet.pytorch/HEAD/requirements/readthedocs.txt -------------------------------------------------------------------------------- /requirements/runtime.txt: -------------------------------------------------------------------------------- 1 | matplotlib 2 | numpy 3 | -------------------------------------------------------------------------------- /requirements/tests.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DequanWang/weightnet.pytorch/HEAD/requirements/tests.txt -------------------------------------------------------------------------------- /resources/mmcls-logo.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DequanWang/weightnet.pytorch/HEAD/resources/mmcls-logo.png -------------------------------------------------------------------------------- /setup.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DequanWang/weightnet.pytorch/HEAD/setup.cfg -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DequanWang/weightnet.pytorch/HEAD/setup.py -------------------------------------------------------------------------------- /tests/data/color.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DequanWang/weightnet.pytorch/HEAD/tests/data/color.jpg -------------------------------------------------------------------------------- /tests/data/gray.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DequanWang/weightnet.pytorch/HEAD/tests/data/gray.jpg -------------------------------------------------------------------------------- /tests/test_backbones/test_mobilenet_v2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DequanWang/weightnet.pytorch/HEAD/tests/test_backbones/test_mobilenet_v2.py -------------------------------------------------------------------------------- /tests/test_backbones/test_mobilenet_v3.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DequanWang/weightnet.pytorch/HEAD/tests/test_backbones/test_mobilenet_v3.py -------------------------------------------------------------------------------- /tests/test_backbones/test_regnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DequanWang/weightnet.pytorch/HEAD/tests/test_backbones/test_regnet.py -------------------------------------------------------------------------------- /tests/test_backbones/test_resnest.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DequanWang/weightnet.pytorch/HEAD/tests/test_backbones/test_resnest.py -------------------------------------------------------------------------------- /tests/test_backbones/test_resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DequanWang/weightnet.pytorch/HEAD/tests/test_backbones/test_resnet.py -------------------------------------------------------------------------------- /tests/test_backbones/test_resnet_cifar.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DequanWang/weightnet.pytorch/HEAD/tests/test_backbones/test_resnet_cifar.py -------------------------------------------------------------------------------- /tests/test_backbones/test_resnext.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DequanWang/weightnet.pytorch/HEAD/tests/test_backbones/test_resnext.py -------------------------------------------------------------------------------- /tests/test_backbones/test_seresnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DequanWang/weightnet.pytorch/HEAD/tests/test_backbones/test_seresnet.py -------------------------------------------------------------------------------- /tests/test_backbones/test_seresnext.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DequanWang/weightnet.pytorch/HEAD/tests/test_backbones/test_seresnext.py -------------------------------------------------------------------------------- /tests/test_backbones/test_shufflenet_v1.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DequanWang/weightnet.pytorch/HEAD/tests/test_backbones/test_shufflenet_v1.py -------------------------------------------------------------------------------- /tests/test_backbones/test_shufflenet_v2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DequanWang/weightnet.pytorch/HEAD/tests/test_backbones/test_shufflenet_v2.py -------------------------------------------------------------------------------- /tests/test_backbones/test_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DequanWang/weightnet.pytorch/HEAD/tests/test_backbones/test_utils.py -------------------------------------------------------------------------------- /tests/test_backbones/test_vgg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DequanWang/weightnet.pytorch/HEAD/tests/test_backbones/test_vgg.py -------------------------------------------------------------------------------- /tests/test_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DequanWang/weightnet.pytorch/HEAD/tests/test_dataset.py -------------------------------------------------------------------------------- /tests/test_eval_hook.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DequanWang/weightnet.pytorch/HEAD/tests/test_eval_hook.py -------------------------------------------------------------------------------- /tests/test_heads.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DequanWang/weightnet.pytorch/HEAD/tests/test_heads.py -------------------------------------------------------------------------------- /tests/test_losses.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DequanWang/weightnet.pytorch/HEAD/tests/test_losses.py -------------------------------------------------------------------------------- /tests/test_metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DequanWang/weightnet.pytorch/HEAD/tests/test_metrics.py -------------------------------------------------------------------------------- /tests/test_pipelines/test_loading.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DequanWang/weightnet.pytorch/HEAD/tests/test_pipelines/test_loading.py -------------------------------------------------------------------------------- /tests/test_pipelines/test_transform.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DequanWang/weightnet.pytorch/HEAD/tests/test_pipelines/test_transform.py -------------------------------------------------------------------------------- /tools/analyze_results.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DequanWang/weightnet.pytorch/HEAD/tools/analyze_results.py -------------------------------------------------------------------------------- /tools/convert_models/mobilenetv2_to_mmcls.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DequanWang/weightnet.pytorch/HEAD/tools/convert_models/mobilenetv2_to_mmcls.py -------------------------------------------------------------------------------- /tools/convert_models/shufflenetv2_to_mmcls.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DequanWang/weightnet.pytorch/HEAD/tools/convert_models/shufflenetv2_to_mmcls.py -------------------------------------------------------------------------------- /tools/convert_models/vgg_to_mmcls.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DequanWang/weightnet.pytorch/HEAD/tools/convert_models/vgg_to_mmcls.py -------------------------------------------------------------------------------- /tools/dist_test.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DequanWang/weightnet.pytorch/HEAD/tools/dist_test.sh -------------------------------------------------------------------------------- /tools/dist_train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DequanWang/weightnet.pytorch/HEAD/tools/dist_train.sh -------------------------------------------------------------------------------- /tools/get_flops.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DequanWang/weightnet.pytorch/HEAD/tools/get_flops.py -------------------------------------------------------------------------------- /tools/onnx2tensorrt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DequanWang/weightnet.pytorch/HEAD/tools/onnx2tensorrt.py -------------------------------------------------------------------------------- /tools/publish_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DequanWang/weightnet.pytorch/HEAD/tools/publish_model.py -------------------------------------------------------------------------------- /tools/pytorch2onnx.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DequanWang/weightnet.pytorch/HEAD/tools/pytorch2onnx.py -------------------------------------------------------------------------------- /tools/slurm_test.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DequanWang/weightnet.pytorch/HEAD/tools/slurm_test.sh -------------------------------------------------------------------------------- /tools/slurm_train.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DequanWang/weightnet.pytorch/HEAD/tools/slurm_train.sh -------------------------------------------------------------------------------- /tools/test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DequanWang/weightnet.pytorch/HEAD/tools/test.py -------------------------------------------------------------------------------- /tools/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/DequanWang/weightnet.pytorch/HEAD/tools/train.py --------------------------------------------------------------------------------