0b70c108b0
update riformer mmpretrain
2023-04-06 21:56:25 +08:00
1ee9bbe050
[Docs] Update links ( #1457 )
...
* update links
* update readtherdocs
* update
* update
* fix lint
* update
* update
* update
* update cov branch
* update
* update
* update
2023-04-06 20:58:52 +08:00
3069e43f77
[Docs] Update readme ( #1449 )
...
* update readme
* update
* refine
* refine
* update cn version
* update installation
* update modelzoo table
* fix lint
* update
* update
* update
* update
* fix lint
* update
* update
* update changelog
* remove gif
* fix typo
* update announcement
* update
* fixtypo
* update
2023-04-06 17:17:56 +08:00
75dceaa78f
[Refactor] Add ln to vit avg_featmap output ( #1447 )
2023-04-06 11:59:39 +08:00
3a25b13eb3
[Fix] Update CI ( #1454 )
...
* update ci
* update window ci
* update window ci
* update window ci
* update window ci
* update window ci
* update window ci
* update window ci
* update window ci
* update window ci
* update window ci
* update window ci
* ignore tests/test_tools.py
2023-04-06 10:52:08 +08:00
568188a6b0
[Docs] add overview in the homepage ( #1446 )
...
* update index overview
* fix lint
* refine with chatgpt
* fix lint
* update according to review
* update cn version
2023-04-03 16:39:58 +08:00
9fb4e9c911
[Fix] fix config of maskfeat ( #1424 )
2023-03-30 11:45:18 +08:00
445eb3223a
[Docs] Refine advanced guides ( #1428 )
...
* refine
* update description
* update links
* update links
* update installation
* refine
2023-03-29 16:23:57 +08:00
b017670e1b
[Improve] Use PyTorch official `scaled_dot_product_attention` to accelerate `MultiheadAttention`. ( #1434 )
...
* [Improve] Use PyTorch official `scaled_dot_product_attention` to accelerate `MultiheadAttention`.
* Support `--local-rank` and `--amp` option for new version PyTorch.
* Fix imports and UT.
2023-03-29 15:50:44 +08:00
164f16e248
[Fix] Fix init bug of r50 in contrastive leanrning ( #1445 )
2023-03-29 15:49:51 +08:00
555adab0a0
[Doc] Update logos and Add more social networking links ( #1440 )
...
* add social media link in README
* replace logos
* update links
* update logo
* upodate logo size
2023-03-29 13:35:56 +08:00
53dc810c08
[Refactor] Add projects from mmselfsup ( #1410 )
...
* update projects/fgia
* add video maskfeat in projects
* update according to review
2023-03-27 16:59:08 +08:00
c4ccae40db
[Docs] Update user guides docs and tools for MMPretrain. ( #1429 )
...
* [Docs] Update user guides docs and tools for MMPretrain.
* Fix UT
* Fix Chinese docs.
* Improve according to comments.
* Fix windows CI.
2023-03-27 14:32:26 +08:00
a50d96f7f1
Update docs.
2023-03-20 16:12:10 +08:00
175d19f67e
Update docs.
2023-03-20 16:10:33 +08:00
1f78ab410f
Update docs.
2023-03-20 16:05:53 +08:00
6038df9514
Update docs.
2023-03-20 16:03:57 +08:00
f6b65fcbe7
[Docs] Update get start docs and user guides. ( #1407 )
...
* update user_guides
* update test.md
* fix lint
* fix typo
* refine
* fix typo
* update retriever to api
* update rst and downstream
* update index.rst
* update index.rst
* update custom.js
* update chinese docs
* update config.md
* update train and test
* add pretrain on custom dataset
* fix lint
2023-03-20 15:56:09 +08:00
04e15ab347
Update circle-CI
2023-03-20 14:55:08 +08:00
6cedce234e
[Refactor] Update dev scripts to be compatible with selfsup tasks. ( #1412 )
...
* [Refactor] Update dev scripts to be compatible with selfsup tasks.
* Fix some missing fields in config files.
* Set maximum number of gpus for local training.
* Update README files
* Update according to comments.
2023-03-20 14:30:57 +08:00
4f5b38f225
[Refactor] Update almost tools and add unit tests for these tools. ( #1418 )
...
* [Refactor] Update almost tools and add unit tests for these tools.
* Fix Windows UT.
2023-03-17 10:50:51 +08:00
8875e9da92
[Docs] Update migration.md ( #1417 )
...
* update migration
* refine table
* update zh_cn
* fix lint
* Polish the documentation by ChatGPT.
* Update sphinx version and fix some warning.
---------
Co-authored-by: mzr1996 <mzr1996@163.com>
2023-03-17 10:30:09 +08:00
76a1f3f735
[Refactor] Refactor the `browse_dataset.py` to support selfsup pipeline. ( #1414 )
...
* refactor browsedataset to support selfsup pipeline
* update make_grid to support list input
* mode 'transformed' supports list
* Beautify the visualization image.
* Fix compatitibly bug with matplotlib=3.5
* remove print
* fix bug of resize
* Apply mask only on the first image.
* Remove master only for some API in visualizer.
---------
Co-authored-by: mzr1996 <mzr1996@163.com>
2023-03-15 14:18:36 +08:00
3472ee5d2c
[Feature] Implememnt the universal visualizer for multiple task. ( #1404 )
...
* [Feature] Implememnt the universal visualizer for multiple task.
* Update tools
* Improve according to comments.
* Fix tools docs
* Add --test-cfg option and set default collate function.
2023-03-09 11:36:54 +08:00
dbf3df21a3
[Refactor] Use `out_type` to specify ViT-like backbone output. ( #1408 )
...
* [Refactor] Use to specify ViT-like backbone output.
* Fix ClsBatchNormNeck
* Update mmpretrain/models/necks/mae_neck.py
---------
Co-authored-by: Yixiao Fang <36138628+fangyixiao18@users.noreply.github.com>
2023-03-09 11:02:58 +08:00
63e5b512cc
[Refactor] Move part of tool scripts from mmselfsup. ( #1403 )
...
* add dataset converters and benchmark .sh
* refine
* fix lint
* add tsne
* rename visualizaition
* update configs and script
2023-03-07 17:57:08 +08:00
274a67223e
[Feature] Implement layer-wise learning rate decay optimizer constructor. ( #1399 )
...
* [Feature] Implement layer-wise learning rate decay optimizer constructor.
* Use num_layers instead of max_depth to avoid misleading
* Add UT
* Update docstring
* Update log info
* update LearningRateDecay configs
---------
Co-authored-by: fangyixiao18 <fangyx18@hotmail.com>
2023-03-07 17:30:39 +08:00
827be6e22d
[Fix] Fix value error while start training. ( #1400 )
...
* fix value error of randint
* add missing key in configs
2023-03-07 08:51:31 +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
a05c79e806
[Refactor] Move transforms in mmselfsup to mmpretrain. ( #1396 )
...
* [Refactor] Move transforms in mmselfsup to mmpretrain.
* Update transform docs and configs. And register some mmcv transforms in
mmpretrain.
* Fix missing transform wrapper.
* update selfsup transforms
* Fix UT
* Fix UT
* update gaussianblur inconfigs
---------
Co-authored-by: fangyixiao18 <fangyx18@hotmail.com>
2023-03-03 15:01:11 +08:00
1d6e37e56b
Avoid build pdf docs.
2023-03-02 18:14:35 +08:00
e035e03d59
Update docs style.
2023-03-02 13:29:48 +08:00
dda3d6565b
[Docs] Update generate_readme.py and readme files. ( #1388 )
...
* Update generate_readme.py and readme files.
* Update reamde
* Update docs
* update metafile
* update simmim readme
* update
* update mae
* fix lint
* update mocov2
* update readme pic
* fix lint
* Fix mmcls download links.
* Fix Chinese docs.
* Decrease readthedocs requirements.
---------
Co-authored-by: fangyixiao18 <fangyx18@hotmail.com>
2023-03-02 13:29:07 +08:00
c9670173aa
[Refactor] Move and refactor utils from mmselfsup. ( #1385 )
...
* add heads
* add losses
* fix
* remove mim head
* add modified backbones and target generators
* fix lint
* fix lint
* add heads
* add losses
* fix
* add data preprocessor from mmselfsup
* add ut for data prepocessor
* add GatherLayer
* add ema
* add batch shuffle
* add misc
* fix lint
* update
* update docstring
2023-02-28 17:04:40 +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
e453a45d31
[Refactor] Add self-supervised backbones and target generators. ( #1379 )
...
* add heads
* add losses
* fix
* remove mim head
* add modified backbones and target generators
* add unittest
* refactor caevit
* add window_size check
* fix lint
* apply new DataSample
* fix ut error
* update ut
* fix ut
* fix lint
* Update base modules.
---------
Co-authored-by: mzr1996 <mzr1996@163.com>
2023-02-28 15:59:17 +08:00
63d9f27fde
[Refactor] Add necks, heads and losses for the self-supervised task. ( #1376 )
...
* add necks
* refactor linear neck
* rename simmim neck
* add heads
* add losses
* fix
* add unittest
* update
* update cae
* remove mim head
* update config
2023-02-28 10:05:00 +08:00
75c79311f4
[Refactor] Update datasets ( #1375 )
...
* add ut
* add places205
* support ann_file without labels
* temp test
* update custom
* update
* update ut
* Update CustomDataset.
* Update Places205.
---------
Co-authored-by: mzr1996 <mzr1996@163.com>
2023-02-27 15:42:22 +08:00
89000c10eb
[Refactor] Refactor configs and metafile ( #1369 )
...
* update base datasets
* update base
* update barlowtwins
* update with new convention
* update
* update
* update
* add schedule
* add densecl
* add eva
* add mae
* add maskfeat
* add milan and mixmim
* add moco
* add swav simclr
* add simmim and simsiam
* refine
* update
* add to model index
* update config inheritance
* fix error in metafile
* Update pre-commit and metafile check script
* update metafile
* fix name error
* Fix classification model name and config name
---------
Co-authored-by: mzr1996 <mzr1996@163.com>
2023-02-23 11:17:16 +08:00
36bea13fca
[Refactor] Refactor ClsDatasample to a union DataSample. ( #1371 )
...
* [Refactor] Refactor ClsDatasample to a union DataSample.
* Add method
* Fix docstring
* Update docstring.
2023-02-23 10:07:53 +08:00
4016f1348e
Fix CI
2023-02-17 15:33:16 +08:00
0979e78573
Rename the package name to `mmpretrain`.
2023-02-17 15:20:55 +08:00
8352951f3d
[Feature] Support XCiT Backbone. ( #1305 )
...
* update model file
* Update XCiT implementation and configs.
* Update metafiles
* Update metafile
* Fix floor divide
* Imporve memory usage
---------
Co-authored-by: qingtian <459291290@qq.com>
Co-authored-by: mzr1996 <mzr1996@163.com>
2023-02-15 10:32:35 +08:00
bedf4e9f64
[Refactor] Update analysis tools and documentations. ( #1359 )
...
* [Refactor] Update analysis tools and documentations.
* Update migration.md and add unit test.
* Fix print_config.py
2023-02-15 10:28:08 +08:00
b4ee9d2848
[Feature] Support calculate confusion matrix and plot it. ( #1287 )
...
* [Feature] Support calculate confusion matrix and plot it.
* Fix keepdim
* Update confusion_matrix tools and the plot graph.
* Revert accidental modification.
* Update docstring
* Move confusion matrix tool to
2023-02-14 12:58:11 +08:00
841256b630
[Feature] Support RetrieverRecall metric & Add ArcFace config ( #1316 )
...
* rebase
* add ap metric
* fix mlti-gpu bug in retrevel
* rebase
* rebase
* add training cfgs and update readme.md
* fix bugs(cannot load vecs in dist and diff test-val recall\)
* update configs and readme
* fix ut
* fix doc
* rebase
* fix rebase conflicts
* fix rebase error
* fix UT error
* fix docs
* fix typo
---------
Co-authored-by: Ezra-Yu <18586273+Ezra-Yu@users.noreply.github.com>
2023-02-14 12:46:21 +08:00
1c1273abca
[Docs] Translate some tools tutorials to Chinese. ( #1321 )
...
* Delete print_config.md
* Add files via upload
2023-02-09 16:40:13 +08:00
705ed2be49
[Fix] Fix retrieval multi gpu bug ( #1319 )
...
* fix mlti-gpu bug in retrevel
* fix bugs(cannot load vecs in dist and diff test-val recall\)
* load weight each process
2023-02-09 15:55:47 +08:00
7ec6062415
[Refactor] Unify the `--out` and `--dump` in `tools/test.py`. ( #1307 )
2023-02-09 14:05:03 +08:00
58cefa5c0f
[Fix] Fix error repvgg-deploy base config path. ( #1357 )
2023-02-09 11:27:35 +08:00
techmonsterwang
Yixiao Fang
Ezra-Yu
Ma Zerun
QINGTIAN
takuoko
huangxuming