mirrors/mmselfsup
1
0
Fork 0
mirror of https://github.com/open-mmlab/mmselfsup.git synced 2025-06-03 14:59:38 +08:00
Code Issues Packages Projects Releases Wiki Activity
417 Commits 5 Branches 19 Tags
Commit Graph

417 Commits

This Branch
This Branch
All Branches
Author SHA1 Message Date
YuanLiuuuuuu
a10a6bd64f [Refactor]: Refactor layer wise decay learning rate optimizer constructor 2022-07-18 11:06:44 +08:00
YuanLiuuuuuu
2f2813ecd4 [Refactor]: Refactor ODC 2022-07-18 11:06:44 +08:00
YuanLiuuuuuu
0baac605d1 [Refactor]: Refactor NPID 2022-07-18 11:06:44 +08:00
fangyixiao.vendor
09c721a33e [Refactor] refactor sampler and clean redundant codes 2022-07-18 11:06:44 +08:00
YuanLiuuuuuu
0c88e8dceb [Refactor]: Refactor loss 2022-07-18 11:06:44 +08:00
fangyixiao18
1f69aa84f7 add 'set_reweight' to hook 2022-07-18 11:06:44 +08:00
renqin
99de0c1aeb [Refactor]: refactor RelativeLoc model and RandomPatchWithLabels transform 2022-07-18 11:06:44 +08:00
renqin
e39e3bbac0 [Refactor]: refactor RotationPred model and RotationWithLabels transform 2022-07-18 11:06:44 +08:00
fangyixiao.vendor
5dc3696df7 [Refactor] refactor DeepCluster and ODC hook and related files 2022-07-18 11:06:44 +08:00
YuanLiuuuuuu
d1b7466d9b [Refactor]: Refactor deepcluster 2022-07-18 11:06:44 +08:00
fangyixiao.vendor
063288240a [Refactor] refactor hooks 2022-07-18 11:06:44 +08:00
renqin
9565da23a4 [Feature]: Add RandomRotation transform 2022-07-18 11:06:44 +08:00
fangyixiao.vendor
2ed7e90052 [Feature] add RandomCrop 2022-07-18 11:06:44 +08:00
YuanLiuuuuuu
f19f8f4c31 [Refactor]: Delete stepfixcosine and create a sample for new lr scheduler 2022-07-18 11:06:44 +08:00
fangyixiao.vendor
953f790328 [Refactor] refactor dataset base config 2022-07-18 11:06:44 +08:00
YuanLiuuuuuu
25a5f71c7e [Refactor]: Logging 2022-07-18 11:06:44 +08:00
renqin
9ad2f6e6d5 [Refactor]: refactor the interface of byol, densecl, swav, barlowtwins and simclr 2022-07-18 11:06:44 +08:00
fangyixiao18
c119c4677c add RandomResizedCrop 2022-07-18 11:06:44 +08:00
liuyuan1.vendor
be1dd2f5c2 [Refactor]: Refactor CAE 2022-07-18 11:06:44 +08:00
liuyuan1.vendor
e687aff595 [Refactor]: Change the interface of SimMIM 2022-07-18 11:06:44 +08:00
fangyixiao.vendor
e87be11a98 [Refactor] refactor moco and mocov3 2022-07-18 11:06:44 +08:00
fangyixiao.vendor
dfbe3f6235 [Feature] add ColorJitter 2022-07-18 11:06:44 +08:00
fangyixiao.vendor
11c3a6a3af [Refactor] refactor SimSiam 2022-07-18 11:06:44 +08:00
YuanLiuuuuuu
c8c0526582 [Fix]: Fix UT and use LabelData 2022-07-18 11:06:44 +08:00
liuyuan1.vendor
efb9255d7e [Fix]: Support numpy array in random crop resize with two pic 2022-07-18 11:06:44 +08:00
fangyixiao.vendor
4dc2ff1b79 [Refactor] refactor train/test interface 2022-07-18 11:06:44 +08:00
liuyuan1.vendor
2063a54c2a [Refactor]: Refactor the interface of MAE 2022-07-18 11:06:44 +08:00
renqin
3426b49b39 [Refactor]: refactor RandomRotation and RandomPatch 2022-07-18 11:06:44 +08:00
fangyixiao.vendor
d9c7bd6a7b [Fix] revise __repr__ format of some transforms 2022-07-18 11:06:44 +08:00
liuyuan1.vendor
80916a0966 [Feature]: Add base model 2022-07-18 11:06:44 +08:00
fangyixiao.vendor
472a0ab548 [Refactor] revise default_rutime.py 2022-07-18 11:06:44 +08:00
fangyixiao.vendor
fa9d148ee8 [Refactor] add MultiView transform wrapper 2022-07-18 11:06:44 +08:00
liuyuan1.vendor
4791407fd7 [Feature]: Add pack selfsup input transform 2022-07-18 11:06:44 +08:00
fangyixiao.vendor
5a8b6b29e6 [Refactor] refactor lighting, gaussian_blur and solarize 2022-07-18 11:06:44 +08:00
liuyuan1.vendor
4501b37c86 [Refactor]: Refacor BEiT, SimMIM mask generator and random crop resize with two pic 2022-07-18 11:06:44 +08:00
liuyuan1.vendor
2e0772ad33 [Feature]: Add Registry 2022-07-18 11:06:44 +08:00
fangyixiao18
b414011530 [Refactor] clean ut 2022-07-18 11:06:44 +08:00
liuyuan1.vendor
40b58076f7 [Feature]: Add self data sample 2022-07-18 11:06:44 +08:00
fangyixiao.vendor
eb417cf112 [CI] add gitlab ci 2022-07-18 11:06:44 +08:00
fangyixiao18
751c817b1f [UT] clean unit test 2022-07-18 11:06:44 +08:00
Yuan Liu
381fd9ad38
[Fix]: Add attrs in requirements (#344) 2022-07-15 13:06:09 +08:00
Yixiao Fang
a7dbc55eb0
Clean pillow version check (#333) 2022-06-22 13:57:22 +08:00
Yixiao Fang
18543dbf6e
[Enhance] Test commands of mim in CI (#332) 
* Test commands of mim in CI

* fix lint
 2022-06-22 13:56:46 +08:00
Yuan Liu
86290150d8
[Fix]: Fix the link of switch language (#327) 2022-06-15 16:15:17 +08:00
Yixiao Fang
69033acfe7
Add mim to extras_require in setup.py (#326) 2022-06-15 16:14:19 +08:00
Yixiao Fang
880ddef6cf
[Docs] Update mdformat settings (#323) 
* update mdformat setting

* update mdformat

* remove pip upgrade
 2022-06-08 14:27:47 +08:00
Yixiao Fang
bbdf670d00
Bump version to v0.9.1 (#322) 
* [Fix]: Set qkv bias to False for cae and True for mae (#303)

* [Fix]: Add mmcls transformer layer choice

* [Fix]: Fix transformer encoder layer bug

* [Fix]: Change UT of cae

* [Feature]: Change the file name of cosine annealing hook (#304)

* [Feature]: Change cosine annealing hook file name

* [Feature]: Add UT for cosine annealing hook

* [Fix]: Fix lint

* read tutorials and fix typo (#308)

* [Fix] fix config errors in MAE (#307)

* update readthedocs algorithm readme (#310)

* [Docs] Replace markdownlint with mdformat (#311)

* Replace markdownlint with mdformat to avoid installing ruby

* fix typo

* add 'ba' to codespell ignore-words-list

* Configure Myst-parser to parse anchor tag (#309)

* [Docs] rewrite install.md (#317)

* rewrite the install.md

* add faq.md

* fix lint

* add FAQ to README

* add Chinese version

* fix typo

* fix format

* remove modification

* fix format

* [Docs] refine README.md file (#318)

* refine README.md file

* fix lint

* format language button

* rename getting_started.md

* revise index.rst

* add model_zoo.md to index.rst

* fix lint

* refine readme

Co-authored-by: Jiahao Xie <52497952+Jiahao000@users.noreply.github.com>

* [Enhance] update byol models and results (#319)

* Update version information (#321)

Co-authored-by: Yuan Liu <30762564+YuanLiuuuuuu@users.noreply.github.com>
Co-authored-by: Yi Lu <21515006@zju.edu.cn>
Co-authored-by: RenQin <45731309+soonera@users.noreply.github.com>
Co-authored-by: Jiahao Xie <52497952+Jiahao000@users.noreply.github.com>
v0.9.1 		2022-06-01 09:59:05 +08:00
Yuan Liu
399b5a0d6e
Bump version to v0.9.0 (#299) 
* [Feature]: MAE pre-training with fp16 (#271)

* [Feature]: MAE pre-training with fp16

* [Fix]: Fix lint

* [Fix]: Fix SimMIM config link, and add SimMIM to model_zoo (#272)

* [Fix]: Fix link error

* [Fix]: Add SimMIM to model zoo

* [Fix]: Fix lint

* [Fix] fix 'no init_cfg' error for pre-trained model backbones (#256)

* [UT] add unit test for apis (#276)

* [UT] add unit test for apis

* ignore pytest log

* [Feature] Add extra dataloader settings in configs. (#264)

* [Feature] support to set validation samples per gpu independently

* set default to be cfg.data.samples_per_gpu

* modify the tools/test.py

* using 'train_dataloader', 'val_dataloader', 'test_dataloader' for specific settings

* test 'evaluation' branch

* [Fix]: Change imgs_per_gpu to samples_per_gpu MAE (#278)

* [Feature]: Add SimMIM 192 pt 224 ft (#280)

* [Feature]: Add SimMIM 192 pt 224 ft

* [Feature]: Add simmim 192 pt 224 ft to readme

* [Fix] fix key error bug when registering custom hooks (#273)

* [UT] remove pytorch1.5 test (#288)

* [Benchmark] rename linear probing config file names (#281)

* [Benchmark] rename linear probing config file names

* update config links

* Avoid GPU memory leak with prefetch dataloader (#277)

* [Feature] barlowtwins (#207)

* [Fix]: Fix mmcls upgrade bug (#235)

* [Feature]: Add multi machine dist_train (#232)

* [Feature]: Add multi machine dist_train

* [Fix]: Change bash to sh

* [Fix]: Fix missing sh suffix

* [Refactor]: Change bash to sh

* [Refactor] Add unit test (#234)

* [Refactor] add unit test

* update workflow

* update

* [Fix] fix lint

* update test

* refactor moco and densecl unit test

* fix lint

* add unit test

* update unit test

* remove modification

* [Feature]: Add MAE metafile (#238)

* [Feature]: Add MAE metafile

* [Fix]: Fix lint

* [Fix]: Change LARS to AdamW in the metafile of MAE

* Add barlowtwins

* Add unit test for barlowtwins

* Adjust training params

* add decorator to pass CI

* adjust params

* Add barlowtwins

* Add unit test for barlowtwins

* Adjust training params

* add decorator to pass CI

* adjust params

* add barlowtwins configs

* revise LatentCrossCorrelationHead

* modify ut to save memory

* add metafile

* add barlowtwins results to model zoo

* add barlow twins to homepage

* fix batch size bug

* add algorithm readme

* add type hints

* reorganize the model zoo

* remove one config

* recover the config

* add missing docstring

* revise barlowtwins

* reorganize coco and voc benchmark

* add barlowtwins to index.rst

* revise docstring

Co-authored-by: Yuan Liu <30762564+YuanLiuuuuuu@users.noreply.github.com>
Co-authored-by: Yixiao Fang <36138628+fangyixiao18@users.noreply.github.com>
Co-authored-by: fangyixiao18 <fangyx18@hotmail.com>

* [Fix] fix --local-rank (#290)

* [UT] reduce memory usage while runing unit test (#291)

* [Feature]: CAE Supported (#284)

* [Feature]: Add mc

* [Feature]: Add dataset of CAE

* [Feature]: Init version of CAE

* [Feature]: Add mc

* [Fix]: Change beta to (0.9, 0.999)

* [Fix]: New feature

* [Fix]: Decouple the qkv bias

* [Feature]: Decouple qkv bias in MultiheadAttention

* [Feature]: New mask generator

* [Fix]: Fix TransformEncoderLayer bug

* [Feature]: Add MAE CAE linear prob

* [Fix]: Fix config

* [Fix]: Delete redundant mc

* [Fix]: Add init value in mim cls vit

* [Fix]: Fix cae ft config

* [Fix]: Delete repeated init_values

* [Fix]: Change bs from 64 to 128 in CAE ft

* [Fix]: Add mc in cae pt

* [Fix]: Fix momemtum update bug

* [Fix]: Add no weight_decay for gamma

* [Feature]: Add mc for cae pt

* [Fix]: Delete mc

* [Fix]: Delete redundant files

* [Fix]: Fix lint

* [Feature]: Add docstring to algo, backbone, neck and head

* [Fix]: Fix lint

* [Fix]: network

* [Feature]: Add docstrings for network blocks

* [Feature]: Add docstring to ToTensor

* [Feature]: Add docstring to transoform

* [Fix]: Add type hint to BEiTMaskGenerator

* [Fix]: Fix lint

* [Fix]: Add copyright to dalle_e

* [Fix]: Fix BlockwiseMaskGenerator

* [Feature]: Add UT for CAE

* [Fix]: Fix dalle state_dict path not existed bug

* [Fix]: Delete file_client_args related code

* [Fix]: Remove redundant code

* [Refactor]: Add fp16 to the name of cae pre-train config

* [Refactor]: Use FFN from mmcv

* [Refactor]: Change network_blocks to trasformer_blocks

* [Fix]: Fix mask generator name bug

* [Fix]: cae pre-train config bug

* [Fix]: Fix docstring grammar

* [Fix]: Fix mc related code

* [Fix]: Add object parent to transform

* [Fix]: Delete unnecessary modification

* [Fix]: Change blockwisemask generator to simmim mask generator

* [Refactor]: Change cae mae pretrain vit to cae mae vit

* [Refactor]: Change lamb to lambd

* [Fix]: Remove blank line

* [Fix]: Fix lint

* [Fix]: Fix UT

* [Fix]: Delete modification to swin

* [Fix]: Fix lint

* [Feature]: Add README and metafile

* [Feature]: Update index.rst

* [Fix]: Update model_zoo

* [Fix]: Change MAE to CAE in algorithm

* [Fix]: Change SimMIMMaskGenerator to CAEMaskGenerator

* [Fix]: Fix model zoo

* [Fix]: Change to dalle_encoder

* [Feature]: Add download link for dalle

* [Fix]: Fix lint

* [Fix]: Fix UT

* [Fix]: Update metafile

* [Fix]: Change b to base

* [Feature]: Add dalle download link in warning

* [Fix] add arxiv link in readme

Co-authored-by: Jiahao Xie <52497952+Jiahao000@users.noreply.github.com>

* [Enhance] update SimCLR models and results (#295)

* [Enhance] update simclr models and results

* [Fix] revise comments to indicate settings

* Update version (#296)

* [Feature]: Update to 0.9.0

* [Feature]: Add version constrain for mmcls

* [Fix]: Fix bug

* [Fix]: Fix version bug

* [Feature]: Update version in install.md

* update changelog

* update readme

* [Fix] fix uppercase

* [Fix] fix uppercase

* [Fix] fix uppercase

* update version dependency

* add cae to readme

Co-authored-by: fangyixiao18 <fangyx18@hotmail.com>
Co-authored-by: Jiahao Xie <52497952+Jiahao000@users.noreply.github.com>

Co-authored-by: Yixiao Fang <36138628+fangyixiao18@users.noreply.github.com>
Co-authored-by: Ming Li <73068772+mitming@users.noreply.github.com>
Co-authored-by: xcnick <xcnick0412@gmail.com>
Co-authored-by: fangyixiao18 <fangyx18@hotmail.com>
Co-authored-by: Jiahao Xie <52497952+Jiahao000@users.noreply.github.com>
v0.9.0 		2022-04-29 20:01:30 +08:00
Yixiao Fang
df907e5ce0
Bump version to v0.8.0 (#269) 
* [Fix]: Fix mmcls upgrade bug (#235)

* [Feature]: Add multi machine dist_train (#232)

* [Feature]: Add multi machine dist_train

* [Fix]: Change bash to sh

* [Fix]: Fix missing sh suffix

* [Refactor]: Change bash to sh

* [Refactor] Add unit test (#234)

* [Refactor] add unit test

* update workflow

* update

* [Fix] fix lint

* update test

* refactor moco and densecl unit test

* fix lint

* add unit test

* update unit test

* remove modification

* [Feature]: Add MAE metafile (#238)

* [Feature]: Add MAE metafile

* [Fix]: Fix lint

* [Fix]: Change LARS to AdamW in the metafile of MAE

* [Fix] fix codecov bug (#241)

* [Fix] fix codecov bug

* update comment

* [Refactor] Using MMCls backbones (#233)

* [Refactor] using backbones from MMCls

* [Refactor] modify the unit test

* [Fix] modify default setting of out_indices

* [Docs] fix lint

* [Refactor] modify super init

* [Refactore] remove res_layer.py

* using mmcv PatchEmbed

* [Fix]: Fix outdated problem (#249)

* [Fix]: Fix outdated problem

* [Fix]: Update MoCov3 bibtex

* [Fix]: Use abs path in README

* [Fix]: Reformat MAE bibtex

* [Fix]: Reformat MoCov3 bibtex

* [Feature] Resume from the latest checkpoint automatically. (#245)

* [Feature] Resume from the latest checkpoint automatically.

* fix windows path problem

* fix lint

* add code reference

* [Docs] add docstring for ResNet and ResNeXt (#252)

* [Feature] support KNN benchmark (#243)

* [Feature] support KNN benchmark

* [Fix] add docstring and multi-machine testing

* [Fix] fix lint

* [Fix] change args format and check init_cfg

* [Docs] add benchmark tutorial

* [Docs] add benchmark results

* [Feature]: SimMIM supported (#239)

* [Feature]: SimMIM Pretrain

* [Feature]: Add mix precision and 16x128 config

* [Fix]: Fix config import bug

* [Fix]: Fix config bug

* [Feature]: Simim Finetune

* [Fix]: Log every 100

* [Fix]: Fix eval problem

* [Feature]: Add docstring for simmim

* [Refactor]: Merge layer wise lr decay to Default constructor

* [Fix]:Fix simmim evaluation bug

* [Fix]: Change model to be compatible to latest version of mmcls

* [Fix]: Fix lint

* [Fix]: Rewrite forward_train for classification cls

* [Feature]: Add UT

* [Fix]: Fix lint

* [Feature]: Add 32 gpus training for simmim ft

* [Fix]: Rename mmcls classifier wrapper

* [Fix]: Add docstring to SimMIMNeck

* [Feature]: Generate docstring for the forward function of simmim encoder

* [Fix]: Rewrite the class docstring for constructor

* [Fix]: Fix lint

* [Fix]: Fix UT

* [Fix]: Reformat config

* [Fix]: Add img resolution

* [Feature]: Add readme and metafile

* [Fix]: Fix typo in README.md

* [Fix]: Change BlackMaskGen to BlockwiseMaskGenerator

* [Fix]: Change the name of SwinForSimMIM

* [Fix]: Delete irrelevant files

* [Feature]: Create extra transformerfinetuneconstructor

* [Fix]: Fix lint

* [Fix]: Update SimMIM README

* [Fix]: Change SimMIMPretrainHead to SimMIMHead

* [Fix]: Fix the docstring of ft constructor

* [Fix]: Fix UT

* [Fix]: Recover deletion

Co-authored-by: Your <you@example.com>

* [Fix] add seed to distributed sampler (#250)

* [Fix] add seed to distributed sampler

* fix lint

* [Feature] Add ImageNet21k (#225)

* solve memory leak by limited implementation

* fix lint problem

Co-authored-by: liming <liming.ai@bytedance.com>

* [Refactor] change args format to '--a-b' (#253)

* [Refactor] change args format to `--a-b`

* modify tsne script

* modify 'sh' files

* modify getting_started.md

* modify getting_started.md

* [Fix] fix 'mkdir' error in prepare_voc07_cls.sh (#261)

* [Fix] fix positional parameter error (#260)

* [Fix] fix command errors in benchmarks tutorial (#263)

* [Docs] add brief installation steps in README.md (#265)

* [Docs] add colab tutorial (#247)

* [Docs] add colab tutorial

* fix lint

* modify the colab tutorial, using API to train the model

* modify the description

* remove #

* modify the command

* [Docs] translate 6_benchmarks.md into Chinese (#262)

* [Docs] translate 6_benchmarks.md into Chinese

* Update 6_benchmarks.md

change 基准 to 基准评测

* Update 6_benchmarks.md

(1)  Add Chinese translation of  ‘1 folder for ImageNet nearest-neighbor classification task’
(2) 数据预准备 -> 数据准备

* [Docs] remove install scripts in README (#267)

* [Docs] Update version information in dev branch (#268)

* update version to v0.8.0

* fix lint

* [Fix]: Install the latest mmcls

* [Fix]: Add SimMIM in RAEDME

Co-authored-by: Yuan Liu <30762564+YuanLiuuuuuu@users.noreply.github.com>
Co-authored-by: Jiahao Xie <52497952+Jiahao000@users.noreply.github.com>
Co-authored-by: Your <you@example.com>
Co-authored-by: Ming Li <73068772+mitming@users.noreply.github.com>
Co-authored-by: liming <liming.ai@bytedance.com>
Co-authored-by: RenQin <45731309+soonera@users.noreply.github.com>
Co-authored-by: YuanLiuuuuuu <3463423099@qq.com>
v0.8.0 		2022-03-31 18:47:54 +08:00
Yixiao Fang
16d9bf2e8c
Bump version to 0.7.1 (#236) 
* [Fix] fix mmcls version bug

* Update version

* [Docs] fix typo

* Update MMCls version dependency

* update version dependency
0.7.1 		2022-03-10 09:32:09 +08:00