66 Commits

Author SHA1 Message Date
Yuan Liu
90f6292d89
[Refactor]: Refactor CAE (#645)
* [Feature]: Add target generator config and param

* [Feature]: Use target generator to generate target

* [Fix]: Fix lint

* [Fix]: Fix UT

* [Fix]: Delete petrel

* [Fix]: Delete download ckpt

* [Fix]: Refactor CAE config
2022-12-30 17:27:20 +08:00
Wangbo Zhao(黑色枷锁)
da8b6c5146
[Feature] Support EVA-MAE style (#632)
* [Feature] Support EVA-MAE style

* [Feature] Refine After Review1

* [Feature] Refine After Review2 and Review3. Add ckpt links

* [Feature] Refine After Review4

* [Feature] Refine After Review5

* [Feature] Refine After Review6

* [Feature] Fix file name
2022-12-27 15:47:32 +08:00
Yixiao Fang
73cd764b5f
[Feature] Support pixel reconstruction visualization (#570)
* refactor reconstruction visualization

* support simmim visualization

* fix reconstruction bug of MAE

* support visualization of MaskFeat

* refaction mae visualization demo

* add unit test

* fix lint and ut

* update

* add docs

* set random seed

* update

* update docstring

* add torch version check

* update

* rename

* update version

* update

* fix lint

* add docstring

* update docs
2022-12-06 19:45:01 +08:00
Yuan Liu
d73c953804
[Feature]: MILAN supported (#600)
* [Feature]: Add milan ft config

* [Feature]: Add milan linear prob

* [Feature]: Set diff rank seed in mae ft

* [Feature]: V1

* [Feature]: Add target generator

* [Feature]: Add MILAN head and loss

* [Feature]: Refine milan

* [Feature]: Delete redundant mask and ids_shuffle

* [Feature]: Delete redundant return value of attention masking

* [Feature]: Detele return attention param

* [Feature]: Add typehint and docstring for PromptDecoder and PromptAttention

* [Feature]: Add type hint and docstring

* [Feature]: Fix lint

* [Fix]: Remove petrel backend

* [Feature]: Delete redundant code in clip

* [Feature]: Add ut for milan algorithm

* [Feature]: Delete redundant code

* [Feature]: Use mock for target generator

* [Feature]: Add docstring

* [Feature]: Create classification folder in milan

* [Feature]: Add README

* [Feature]: Add metafile

* [Feature]: Add main paper readme

* [Feature]: Update model zoom

* [Feature]: Fix review

* [Feature]: Fix config path bug

* [Feature]: Fix review#2

* [Feature]: Delete MILAN loss

* [Fix]: Add metafile

* [Fix]: Fix lint

* [Feature]: Change the test milan

* [Feature]: Update the config file name
2022-12-06 19:42:13 +08:00
RenQin
7a7b048f23
[Feature]: Add BEiT Support (#425)
* [Feature]: Add BEiT Support

* [Fix]: fix bugs after update

* [Fix]: fix bugs in backbone

* [Refactor]: refactor config

* [Feature]: Support BEiTv2

* [Fix]: Fix UT

* [Fix]: rename some configs

* [Fix]: fix beitv2neck

* [Refactor]: refactor beitv2

* [Fix]: fix lint

* refactor configs

* refactor beitv2

* update configs

* add dalle target generator

* refactor for beitv1

* refactor rel_pos_bias of beit

* update configs

* update configs

* update v1 configs

* update v2 configs

* refactoe layer decay

* update unittest

* fix lint

* fix ut

* add docstrings

* rename

* fix lint

* add beit model and log links

* fix lint

* update according to review

* update

* update

* update LearningRateDecayOptimWrapperConstructor
related configs

* update init and backbone

* update neck and vqkd

* refactor neck

* fix lint

* add some comments

* fix typo

Co-authored-by: 任琴 <PJLAB\renqin@shai14001114l.pjlab.org>
Co-authored-by: fangyixiao18 <fangyx18@hotmail.com>
2022-12-06 16:40:05 +08:00
Yixiao Fang
bc0807d3cf
Update CI and UT (#601)
* update ci

* update ut

* fix lint

* update
2022-12-02 17:09:00 +08:00
lkylkylky
e761acd1bd [Feature] Add Maskfeat-1.x Support (#494)
* [Feature]: Add MaskfeatMaskGenerator Pipeline

* [Feature]: Add MaskFeatMaskGenerator Pipeline

* [Feature]: Add Backbone of MaskFeat

* [Feature]: Add HogLayerC for MaskFeat

* [Feature]: Add Loss of MaskFeat

* [Feature]: Add Head of MaskFeat

* [Feature]: Add Algorithms of MaskFeat

* [Feature]: Add Config of MaskFeat

* [Fix] fix ut of MaskFeatMaskGenerator

* refine configs

* update

* refactor to support hog generator

* update config

* update

* update config and metafiel

* update maskfeat model link

* fix ut

* refine codes

* fix lint

* refine docstring

* refactor maskfeat head

* update model links

* fix ut

* refine docstring

* update model-index

* using BEiTMaskGenerator

* refine configs

* update ut

* fix lint

Co-authored-by: fangyixiao18 <fangyx18@hotmail.com>
2022-11-03 16:09:36 +08:00
Yixiao Fang
bd75fc67b4 [Refactor] Add target generator (#518)
* add target generator

* add copyright

* add docstring for target_generator

* refine docstring for HOG
2022-11-03 16:09:36 +08:00
RenQin
41747f73c7 [Refactor]: refactor MAE visualization (#471)
* [Refactor]: refactor MAE visualization

* [Fix]: fix lint

* [Refactor]: refactor MAE visualization

* [Feature]: add mae_visualization.py

* [UT]: add unit test

* [Refactor]: move mae_visualization.py to tools/analysis_tools

* [Docs]: Add the purpose of the function unpatchify()

* [Fix]: fix lint
2022-11-03 16:09:36 +08:00
Yixiao Fang
ff5f5ae2a9 [CI] update CI package (#476)
* update circleci

* update github action

* fix windows error

* update

* update pr and merge state

* fix lint

* update

* update

* update circle ci

* fix

* fix ut
2022-10-12 19:35:44 +08:00
Yuan Liu
20488d01b4
[Refactor]: Refactor data flow (#429)
* [Refactor]: Refactor data flow

* [Fix]: Change data sample to data samples

* [Fix]: Change batch_inputs to inputs

* [Fix]: Fix lint and UT

* [Fix]: Fix UT

* [Fix]: Fix lint

* [Fix]: Fix docstring

* [Fix]: Fix UT

* [Refactor]: Add assert in data preprocessor

* [Fix]: Fix lint
2022-08-30 11:34:04 +08:00
Yixiao Fang
3c51e86d44
[Fix] Fix ut (#422)
* add register all modules

* add scope
2022-08-19 14:36:26 +08:00
Yuan Liu
5d6133d60c
[Refactor]: Remove MIM ViT related code (#387)
* [Refactor]: Delete mim vit and related files

* [Fix]: Remove mim-vit from __init__.py
2022-08-01 14:14:03 +08:00
Yixiao Fang
ba50b7450b
[Refactor] Rename mmselfsup.data to mmselfsup.structures (#385) 2022-07-30 16:36:48 +08:00
Yuan Liu
10d9539f67
[Feature]: Add reconstruction loss (#365)
* [Feature]: Add pixel reconstruction loss

* [Fix]: Fix lint
2022-07-22 17:28:50 +08:00
Yixiao Fang
c7101c7648
[Enhance] support non-dist training of simclr (#352)
* support non-dist training of simclr

* revise ut
2022-07-19 14:21:26 +08:00
Yixiao Fang
af3b261dce [Enhance] support non-dist training of mocov3 (#346)
* support non-dist training of mocov3

* revise docstring
2022-07-18 11:06:44 +08:00
Yixiao Fang
210ea79316 fix import error (#348) 2022-07-18 11:06:44 +08:00
fangyixiao.vendor
4cb6aa3895 [Refactor] refactor package 2022-07-18 11:06:44 +08:00
xiejiahao.vendor
f4c19381c9 [Refactor] refactor dc and odc 2022-07-18 11:06:44 +08:00
fangyixiao.vendor
d390a03a47 [Refactor] support to train moco in non-dist environment 2022-07-18 11:06:44 +08:00
renqin
4c5c4b88f4 [Refactor]: refactor npid algorithm 2022-07-18 11:06:44 +08:00
renqin
dfa4d180df [Refactor]: refactor swav algorithm 2022-07-18 11:06:44 +08:00
renqin
5f778aa552 [Refactor]: refactor densecl algorithm 2022-07-18 11:06:44 +08:00
YuanLiuuuuuu
d6dfa9fe40 [Fix]: Move CAE data preprocessor to data preprocessor file and fix UT 2022-07-18 11:06:44 +08:00
YuanLiuuuuuu
8910743c6e [Fix]: Fix multi-head config 2022-07-18 11:06:44 +08:00
fangyixiao.vendor
1e16016b27 [Refactor] refactor byol 2022-07-18 11:06:44 +08:00
YuanLiuuuuuu
acf001d554 [Refactor]: Refactor CAE 2022-07-18 11:06:44 +08:00
fangyixiao.vendor
61fdc30c59 [Refactor] refactor simclr 2022-07-18 11:06:44 +08:00
fangyixiao.vendor
c9a532e1c0 [Refactor] refactor mocov3 benchmark 2022-07-18 11:06:44 +08:00
fangyixiao.vendor
bc199203a0 [Refactor] refactor rotation pred 2022-07-18 11:06:44 +08:00
fangyixiao.vendor
87ed42aaeb [Refactor] refactor barlowtwins 2022-07-18 11:06:44 +08:00
renqin
7528fbcbe8 [Refactor]: refactor simsiam algorithm 2022-07-18 11:06:44 +08:00
fangyixiao.vendor
5a84e88609 [Refactor] refactor mocov3 2022-07-18 11:06:44 +08:00
YuanLiuuuuuu
bcc4576ace [Refactor]: Refactor relative loc 2022-07-18 11:06:44 +08:00
YuanLiuuuuuu
5ba17adb23 [Refactor]: Refactor SimMIM 2022-07-18 11:06:44 +08:00
fangyixiao.vendor
b9df061e7f [Refactor] refactor moco according to new basemodel 2022-07-18 11:06:44 +08:00
YuanLiuuuuuu
35e0988527 [Refactor]: Refactor base model and create a sample for MAE 2022-07-18 11:06:44 +08:00
fangyixiao.vendor
962f9b9752 [Refactor] refactor multi_cls_head and clean cls algorithm 2022-07-18 11:06:44 +08:00
fangyixiao18
5ea54a48c8 fix ut 2022-07-18 11:06:44 +08:00
fangyixiao.vendor
ceda0eed68 [Refactor] add unit test for part of model utils 2022-07-18 11:06:44 +08:00
YuanLiuuuuuu
0f10a4debe [Feature]: Add ut for backbones 2022-07-18 11:06:44 +08:00
YuanLiuuuuuu
0a2a811acf [Fix]: Fix lint 2022-07-18 11:06:44 +08:00
fangyixiao.vendor
1176808264 [Refactor] refactor utils 2022-07-18 11:06:44 +08:00
fangyixiao.vendor
c39277b216 [Feature] apply EMA from mmengine and add CosineEMA 2022-07-18 11:06:44 +08:00
YuanLiuuuuuu
17136bf935 [Feature]: Add pseudo_label in selfsupdatasample 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
YuanLiuuuuuu
0c88e8dceb [Refactor]: Refactor loss 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