496e098b21
[Feature] Support some downstream classification datasets. ( #1467 )
...
* feat: support some downstream classification datasets
* update sun397
* sum
* update sun397
* [CI] Add test mim CI. (#879 )
* feat: support some downstream classification datasets
* update sun397
* sum
* update sun397
* rebase
* feat: support some downstream classification datasets
* update sun397
* update sun397
* update sun397
* update sun397
* fix unittest
* update docstring
* rm
* update
* update
* refactor names of datasets
* refactor some implements of datasets
* refactor some implements of datasets
* fix datasets unittest
* refactor cub and stanford cars
* refactor cub and cifar
* refactor cub and cifar
* refactor cub and cifar
* update downstream datasets and docs
* update docstring
---------
Co-authored-by: Ma Zerun <mzr1996@163.com>
Co-authored-by: Ezra-Yu <18586273+Ezra-Yu@users.noreply.github.com>
2023-05-05 14:43:14 +08:00
08dc8c75d3
[Refactor] Add selfsup algorithms. ( #1389 )
...
* remove basehead
* add moco series
* add byol simclr simsiam
* add ut
* update configs
* add simsiam hook
* add and refactor beit
* update ut
* add cae
* update extract_feat
* refactor cae
* add mae
* refactor data preprocessor
* update heads
* add maskfeat
* add milan
* add simmim
* add mixmim
* fix lint
* fix ut
* fix lint
* add eva
* add densecl
* add barlowtwins
* add swav
* fix lint
* update readtherdocs rst
* update docs
* update
* Decrease UT memory usage
* Fix docstring
* update DALLEEncoder
* Update model docs
* refactor dalle encoder
* update docstring
* fix ut
* fix config error
* add val_cfg and test_cfg
* refactor clip generator
* fix lint
* pass check
* fix ut
* add lars
* update type of BEiT in configs
* Use MMEngine style momentum in EMA.
* apply mmpretrain solarize
---------
Co-authored-by: mzr1996 <mzr1996@163.com>
2023-03-06 16:53:15 +08:00
414ba80274
[Refactor] Refactor APIs, add `ImageRetrievalInferencer` and `FeatureExtractor`. ( #1380 )
...
* [Refactor] Refactor APIs, add `ImageRetrievalInferencer` and `FeatureExtractor'.
* Update image retrieval
* Update FeatureExtractor
* Fix UT
2023-02-28 16:31:42 +08:00
0979e78573
Rename the package name to `mmpretrain`.
2023-02-17 15:20:55 +08:00
6ea59bd846
[Fix] Fix the requirements and lazy register mmcls models. ( #1275 )
2022-12-19 13:01:11 +08:00
3b6d44ea05
[Refactor] Refactor CIFAR10 and CIFAR100 dataset.
2022-07-18 10:53:56 +08:00
0b20aa6873
[Fix]: Fix missing runtime requirement `packaging`. ( #459 )
2021-09-23 18:04:42 +08:00
fe45b241c5
Do not install mmcv through requirements ( #90 )
...
* Delete the mmcv requirement to support mmcv-full installed environment
* install mmcv in CI
* fix mmcv version error
Co-authored-by: Xinran <lixinran@sensetime.com>
2020-11-19 00:00:12 +08:00
4f4f2957ef
Use build_runner ( #54 )
...
* Use build_runner in train api
* Support iter in eval_hook
* Add runner section
* Add test_eval_hook
* Pin mmcv version in install docs
* Replace max_iters with max_epochs
* Set by_epoch=True as default
* Remove trailing space
* Replace DeprecationWarning with UserWarning
* pre-commit
* Fix tests
2020-10-15 21:12:50 +08:00
c00ee6a876
workflow: modify build, add deploy ( #61 )
...
* workflow: modify build, add deploy
* fix type: mmseg -> mmcls
* simplify build by delete cuda related settings
* re-sort requirements by pre-commit hook
* build: support more pytorch versions
* build: fix torchvision version bug
* del useless test imports
* merge upstream
* build: del docstring checking to pass coverage check
2020-10-10 22:15:50 +08:00
24fd4fb627
Visualize results on image demo ( #58 )
...
* visualize results on image demo
* add matplotlib in requirements
2020-10-10 16:33:27 +08:00
71e3b86ed3
add github action (manually squash)
2020-07-12 00:07:05 +08:00
zzc98
Yixiao Fang
Ma Zerun
RangiLyu
Wenwei Zhang
David de la Iglesia Castro
Y. Xiong
Lei Yang
yl-1993