mirrors
/
mmpretrain
mirror of https://github.com/open-mmlab/mmpretrain.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity
657 Commits
8 Branches
41 Tags
32 MiB
Commit Graph

657 Commits (e4e8047563f93cfb1695638107f0b47f5afb8d7b)
 

Author SHA1 Message Date
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 9999da646c
[Imporve] Update tools to enable `pin_memory` and `persistent_workers` by default. (#1024) 2022-09-14 11:57:32 +08:00
Ma Zerun 96a1a34415
[Docs] Auto generate model summary table. (#1010) 
* Fix wrong metafile

* Auto generate model summary table.

* Fix all TODO link
 2022-09-13 15:06:17 +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 c95ab99289
[Docs] Add training and test docs. (#994) 
* [Docs] Add train and test docs.

* Improve according to comments.
 2022-08-31 14:22:30 +08:00
Ezra-Yu 18ce94bd5f
[Docs] Add Inference Doc (#999) 
* add EN inference doc

* add CN inference doc

* refine

* refine

* refine

* update inference dics

* refine CN docs
 2022-08-31 14:21:56 +08:00
Songyang Zhang d0de93746f
[Docs] Update the contributing tutorial. (#1002) 
* [Doc] update contributing

* update

* Update CONTRIBUTING.md

Co-authored-by: Ma Zerun <mzr1996@163.com>
 2022-08-31 14:11:38 +08:00
Songyang Zhang 80ad9432f3
[Docs] Add projects page in documentations. (#1003) 
* [Doc] update projects page

* update

* update

* update
 2022-08-31 14:07:07 +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
Hubert 385cf2d98c
[Docs] Refactor new modules tutorial. (#998) 
* [Docs] refactor new modules tutorial

* Imporve the modules customization documentations.

* [Fix] fix for comments

* minor add

* Fix grammar of English tutorial.

Co-authored-by: mzr1996 <mzr1996@163.com>
 2022-08-31 11:21:44 +08:00
Songyang Zhang f52aab488b
[Doc] Update analysis tools tutorial. (#1001) 
* [Doc] update analysis doc

* Rename the tutorial title.

Co-authored-by: mzr1996 <mzr1996@163.com>
 2022-08-31 11:07:25 +08:00
Songyang Zhang 4367d05dfc
[Enhance] Use `fvcore` to calculate FLOPS. (#1000) 
* [Feature] Use fvcore for flops count

* update requirements

* update
 2022-08-31 11:04:42 +08:00
Ezra-Yu ccece7e6dc
[Docs] Refactor dataset tutorial (#916) 
* refactor dataset tutorials

* split into user_guide and advance_guide

* refine

* Fix dataset preparasion tutorial.

* refine CN docs

* update docs API doc link

* refine new a dataset

* refine new a dataset

* refine new a dataset

Co-authored-by: mzr1996 <mzr1996@163.com>
 2022-08-30 18:45:58 +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
Hubert 344fa64400
[Docs] Refactor fine-tune tutorial. (#991) 
* [Docs] refactor fine-tune tutorial

* fix comment
 2022-08-29 10:18:38 +08:00
Ma Zerun 12d8c78222
[Docs] Refactor get started docs and fix the demo and docker file. (#990) 
* [Docs] Refactor get started docs and fix the demo and docker file.

* Improve docs according to comments
 2022-08-26 18:03:18 +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 b4e39d51d6
[Refactor] Refactor the docs structure. (#988) 
* [Refactor] Refactor the rest data transforms and fix some docstring.

* Use `mmengine` instead of `mmcv`

* Refactor KFold dataset tools

* Fix docstring according to comments

* [Refactor] Refactor the docs structure.

* Rename `getting_started.md` to `get_started.md`

* Rename `transforms` to `data_process`
 2022-08-24 16:00:36 +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
Ma Zerun 7f0502867b
[Improve] Add `--amp`, `--resume` and `--auto-scale-lr` options in `tools/train.py`. (#914) 2022-07-21 17:34:04 +08:00
mzr1996 0c7a04b1c7 Fix lint 2022-07-18 14:14:31 +08:00
mzr1996 735a3ee11f Update auto_scale_lr fields 2022-07-18 11:11:13 +08:00
yingfhu a667b488ae minor fix 2022-07-18 11:11:13 +08:00
yingfhu ce81a07059 [Refactor] add auto_scale_lr 2022-07-18 11:11:13 +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
mzr1996 12c982f939 [Refactor] Refactor `RepeatAugSampler`. 2022-07-18 11:11:13 +08:00
yingfhu e8d69cf2ff [Fix] adjust default dataset path to empty string 2022-07-18 11:11:13 +08:00
mzr1996 c0f3ba68a8 Add MAE version ViT-base training results and fix some errors in configs. 2022-07-18 11:11:13 +08:00