Commit Graph

596 Commits (90f6292d899de0c23ae9420ad15e284a461174bb)
 

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
ZhaoQiiii c4e1ebb25a
[Docs]: Update datasets.md (#633)
* Update datasets.md

* [Fix]: Fix lint

* Update datasets.md

* Update docs/zh_cn/advanced_guides/datasets.md

Co-authored-by: Yuan Liu <30762564+YuanLiuuuuuu@users.noreply.github.com>

* Update docs/zh_cn/advanced_guides/datasets.md

Co-authored-by: Yuan Liu <30762564+YuanLiuuuuuu@users.noreply.github.com>

* Update docs/zh_cn/advanced_guides/datasets.md

Co-authored-by: Yuan Liu <30762564+YuanLiuuuuuu@users.noreply.github.com>

* Update docs/zh_cn/advanced_guides/datasets.md

Co-authored-by: Yixiao Fang <36138628+fangyixiao18@users.noreply.github.com>

* Update docs/zh_cn/advanced_guides/datasets.md

Co-authored-by: Yixiao Fang <36138628+fangyixiao18@users.noreply.github.com>

* fix lint

* update

Co-authored-by: YuanLiuuuuuu <3463423099@qq.com>
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>
2022-12-27 15:21:06 +08:00
Yixiao Fang b6585cc1f1
[Fix] Fix knn multi-gpu bug (#634)
* update knn

* update

* fix bugs of knn

* update entrance scripts

* update configs and related codes

* update sampler config

* remove redundance

* update

* update docs

* fix lint

* update logic of loading ckpt
2022-12-23 11:45:48 +08:00
Yixiao Fang 347b49e77f
[Refactor] Refactor t-SNE (#629)
* refactor tsne

* update configs

* update docs

* add cls_token option
2022-12-21 16:21:12 +08:00
YuanLiuuuuuu fd6659bba5 [Fix]: Add space 2022-12-19 15:16:40 +08:00
ZhaoQiiii 4da640b8d0 Update docs/zh_cn/advanced_guides/data_flow.md
Co-authored-by: Yuan Liu <30762564+YuanLiuuuuuu@users.noreply.github.com>
2022-12-19 15:16:40 +08:00
YuanLiuuuuuu 31170f8fc4 [Fix]: Fix lint 2022-12-19 15:16:40 +08:00
ZhaoQiiii c63d1962c0 Update data_flow.md 2022-12-19 15:16:40 +08:00
Yuan Liu 1049791766
[Fix]: Change np.int to int (#636) 2022-12-19 15:14:03 +08:00
Yixiao Fang fbe0ff6ec7
[Fix] Fix key error in configs (#630) 2022-12-14 10:26:52 +08:00
fangyixiao18 bd4609658a [Fix] Fix import error 2022-12-14 09:47:25 +08:00
Yuan Liu a194464863
[Feature]: Add pretraining for FGIA (#607)
* [Feature]: Add pretraining for FGIA

* [Fix]: Add requirements and title

* [Refactor]: Move readme to root folder

* [Feature]: Add cls link

* [Fix]: Fix typo
2022-12-12 18:26:33 +08:00
Yixiao Fang 83e0917482
[Feature] Support BEiT v2 (#627)
* update

* update configs

* update model zoo

* fix lint

* update config
2022-12-12 14:14:49 +08:00
Yixiao Fang a4b96dc662
[Refactor] Refactor projects folder (#620)
* refactor projects folder

* fix lint

* fix typo

* refine readme template

* update downsteam
2022-12-12 14:12:50 +08:00
Yuan Liu aa836ac835
Merge pull request #623 from YuanLiuuuuuu/hotfix/mmcls2timm
[Feature]: Change map_location to cpu
2022-12-09 14:42:19 +08:00
YuanLiuuuuuu 976de2e518 [Feature]: Change map_location to cpu 2022-12-09 14:40:40 +08:00
Yuan Liu 5244b66d5c
[Feature]: Convert scripts and instructions on seg and det (#621)
* [Feature]: Add converting script

* [Feature]: Add instruction for det and seg

* [Feature]: Fix citation

* [Fix]: Use absolute path

* [Fix]: Change tree to blob
2022-12-09 13:42:35 +08:00
Yuan Liu f6049ed379
[Fix]: Fix the link of SimMIM 800pt 100ft (#622)
* [Feature]: Fix test dataloader bug

* [Fix]: Fix the link of 800pt 100ft SimMIM
2022-12-08 16:52:11 +08:00
Yuan Liu ce4b78b171
[Fix]: Change pseudo_collect to default_collect (#616) 2022-12-08 11:12:29 +08:00
Yuan Liu 0d6632a5c7
Merge pull request #618 from open-mmlab/dev-1.x
Merge dev-1.x to 1.x
2022-12-07 11:46:22 +08:00
Yixiao Fang c11ac99355
Bump version to v1.0.0rc4 (#613)
* Bump version to v1.0.0rc4

* fix typo
2022-12-07 11:35:05 +08:00
Yixiao Fang b74c3bb25c
[Docs] add readthedocs algorithm pages and fix some displaying error (#599)
* update algo readme

* add script to generate algo pages in model zoo

* fix lint

* update requirements

* fix lint

* update title

* fix readthedocs docstring displaying error

* update gitignore

* fix displaying

* fix lint
2022-12-06 20:17:34 +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
Yixiao Fang 8b56a78f93
[Fix] Fix dependence and key bug (#611)
* fix dependence bug

* fix
2022-12-06 18:58:54 +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
Yuan Liu ce08509014
[Fix]: register data preprocessor (#603) 2022-12-05 14:07:49 +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 6db04339a6
[Community] Add './projects' folder and an example (#586)
* add projects folder and readme

* refine description

* fix lint

* update ci

* update

* refine docs
2022-11-23 20:42:31 +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
Yixiao Fang 1835af3a91 [Docs] update readthedocs rst and menu button (#572)
* update the version switcher menu

* update rst

* update

* update
2022-11-10 17:45:27 +08:00
Yixiao Fang 7980414026 Bump version to v1.0.0rc3 (#566)
* change mmcv to mmengine

* update version

* update

* update readme

* update requirements

* update changelog

* fix lint
2022-11-03 16:09:36 +08:00
Songyang Zhang f0d03d4fcd [Docs] translate 4 custom dataset (#561)
* [Docs] translate 4 custom dataset

* Update docs/zh_cn/user_guides/4_pretrain_custom_dataset.md

Co-authored-by: Yixiao Fang <36138628+fangyixiao18@users.noreply.github.com>

* update

Co-authored-by: Yixiao Fang <36138628+fangyixiao18@users.noreply.github.com>
2022-11-03 16:09:36 +08:00
Songyang Zhang e4703e382b [Docs] translate overview (#562)
* [Docs] translate overview

* upadte figure of design
2022-11-03 16:09:36 +08:00
Yixiao Fang 35cb9b81cd [Docs] Translate model_zoo.md (#556)
* translate model_zoo.md

* fix lint

* update maskfeat link

* Update docs/zh_cn/model_zoo.md

Co-authored-by: Songyang Zhang <tonysy@users.noreply.github.com>

* Update docs/zh_cn/model_zoo.md

Co-authored-by: Songyang Zhang <tonysy@users.noreply.github.com>

* Update docs/zh_cn/model_zoo.md

Co-authored-by: Songyang Zhang <tonysy@users.noreply.github.com>

Co-authored-by: Songyang Zhang <tonysy@users.noreply.github.com>
2022-11-03 16:09:36 +08:00
Yixiao Fang 27374e1f0d [Docs] Refine some English version docs (#563)
* add visualization examples

* refine docs

* remove old version contents

* update

* update

* fix lint
2022-11-03 16:09:36 +08:00
Yixiao Fang 80442e21c6 [Docs] Refine migration.md and translate it (#544)
* refine migration.md

* translate to Chinese

* Update docs/en/migration.md

Co-authored-by: Yuan Liu <30762564+YuanLiuuuuuu@users.noreply.github.com>

* Update docs/zh_cn/migration.md

Co-authored-by: Yuan Liu <30762564+YuanLiuuuuuu@users.noreply.github.com>

* refine data_preprocessor description

* fix lint

Co-authored-by: Yuan Liu <30762564+YuanLiuuuuuu@users.noreply.github.com>
2022-11-03 16:09:36 +08:00
RenQin 88e1b5d4db [Docs]: Translate 2_dataset_prepare.md to Chinese (#551)
* [Docs]: translate 2_dataset_prepare.md to zh_cn

* [Fix]: fix cifar10 dataset prepare

* [Fix]: fix cifar10 dataset prepare

* [Fix]: fix cifar10 dataset prepare
2022-11-03 16:09:36 +08:00
RenQin 536ac8f448 [Docs]: Translate 3_pretrain.md to Chinese (#552)
* [Docs]: Translate 3_pretrain.md to Chinese

* [Docs]: adjust the details

* [Fix]: fix lint
2022-11-03 16:09:36 +08:00
Jiahao Xie 5e56d2a1e1 [Docs] translate get_started.md to zh_cn (#548)
* [Docs] translate

* [Docs] translate

* [Docs] fix

* [Docs] fix lint

* [Docs] fix lint

* [Docs] fix
2022-11-03 16:09:36 +08:00
Yuan Liu 811ccdad46 [Docs] translate config.md to zh_cn (#543)
* [Feature]: Add Chinese doc for config

* [Fix]: Fix incomplete translation

* [Feature]: Refine config_cn
2022-11-03 16:09:36 +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
Jiahao Xie d142699265 [Docs] refactor add_modules.md (#524)
* [Docs] refactor add_modules.md

* [Docs] fix lint
2022-11-03 16:09:36 +08:00
Songyang Zhang 5a0e0eedc4 [Doc] Add custom dataset tutorial (#522)
* [Doc] Add the tutorial for using custom dataset

* [Doc] Add coco example

* [Doc] Fix bug in command

* [Doc] Fix bug in command

* Update docs/en/user_guides/4_pretrain_custom_dataset.md

Co-authored-by: Yuan Liu <30762564+YuanLiuuuuuu@users.noreply.github.com>

* [Doc] Update the doc according to the reviews

Co-authored-by: Yuan Liu <30762564+YuanLiuuuuuu@users.noreply.github.com>
2022-11-03 16:09:36 +08:00
Yuan Liu f8a33ae543 [Feature] Refine mae-h-448 fine-tuning config (#509)
* [Feature]: Add ceph to vit-h-448

* [Feature]: Refine ViT-H 448 test setting

* [Fix]: Fix lr

* [Fix]: Use disk backend
2022-11-03 16:09:36 +08:00