mmpretrain/mmcls/models/backbones
Ma Zerun 65410b05ad
Fix Mobilenetv3 structure and add pretrained model (#291)
* Refactor Mobilenetv3 structure and add ConvClsHead.

* Change model's name from 'MobileNetv3' to 'MobileNetV3'

* Modify configs for MobileNetV3 on CIFAR10.

And add MobileNetV3 configs for imagenet

* Fix activate setting bugs in MobileNetV3.

And remove bias in SELayer.

* Modify unittest

* Remove useless config and file.

* Fix mobilenetv3-large arch setting

* Add dropout option in ConvClsHead

* Fix MobilenetV3 structure according to torchvision version.

1. Remove with_expand_conv option in InvertedResidual, it should be decided by channels.

2. Revert activation function, should before SE layer.

* Format code.

* Rename MobilenetV3 arch "big" to "large".

* Add mobilenetv3_small torchvision training recipe

* Modify default `out_indices` of MobilenetV3, now it will change
according to `arch` if not specified.

* Add MobilenetV3 large config.

* Add mobilenetv3 README

* Modify InvertedResidual unit test.

* Refactor ConvClsHead to StackedLinearClsHead, and add unit tests.

* Add unit test for `simple_test` of `StackedLinearClsHead`.

* Fix typo

Co-authored-by: Yidi Shao <ydshao@smail.nju.edu.cn>
2021-06-27 23:19:36 +08:00
..
__init__.py Fix Mobilenetv3 structure and add pretrained model (#291) 2021-06-27 23:19:36 +08:00
alexnet.py revise docstr of alexnet to correct input_size (#26) 2020-08-12 16:14:19 +08:00
base_backbone.py [Feature]Delete comments (#298) 2021-06-12 21:45:22 +08:00
lenet.py Add AlexNet and LeNet5 2020-07-12 00:06:54 +08:00
mobilenet_v2.py [WIP] Refactoring weights initialization (#270) 2021-06-10 10:54:34 +08:00
mobilenet_v3.py Fix Mobilenetv3 structure and add pretrained model (#291) 2021-06-27 23:19:36 +08:00
regnet.py [WIP] Refactoring weights initialization (#270) 2021-06-10 10:54:34 +08:00
resnest.py [Fix] Fix optional issues in docstring (#138) 2021-01-14 11:09:08 +08:00
resnet.py [Feature]Delete comments (#298) 2021-06-12 21:45:22 +08:00
resnet_cifar.py Add ResNet_CIFAR 2020-07-12 00:06:56 +08:00
resnext.py [Fix] Fix optional issues in docstring (#138) 2021-01-14 11:09:08 +08:00
seresnet.py Refactoring for ResNet family 2020-06-25 11:57:50 +08:00
seresnext.py [Fix] Fix optional issues in docstring (#138) 2021-01-14 11:09:08 +08:00
shufflenet_v1.py [Feature]Delete comments (#298) 2021-06-12 21:45:22 +08:00
shufflenet_v2.py [Feature]Delete comments (#298) 2021-06-12 21:45:22 +08:00
vgg.py [Feature]Delete comments (#298) 2021-06-12 21:45:22 +08:00
vision_transformer.py [Feature]Delete comments (#298) 2021-06-12 21:45:22 +08:00