Commit Graph

6 Commits (dbf3df21a31ee5404184b3f1ba0588b6b6aaa8f4)

Author SHA1 Message Date
Ma Zerun 414ba80274
[Refactor] Refactor APIs, add `ImageRetrievalInferencer` and `FeatureExtractor`. (#1380)
* [Refactor] Refactor APIs, add `ImageRetrievalInferencer` and `FeatureExtractor'.

* Update image retrieval

* Update FeatureExtractor

* Fix UT
2023-02-28 16:31:42 +08:00
Ma Zerun 36bea13fca
[Refactor] Refactor ClsDatasample to a union DataSample. (#1371)
* [Refactor] Refactor ClsDatasample to a union DataSample.

* Add  method

* Fix docstring

* Update docstring.
2023-02-23 10:07:53 +08:00
mzr1996 0979e78573 Rename the package name to `mmpretrain`. 2023-02-17 15:20:55 +08:00
Ma Zerun a3f2effb17
[Feature] Add `ImageClassificationInferencer`. (#1261)
* [Feature] Add ImageClassificationInferencer.

* Update inferencer implementation and add unit tests.

* Update documentations.

* Update pre-commit hook

* Update docs
2023-02-08 14:30:12 +08:00
Ma Zerun aa53f7790c
[Tool] Add metafile fill and validation tools. (#1297)
* [Tool] Add metafile fill and validation tools.

* Fix pre-commit hooks and update fill tool.

* Update metafile

* Update fill tool

* Remove support no config.

* Add 3rdparty check.
2023-01-10 10:06:56 +08:00
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