mmclassification/mmcls/models/backbones
Ma Zerun a8f4f82b8e
[Enhance] Improve downstream repositories compatibility (#421)
* Defaults to return tuple in all backbones.

* Compat downstream of swin transformer.

* Support tuple input for multi label head and stacked head.

* Fix backbone unit tests for tuple output.

* Add downstream inference unit tests for mmdet.

* Update gitignore

* Add unit tests for `return_tuple` option

* Add unit tests for head input tuple.

* Add warning in `simple_test`

* Add TIMMBackbone return tuple.

* Modify timm backbone unit test.
2021-09-08 10:38:57 +08:00
..
__init__.py [Feature] timm backbones wrapper (#427) 2021-09-06 11:05:31 +08:00
alexnet.py [Enhance] Improve downstream repositories compatibility (#421) 2021-09-08 10:38:57 +08:00
base_backbone.py [Docs] Add Copyright information. (#413) 2021-08-17 19:52:42 +08:00
lenet.py [Enhance] Improve downstream repositories compatibility (#421) 2021-09-08 10:38:57 +08:00
mobilenet_v2.py [Enhance] Improve downstream repositories compatibility (#421) 2021-09-08 10:38:57 +08:00
mobilenet_v3.py [Enhance] Improve downstream repositories compatibility (#421) 2021-09-08 10:38:57 +08:00
regnet.py [Enhance] Improve downstream repositories compatibility (#421) 2021-09-08 10:38:57 +08:00
resnest.py [Docs] Add Copyright information. (#413) 2021-08-17 19:52:42 +08:00
resnet.py [Enhance] Improve downstream repositories compatibility (#421) 2021-09-08 10:38:57 +08:00
resnet_cifar.py [Enhance] Improve downstream repositories compatibility (#421) 2021-09-08 10:38:57 +08:00
resnext.py [Docs] Add Copyright information. (#413) 2021-08-17 19:52:42 +08:00
seresnet.py [Docs] Add Copyright information. (#413) 2021-08-17 19:52:42 +08:00
seresnext.py [Docs] Add Copyright information. (#413) 2021-08-17 19:52:42 +08:00
shufflenet_v1.py [Enhance] Improve downstream repositories compatibility (#421) 2021-09-08 10:38:57 +08:00
shufflenet_v2.py [Enhance] Improve downstream repositories compatibility (#421) 2021-09-08 10:38:57 +08:00
swin_transformer.py [Enhance] Improve downstream repositories compatibility (#421) 2021-09-08 10:38:57 +08:00
timm_backbone.py [Enhance] Improve downstream repositories compatibility (#421) 2021-09-08 10:38:57 +08:00
tnt.py [Docs] Add Copyright information. (#413) 2021-08-17 19:52:42 +08:00
vgg.py [Enhance] Improve downstream repositories compatibility (#421) 2021-09-08 10:38:57 +08:00
vision_transformer.py [Docs] Add Copyright information. (#413) 2021-08-17 19:52:42 +08:00