mirrors/mmclassification
1
0
Fork 0
mirror of https://github.com/open-mmlab/mmclassification.git synced 2025-06-03 21:53:55 +08:00
Code Issues Projects Releases Wiki Activity
763 Commits 8 Branches 41 Tags
Commit Graph

8 Commits

Author SHA1 Message Date
Ma Zerun
c127c474b9
[Feature] Support getting model from the name defined in the model-index file. (#1236) 
* [Feature] Support getting model from the name defined in the model-index file.

* Add unit tests.

* Prevent import `timm` if the `TIMMBackbone` is not used.

* Fix Windows CI.

* Move `init_model` to `mmcls.apis.hub`, and support pass nn.Module to all
model components.

* Fix requirements

* Rename `hub.py` to `model.py` and add unit tests.
 2022-12-06 17:00:22 +08:00
Yixiao Fang
5665b8349a
[Refactor] Refactor import dependency. (#977) 
Co-authored-by: mzr1996 <mzr1996@163.com>
 2022-08-22 15:17:33 +08:00
mzr1996
24bcf069f8 [Refactor] Refactor dev scripts 2022-07-18 11:11:13 +08:00
Ezra-Yu
088d5b5add [Refactor] Inherit all registries from MMEngine and use inherited registries. 2022-07-18 10:53:27 +08:00
Yosuke Shinya
16864c7495
[Feature] Support features_only in TIMMBackbone (#668) 
* Support features_only in TIMMBackbone

based on https://github.com/open-mmlab/mmsegmentation/pull/998

* update test for mmdet

* fix unit test for build_without_timm

* Update docstring

Co-authored-by: mzr1996 <mzr1996@163.com>
 2022-01-25 11:22:23 +08:00
takuoko
9f31fc5390
[Refactor] Use reset_classifier to remove head of timm backbones. (#534) 
* refactor timm backbone

* fix comment
 2021-11-16 16:24:55 +08:00
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
Miras Amir
5cfaed6807
[Feature] timm backbones wrapper (#427) 
* Add wrapper to use backbones from timm

* Add tests

* Remove timm from optional deps and modify GitHub workflow.

Co-authored-by: mzr1996 <mzr1996@163.com>
 2021-09-06 11:05:31 +08:00