9506241f73
[Feature] Add arcface head. ( #926 )
...
* feat: add arcface head
* feat: add arcface head
* update arcface
* refactor archface head
* update archface head
* update archface head
* use torch.cat instead of torch.hstack to fix ci
Co-authored-by: Ezra-Yu <18586273+Ezra-Yu@users.noreply.github.com>
2022-11-02 17:45:33 +08:00
693596bc2f
[Feature] Add Base Retriever and Image2Image Retriever for retrieval tasks. ( #1055 )
...
* feat: add image retriever
* feat: add image retriever
* feat: add image retriever
* feat: add image retriever
* feat: add image retriever
* feat: add image retriever
* feat: add image retriever
* feat: add image retriever
* feat: add image retriever
* update retriever
* fix lint
* add hook unit test
* Use `register_buffer` to save prototype vectors and add a progress bar
during preparing prototype.
* update UTs
* update UTs
* fix typo
* modify the hook
Co-authored-by: Ezra-Yu <18586273+Ezra-Yu@users.noreply.github.com>
Co-authored-by: mzr1996 <mzr1996@163.com>
2022-11-02 17:43:56 +08:00
b526f018db
[Fix] Fix broken inference api because of the modification in data
...
preprocessor.
2022-10-18 17:50:35 +08:00
bcca619066
[Feature] Support MobileViT backbone. ( #1068 )
...
* init
* fix
* add config
* add meta
* add unittest
* fix for comments
* Imporvee docstring and support custom arch.
* Update README
* Update windows CI
Co-authored-by: mzr1996 <mzr1996@163.com>
2022-10-18 17:05:59 +08:00
29f066f7fb
[Improve] Speed up data preprocessor. ( #1064 )
...
* [Improve] Speed up data preprocessor.
* Add ClsDataSample serialization override functions.
* Add unit tests
* Modify configs to fit new mixup args.
* Fix `num_classes` of the ImageNet-21k config.
* Update docs.
2022-10-17 17:08:18 +08:00
a49c3076e1
[Feature] Support DeiT3. ( #1065 )
...
* deit3
deit3
lint
* add tools and test
* deit3
* deit3
* fix preprocess
* lint
* Update config names and checkpoint paths
* Update convert tools to use mmengine, and fix docstring.
Co-authored-by: mzr1996 <mzr1996@163.com>
2022-10-10 14:54:20 +08:00
2153a16dc5
[Docs] Fix some docstrings.
2022-10-10 11:17:48 +08:00
8c5d86a388
[Fix] Fix `loss_weight` in `LabelSmoothLoss`. ( #1058 )
2022-09-28 08:18:12 +08:00
51b7c754e7
[Fix] Fix the output position of Swin-Transformer. ( #947 )
...
* [Fix] Fix the output position of Swin-Transformer.
* Rename `downsample` argument to `do_downsample`.
2022-09-22 17:11:52 +08:00
034683cfcd
[Feature] Support MViT for MMCLS 1.x ( #1023 )
...
* [Feature] Support MViT for MMCLS 1.x
* Add unit tests
* Update documentation
2022-09-22 09:23:49 +08:00
f7191ca686
[Feature] Add ViT huge architecture. ( #1049 )
...
* [Feature]: Add ViT huge settings
* [Feature]: Add comments
2022-09-21 13:27:54 +08:00
81f10ccbb7
[Feature] Support EdgeNeXt for dev-1.x. ( #1037 )
...
* [Feature] add EdgeNeXt for dev-1.x
* [Feature] add update
* Imporve EdgeNeXt
* Fix div error in low pytorch version
Co-authored-by: mzr1996 <mzr1996@163.com>
2022-09-21 13:27:04 +08:00
e9e2d48cb2
[Improve] Update benchmark scripts ( #1028 )
...
* Update train benchmark scripts
* Add `--cfg-options` for dev scripts and enhance `--range`.
* Fix bug of regex expression.
* Fix minor bugs
* Update ShuffleNet configs
* Update rsb-a1 configs and label smooth loss mode.
* Update inference dev scripts
* From `mmengine` instead of `mmcv` import fileio.
* Fix lint
* Update pre-commit hook
* Use `use_sigmoid` option instead of "bce" mode in label smooth loss.
2022-09-20 15:50:21 +08:00
e4e8047563
[Feature] Support Swin Transformer V2 for MMCLS 1.x. ( #1029 )
...
* [Feature] support swin v2 in mmcls 1.x
* fix configs
* fix clamp max
* Fix meshgrid
* Imporve swin v2
Co-authored-by: mzr1996 <mzr1996@163.com>
2022-09-20 15:45:27 +08:00
aaf127c5e1
[Feature] Add efficientformer Backbone for MMCls 1.x. ( #1031 )
...
* rebase
* update filename
* update URL
* update UT
* fix lint
* update head
* add efficientformer
* update filename
* update UT
* fix lint
* update configs
* rebase
* fix unit tests
* Fix comments and docs.
Co-authored-by: mzr1996 <mzr1996@163.com>
2022-09-20 14:56:45 +08:00
f1d2f50c21
[Feature] Add MobileOne Backbone For MMCls 1.x. ( #1030 )
...
* add mobileOne
* add train cfg
* update cfgs
* update URL
* update configs
* update inceptionv3 metafile
* add configs
* fix lint
* update checkpoint urls
* Update configs
* Update README
Co-authored-by: mzr1996 <mzr1996@163.com>
2022-09-16 11:55:10 +08:00
0b61b76325
[Docs] Add Migration documentation. ( #997 )
...
* [Docs] Add Migration documentation.
* Fix the train_cfg argument
* [WIP] Improve according to comments
* Imporve according to comments.
2022-08-31 13:31:48 +08:00
2b88df4484
[Refactor] Refactor the data flow. ( #989 )
...
* [Refactor] Refactor the data flow.
* Add comments about data preprocessor.
* Fix after mmengine folder structure refactoring.
2022-08-26 10:40:43 +08:00
b7d0d521eb
[Refactor] Use `mmengine` instead of `mmcv` and refactor some transforms. ( #986 )
...
* [Refactor] Refactor the rest data transforms and fix some docstring.
* Use `mmengine` instead of `mmcv`
* Refactor KFold dataset tools
* Fix docstring according to comments
2022-08-24 15:59:02 +08:00
5665b8349a
[Refactor] Refactor import dependency. ( #977 )
...
Co-authored-by: mzr1996 <mzr1996@163.com>
2022-08-22 15:17:33 +08:00
20f9ace5c5
[CI] Add Circle CI for mmcls 1.x. ( #970 )
...
* [CI] Add Circle CI for mmcls 1.x
* Fix circle-ci config
* Use `torch.cat` instead of `torch.hstack` to compat with PyTorch 1.6
* Compat the behavior of LongTensor in PyTorch 1.6
* Avoid random behavior in the unit test of Lighting
* Fix cuda ci
* Remove github workflow temporarily.
2022-08-22 15:02:08 +08:00
e4252d6848
[Feature]: Support BEiT Transformer layer. ( #919 )
...
* [Feature]: Add BEiT-style transformer encoder layer
* [Feature]: Add average token
* [Fix]: Fix lint
* [Fix]: Refactor CAE config
* [Fix]: Change cv2 backend to pillow backend
* [Fix]: Fix MAE and CAE reshape bug
* [Feature]: Add freeze vit layers
* [Feature]: Add mc
* [Fix]: Fix lint
* [Fix]: Fix dataset bug
* [Fix]: Delete cae selfsup config
* [Fix]: docstring
* [Refactor]: Add init_values to layer_scalue_init_value
* [Fix]: Refine the docstring of avg_token
* [Fix]: Call super init weight in beit attention
* [Fix]: remove mc
* [Fix]: Fix docstring
* [Fix]: Fix docstring
* [Fix]: Fix lint
* [Fix]: Fix init_value bug and change the logic of outputting cls token
* [Fix]: Fix docstring
2022-08-17 00:07:06 +08:00
43e60ad5a6
[Improve] Rename `mmcls.data` to `mmcls.structures`. ( #941 )
2022-07-29 14:18:13 +08:00
9118b97239
refactor evaluation
2022-07-18 11:11:13 +08:00
6ed3860ea6
refactor UTs
2022-07-18 11:11:13 +08:00
0f30c392a3
from core to engine
2022-07-18 11:11:13 +08:00
24bcf069f8
[Refactor] Refactor dev scripts
2022-07-18 11:11:13 +08:00
e9342d9e4c
Add multi_label heads
2022-07-18 11:11:13 +08:00
62b046521e
[Refactor] refactor ViTHead, DeiTHead, ConformerHead, StackedHead
2022-07-18 11:11:13 +08:00
daa716b112
Add some type hints
2022-07-18 11:11:13 +08:00
69e5ab065e
Change "Default to" to "Defaults to"
2022-07-18 11:11:13 +08:00
375fe68f12
Imporve according to comments
2022-07-18 11:11:13 +08:00
f3299b4ca2
[Refactor] Refactor batch augmentations
2022-07-18 11:11:13 +08:00
cecff79a79
Rename 'feat' mode to 'tensor' mode
2022-07-18 11:11:13 +08:00
ee6e585e41
Mazerun/base classifier
2022-07-18 11:11:13 +08:00
0537c4d70c
[Feature] Add `ClsVisualizer`.
2022-07-18 10:53:27 +08:00
de002e455f
[Refactor] Use `MMLogger` from MMEngine in `get_logger` and `print_log`.
2022-07-18 10:53:27 +08:00
088d5b5add
[Refactor] Inherit all registries from MMEngine and use inherited registries.
2022-07-18 10:53:27 +08:00
3e198e63e8
[Docs] Update install tutorials. ( #854 )
...
* [Docs] Update install tutorials.
* [Docs] Improve dataset docs
* Add option to show the results in demo.
* fix typo
2022-06-01 18:31:57 +08:00
7c5ddb1e5b
Bump version to v0.23.0 ( #809 )
2022-05-01 21:58:33 +08:00
b4eefe4794
[Enhance] Support training on IPU and add fine-tuning configs of ViT. ( #723 )
...
* implement training and evaluation on IPU
* fp16 SOTA
* Tput reaches 5600
* 123
* add poptorch dataloder
* change ipu_replicas to ipu-replicas
* add noqa to config long line(website)
* remove ipu dataloder test code
* del one blank line in test_builder
* refine the dataloder initialization
* fix a typo
* refine args for dataloder
* remove an annoted line
* process one more conflict
* adjust code structure in mmcv.ipu
* adjust ipu code structure in mmcv
* IPUDataloader to IPUDataLoader
* align with mmcv
* adjust according to mmcv
* mmcv code structre fixed
Co-authored-by: hudi <dihu@graphcore.ai>
2022-04-29 22:22:19 +08:00
59292b315a
[Feature] Support DenseNet. ( #750 )
...
* init add densenet implementation
* Add config and converted models
* update meta
* add test for memory efficient
* Add docs
* add doc for jit
* Update checkpoint path
* Update readthedocs
Co-authored-by: mzr1996 <mzr1996@163.com>
2022-04-28 13:06:03 +08:00
df6edd7f5a
[Feature] Support VAN. ( #739 )
...
* add van
* fix config
* add metafile
* add test
* model convert script
* fix review
* fix lint
* fix the configs and improve docs
* rm debug lines
* add VAN into api
Co-authored-by: Yu Zhaohui <1105212286@qq.com>
2022-04-28 09:35:17 +08:00
702c196514
[Docs] Refine API reference. ( #774 )
...
* [Docs] Refine API reference
* Add PoolFormer
2022-04-15 20:19:20 +08:00
29b882d4eb
Merge branch 'dev'
2022-04-15 20:10:45 +08:00
2eb27d0d4e
[Feature] Add PoolFormer backbone and checkpoints. ( #746 )
...
* add PoolFormer
* fix some typos in PoolFormer
* fix lint error
* modify out_indices and gap
* fix typo
* fix lint
* fix typo
* fix typo in poolforemr README
* fix lint
* Update some paths
* Refactor freeze_stages method
* Add unit tests
* Fix lint
Co-authored-by: mzr1996 <mzr1996@163.com>
2022-04-15 11:35:15 +08:00
fbcf907dc6
[Feature] Support resize relative position embedding in `SwinTransformer`. ( #749 )
...
* [Feature]: Add resize rel pos embed
* [Refactor]: Create a separated resize_rel_pos_bias_table func
* [Refactor]: Refactor rel pos embed bias
* [Refactor]: Move interpolate into func
* Remove index buffer only when window_size changes
Co-authored-by: mzr1996 <mzr1996@163.com>
2022-04-13 23:06:56 +08:00
875195efe9
[Enhance] Improve CPE performance by reduce memory copy. ( #762 )
2022-04-02 11:12:20 +08:00
922a31a68e
[Feature] ADD RepMLP Backbone and checkpoints. ( #709 )
...
* add repmlp
* refactor backbone code
* add checkpoint, readme, unit tests
* unchange demo
* improve docstring.
* update pic
* refactor patch embed
* refactor unit tests
* fix lint
* update tools
* update tools
* Update checkpoint path
Co-authored-by: mzr1996 <mzr1996@163.com>
2022-03-30 20:40:25 +08:00
292127098f
[Feature] Add CSPNet and backbone and checkpoints ( #735 )
...
* add cspnet backbone
* try
* add backbones
* add cspnet
* add acc
* add unit tests
* chnage mode
* Update checkpoint path
* Fix typo
* Imporve CSPNet
* Update checkpoints
Co-authored-by: mzr1996 <mzr1996@163.com>
2022-03-30 15:29:35 +08:00
zzc98
mzr1996
Hubert
takuoko
whcao
Yuan Liu
Songyang Zhang
Ezra-Yu
Yixiao Fang
Ezra-Yu
yingfhu
Hu Di
Weihao Yu
XiaobingZhang