mirrors
/
mmselfsup
mirror of https://github.com/open-mmlab/mmselfsup.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity
619 Commits
5 Branches
19 Tags
8 MiB
Commit Graph

105 Commits (68129f0a9c84d89057429116021d0544091bb12a)

Author SHA1 Message Date
Yixiao Fang 0f634de85b
[Feature] support MaskFeat pre-training with video (#678) 
* add maskfeat video configs

* add maskfeat video

* update configs

* update readme

* update results and links

* update configs

* update configs

* update links

* refine readme

* refine doc

* add ut

* refine readme

* update ut

* update readme

* update docs

* update config

* update train scripts in readme

* add mmaction2 page link
 2023-02-07 16:40:50 +08:00
Yuan Liu 304e81650a
[Feature]: Shape bias (#635) 
* [Feature]: Add shape bias vis

* [Fix]: Fix lint

* [Feature]: Add shape bias metrics

* [Fix]: Fix lint

* [Fix]: Delete redundant code

* [Feature]: Add shape bias doc

* [Fix]: Fix lint

* [Feature]: add UT

* [Fix]: Fix lint

* [Fix]: Fix typo

* [Fix]: Fix typo

* [Fix]: Fix args param style

* [Feature]: Download pic automatically
 2022-12-30 22:11:18 +08:00
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
Yixiao Fang a4e9f8f0a9
[Refactor] refactor file io and add ut (#582) 
* remove fileclient

* add ut
 2022-11-18 16:44:54 +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
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