Commit Graph

96 Commits (798041402645fd89fea0284667f68ee1d29e75c0)

Author SHA1 Message Date
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
Yixiao Fang 86726ec615 [Refactor] Refine configs (#547)
* update ci workflow

* add test for config

* update configs

* update imagenet 1pct and 10pct

* update configs according to mmcls interface

* update link

* add copyright

* update ci

* update ci install command

* update data_preprocessor field

* fix value of mean, std and add comments

* remove redundant configs
2022-11-03 16:09:36 +08:00
Yixiao Fang 5e44bf7179 [Refactor] Refactor evaluation folder and related configs (#538)
* refactor evaluation folder

* update knn script

* update ut

* update configs

* update config names according to mmdet and mmseg

* update docs
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
RenQin 83b1a33bed
[Refactor]: Rename SelfSupLocalVisualizer to SelfSupVisualizer (#392)
* [Refactor]: rename SelfSupLocalVisualizer to SelfSupVisualizer

* [Fix]: fix example errors in docstring
2022-08-15 16:01:15 +08:00
Yixiao Fang 0ea07c0750
[Refactor] Refactor hooks to support non-dist training (#397)
* apply mmengine.dist instead of torch.dist

* appy get_model to densecl_hook

* fix bug to pass ut

* update typehint
2022-08-08 15:01:47 +08:00
Yuan Liu a703ba2fcb
[Refactor]: Refactor layer wise lr decay (#375)
* [Refactor]: Refactor layer wise lr decay

* [Fix]: Remove is_dcn_module and add log

* [Fix]: Fix lint

* [Fix]: Check relative pos bias table is not decayed

* [Fix]: Fix lint

* [Fix]: Fix UT
2022-08-05 17:38:53 +08:00
Yixiao Fang 045b1fde8e
[Refactor] Apply get_current_instance instead of get_root_logger (#394) 2022-08-04 20:26:09 +08:00
Yixiao Fang c8cf491c4c
[Refactor] Update version check rule (#393) 2022-08-04 17:12:15 +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
Yixiao Fang 067fbce0b5
[CI] Add circleci (#363)
* add circleci

* fix linting

* check version of torchvision

* fix torch dependency bug

* using torch.stack instead of vstack

* comment github actions temporarily

* pass docs

* update docs requirements
2022-07-25 14:11:57 +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
renqin 4e28ee7ff4 [Feature]: add local_visualizer 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
YuanLiuuuuuu 408b9fb187 [Fix]: Fix layer wise learning rate decay optim wrapper 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
fangyixiao.vendor df8c204d75 [Refactor] refactor hooks and add ut 2022-07-18 11:06:44 +08:00
fangyixiao.vendor f78fe71794 [Refactor] refactor swav hook to control gradient 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 ad382c2115 [Refactor] add unit test for deepcluster dataset and sampler 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 62c909d3d2 [Refactor]: Rename optimizer wrapper constructor 2022-07-18 11:06:44 +08:00