mirrors/mmpretrain
1
0
Fork 0
mirror of https://github.com/open-mmlab/mmpretrain.git synced 2025-06-03 14:59:18 +08:00
Code Issues Packages Projects Releases Wiki Activity
693 Commits 8 Branches 41 Tags
Commit Graph

409 Commits

Author SHA1 Message Date
zzc98
693596bc2f
[Feature] Add Base Retriever and Image2Image Retriever for retrieval tasks. (#1055) 
* feat: add image retriever

* feat: add image retriever

* feat: add image retriever

* feat: add image retriever

* feat: add image retriever

* feat: add image retriever

* feat: add image retriever

* feat: add image retriever

* feat: add image retriever

* update retriever

* fix lint

* add hook unit test

* Use `register_buffer` to save prototype vectors and add a progress bar
during preparing prototype.

* update UTs

* update UTs

* fix typo

* modify the hook

Co-authored-by: Ezra-Yu <18586273+Ezra-Yu@users.noreply.github.com>
Co-authored-by: mzr1996 <mzr1996@163.com>
 2022-11-02 17:43:56 +08:00
Hubert
50aaa711ea
[Docs] Add custom evaluation docs (#1130) 
* [Docs] Add evaluation docs

* minor fix

* Fix docs.

Co-authored-by: mzr1996 <mzr1996@163.com>
 2022-11-01 18:54:06 +08:00
mzr1996
b526f018db [Fix] Fix broken inference api because of the modification in data 
preprocessor.
 2022-10-18 17:50:35 +08:00
Hubert
bcca619066
[Feature] Support MobileViT backbone. (#1068) 
* init

* fix

* add config

* add meta

* add unittest

* fix for comments

* Imporvee docstring and support custom arch.

* Update README

* Update windows CI

Co-authored-by: mzr1996 <mzr1996@163.com>
 2022-10-18 17:05:59 +08:00
Ma Zerun
29f066f7fb
[Improve] Speed up data preprocessor. (#1064) 
* [Improve] Speed up data preprocessor.

* Add ClsDataSample serialization override functions.

* Add unit tests

* Modify configs to fit new mixup args.

* Fix `num_classes` of the ImageNet-21k config.

* Update docs.
 2022-10-17 17:08:18 +08:00
kitecats
06c919efc2
[Fix] Fix for inference_model cannot get classes information in checkpoint. (#1093) 
* Fix for MMCLS1.x not being able to get classes information in checkpoint during inference

Let MMCLS1.x get classes information from checkpoint during inference instead of using imagenet classes initialization

* Update inference.py
 2022-10-14 08:27:01 +08:00
Ma Zerun
31c67ffed4
Bump version to v1.0.0rc2. (#1084) 2022-10-12 16:52:31 +08:00
Ma Zerun
dfe0874102
Update requirements. (#1083) 2022-10-12 10:57:37 +08:00
takuoko
a49c3076e1
[Feature] Support DeiT3. (#1065) 
* deit3

deit3

lint

* add tools and test

* deit3

* deit3

* fix preprocess

* lint

* Update config names and checkpoint paths

* Update convert tools to use mmengine, and fix docstring.

Co-authored-by: mzr1996 <mzr1996@163.com>
 2022-10-10 14:54:20 +08:00
mzr1996
043574cbb2 Merge branch '1.x' into dev-1.x 2022-10-10 11:18:05 +08:00
mzr1996
2153a16dc5 [Docs] Fix some docstrings. 2022-10-10 11:17:48 +08:00
takuoko
a1642e42da
[Enhancement] Get scores from inference api. (#1070) 2022-10-08 15:21:34 +08:00
mzr1996
ae37d7fd27 [Fix] Fix visualization hook. 2022-10-08 11:29:18 +08:00
Ma Zerun
7237a646e8
Bump version to v1.0.0rc1 (#1066) 2022-09-30 17:35:44 +08:00
whcao
8c5d86a388
[Fix] Fix loss_weight in LabelSmoothLoss. (#1058) 2022-09-28 08:18:12 +08:00
Ma Zerun
51b7c754e7 [Fix] Fix the output position of Swin-Transformer. (#947) 
* [Fix] Fix the output position of Swin-Transformer.

* Rename `downsample` argument to `do_downsample`.
 2022-09-22 17:11:52 +08:00
Ma Zerun
034683cfcd
[Feature] Support MViT for MMCLS 1.x (#1023) 
* [Feature] Support MViT for MMCLS 1.x

* Add unit tests

* Update documentation
 2022-09-22 09:23:49 +08:00
Yuan Liu
f7191ca686
[Feature] Add ViT huge architecture. (#1049) 
* [Feature]: Add ViT huge settings

* [Feature]: Add comments
 2022-09-21 13:27:54 +08:00
Songyang Zhang
81f10ccbb7
[Feature] Support EdgeNeXt for dev-1.x. (#1037) 
* [Feature] add EdgeNeXt for dev-1.x

* [Feature] add update

* Imporve EdgeNeXt

* Fix div error in low pytorch version

Co-authored-by: mzr1996 <mzr1996@163.com>
 2022-09-21 13:27:04 +08:00
mzr1996
b35ee5d139 [Fix] Fix wrong default CLASSES settings in init_model. 2022-09-20 17:48:03 +08:00
Ma Zerun
e9e2d48cb2
[Improve] Update benchmark scripts (#1028) 
* Update train benchmark scripts

* Add `--cfg-options` for dev scripts and enhance `--range`.

* Fix bug of regex expression.

* Fix minor bugs

* Update ShuffleNet configs

* Update rsb-a1 configs and label smooth loss mode.

* Update inference dev scripts

* From `mmengine` instead of `mmcv` import fileio.

* Fix lint

* Update pre-commit hook

* Use `use_sigmoid` option instead of "bce" mode in label smooth loss.
 2022-09-20 15:50:21 +08:00
Hubert
e4e8047563
[Feature] Support Swin Transformer V2 for MMCLS 1.x. (#1029) 
* [Feature] support swin v2 in mmcls 1.x

* fix configs

* fix clamp max

* Fix meshgrid

* Imporve swin v2

Co-authored-by: mzr1996 <mzr1996@163.com>
 2022-09-20 15:45:27 +08:00
Ezra-Yu
aaf127c5e1
[Feature] Add efficientformer Backbone for MMCls 1.x. (#1031) 
* rebase

* update filename

* update URL

* update UT

* fix lint

* update head

* add efficientformer

* update filename

* update UT

* fix lint

* update configs

* rebase

* fix unit tests

* Fix comments and docs.

Co-authored-by: mzr1996 <mzr1996@163.com>
 2022-09-20 14:56:45 +08:00
Ezra-Yu
f1d2f50c21
[Feature] Add MobileOne Backbone For MMCls 1.x. (#1030) 
* add mobileOne

* add train cfg

* update cfgs

* update URL

* update configs

* update inceptionv3 metafile

* add configs

* fix lint

* update checkpoint urls

* Update configs

* Update README

Co-authored-by: mzr1996 <mzr1996@163.com>
 2022-09-16 11:55:10 +08:00
Ma Zerun
61e9d890a6
[CI] Update circle-ci and github workflow. (#1018) 
* Add deploy workflow.

* [CI] Update circle-ci and github workflow.

* Fix windows CI

* Update unit tests to save memory
 2022-09-13 11:46:15 +08:00
Ma Zerun
85b1eae7f1
Bump to v1.0.0rc0 (#1007) 
* Update docs.

* Update requirements.

* Update config readme and docstring.

* Update CONTRIBUTING.md

* Update README

* Update requirements/mminstall.txt

Co-authored-by: Yifei Yang <2744335995@qq.com>

* Update MMEngine docs link and add to readthedocs requirement.

Co-authored-by: Yifei Yang <2744335995@qq.com>
 2022-08-31 23:57:51 +08:00
Ma Zerun
0b61b76325
[Docs] Add Migration documentation. (#997) 
* [Docs] Add Migration documentation.

* Fix the train_cfg argument

* [WIP] Improve according to comments

* Imporve according to comments.
 2022-08-31 13:31:48 +08:00
Ezra-Yu
e34588a315
[Docs] Refactor API docs. (#958) 
* rebase

* update process

* add TODOs in moved classes

* update APIs

* modify the preprocess_cfg to data_preprocessor

* Fix api docs.

Co-authored-by: mzr1996 <mzr1996@163.com>
 2022-08-29 15:35:10 +08:00
Ezra-Yu
0e3e8e91cc
[Refactor] Refactor Config Doc. (#987) 
* refactor config doc

* refine

* Fix the config docs.

* update CN config

* Fix some grammar erros.

* Fix Chinese docs.

Co-authored-by: mzr1996 <mzr1996@163.com>
 2022-08-29 11:10:05 +08:00
Ma Zerun
2b88df4484
[Refactor] Refactor the data flow. (#989) 
* [Refactor] Refactor the data flow.

* Add comments about data preprocessor.

* Fix after mmengine folder structure refactoring.
 2022-08-26 10:40:43 +08:00
Ma Zerun
b7d0d521eb
[Refactor] Use mmengine instead of mmcv and refactor some transforms. (#986) 
* [Refactor] Refactor the rest data transforms and fix some docstring.

* Use `mmengine` instead of `mmcv`

* Refactor KFold dataset tools

* Fix docstring according to comments
 2022-08-24 15:59:02 +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
Ma Zerun
20f9ace5c5
[CI] Add Circle CI for mmcls 1.x. (#970) 
* [CI] Add Circle CI for mmcls 1.x

* Fix circle-ci config

* Use `torch.cat` instead of `torch.hstack` to compat with PyTorch 1.6

* Compat the behavior of LongTensor in PyTorch 1.6

* Avoid random behavior in the unit test of Lighting

* Fix cuda ci

* Remove github workflow temporarily.
 2022-08-22 15:02:08 +08:00
Yuan Liu
e4252d6848
[Feature]: Support BEiT Transformer layer. (#919) 
* [Feature]: Add BEiT-style transformer encoder layer

* [Feature]: Add average token

* [Fix]: Fix lint

* [Fix]: Refactor CAE config

* [Fix]: Change cv2 backend to pillow backend

* [Fix]: Fix MAE and CAE reshape bug

* [Feature]: Add freeze vit layers

* [Feature]: Add mc

* [Fix]: Fix lint

* [Fix]: Fix dataset bug

* [Fix]: Delete cae selfsup config

* [Fix]: docstring

* [Refactor]: Add init_values to layer_scalue_init_value

* [Fix]: Refine the docstring of avg_token

* [Fix]: Call super init weight in beit attention

* [Fix]: remove mc

* [Fix]: Fix docstring

* [Fix]: Fix docstring

* [Fix]: Fix lint

* [Fix]: Fix init_value bug and change the logic of outputting cls token

* [Fix]: Fix docstring
 2022-08-17 00:07:06 +08:00
Ma Zerun
b8b31e9343
[Improve] Multiple fixes. (#957) 
* Rename some files in unit tests.

* Rename `pipelines` folder to `transforms`.

* Fix version requirement range.

* Remove `init_random_seed` and `set_random_seed`.

* Fix unit tests

* Rename `convert_models` to `model_converters`.
 2022-08-09 15:58:08 +08:00
Ezra-Yu
624d22b441
[Refactor] Refactor Precise BN Hook. (#913) 2022-08-03 15:24:19 +08:00
Ma Zerun
43e60ad5a6
[Improve] Rename mmcls.data to mmcls.structures. (#941) 2022-07-29 14:18:13 +08:00
Ma Zerun
17b24a8230
[Improve] Update apis and packages. (#921) 
* Remove mmcls/core folder

* update dependency version.

* Update `apis/inference.py` and demo.
 2022-07-22 10:59:08 +08:00
Ezra-Yu
f37dc44a25 refactor UTs 2022-07-18 11:11:13 +08:00
Ezra-Yu
35cba87f82 refactor version and utils 2022-07-18 11:11:13 +08:00
Ezra-Yu
9118b97239 refactor evaluation 2022-07-18 11:11:13 +08:00
Ezra-Yu
6ac2f305a6 fix examples in doc 2022-07-18 11:11:13 +08:00
Ezra-Yu
317216efc6 fix visualizer registry bug 2022-07-18 11:11:13 +08:00
Ezra-Yu
1c82863449 fix lint 2022-07-18 11:11:13 +08:00
Ezra-Yu
6ed3860ea6 refactor UTs 2022-07-18 11:11:13 +08:00
Ezra-Yu
0f30c392a3 from core to engine 2022-07-18 11:11:13 +08:00
mzr1996
24bcf069f8 [Refactor] Refactor dev scripts 2022-07-18 11:11:13 +08:00
yingfhu
c992e24617 [Refactor] refactor ClassNumCheckHook 2022-07-18 11:11:13 +08:00
Ezra-Yu
e9342d9e4c Add multi_label heads 2022-07-18 11:11:13 +08:00
yingfhu
4fcd7ee072 Add EfficientNetRandomCrop and EfficientNetCenterCrop 2022-07-18 11:11:13 +08:00