17a886cb58
Bump version to v1.2.0 ( #1860 )
...
* [Fix] Fix resize mix argument bug.
* Bump version to v1.2.0
* Fix UT
2024-01-04 20:43:27 +08:00
fa53174fd9
[Feature]: Add MFF ( #1725 )
...
* [Feature]: Add MFF
* [Feature]: Add mff linear prob
* [Feature]: Add ft
* [Fix]: Update docstring
* [Feature]: Update out_indices
* [Feature]: Add prefix to ft
* [Feature]: Add README
* [Feature]: Update readme
* [Feature]: Update README
* [Feature]: Add metafile
* [Feature]: Update README
* [Fix]: Fix lint
* [Feature]: Add UT
* [Feature]: Update paper link
2023-08-08 16:01:07 +08:00
58a2243d99
Merge branch 'main' into dev
2023-07-28 15:35:55 +08:00
e7fc25cf64
[Fix] Fix nested predict for multi-task prediction. ( #1716 )
...
* fix: multi task predict
* change the loop
---------
Co-authored-by: Pierre Colle <piercus@gmail.com>
2023-07-28 13:44:12 +08:00
c5248b17b7
[Enhance] Adapt test cases on Ascend NPU. ( #1728 )
2023-07-28 13:39:38 +08:00
2b8d8eecb2
[Fix] Fix the issue #1711 "GaussianBlur doesn't work" ( #1722 )
...
* Fix issue 1711. GaussianBlur.
* Fix UT
---------
Co-authored-by: mzr1996 <mzr1996@163.com>
2023-07-25 11:25:32 +08:00
64c446d507
[Feature] Support LoRA. ( #1687 )
...
* [Feature] Support LoRA
* [Feature] Support LoRA
* [Fix] Fix bugs
* [Refactor] Add copyright
* [Fix] Fix bugs
* [Enhancement] Add
* [Fix] Fix bugs
* [Fix] Fix bugs
* [Fix] Fix bugs
* [Fix] Fix bugs
* [Fix] Fix bugs
* [Docs] Update docstring
* [Docs] Update docstring
* [Refactor] Reformat with yapf
* [Docs] Update docstring
* [Refactor] Docformat
* [Refactor] Fix double-quote-string
* [Fix] fix pytorch version
* [Fix] isort
* [Fix] isort
* [Enhancement] Extend forward
* [Enhancement] Extend test
* [Fix] Fix targets
* [Enhancement] Extend LoRA to frozen models
* [Fix] Fix spelling
* [Fix] Override __getattr__
* [Fix] Add init_cfg
* [Enhancement] Add example config
* [Fix] Fix init_cfg
* [Enhancement] Add merging script
* [Fix] Remove init_cfg
* [Fix] Change lora key
* [Fix] Fix merge scripts
* [Fix] Fix merge scripts
* [Docs] Add docs
* [Fix] fix
2023-07-24 11:30:57 +08:00
feb0814b2f
[Feature] Transfer shape-bias tool from mmselfsup ( #1658 )
...
* Transfer shape-bias tool from mmselfsup
* run shape-bias successfully, add CN docs
* fix unit test bug
* add shape_bias to index.rst in docs
* modified mistakes in shape-bias docs
2023-07-03 11:39:23 +08:00
a1cfe888e2
[Feature] Support SparK. ( #1531 )
...
* add spark configs
* fix configs
* remove repeat aug
* add module codes
* support lr layer decay of resnet
* update
* fix lint
* add metafile and readme
* fix lint
* add models and logs
* refactor codes
* fix lint
* update model rst
* update name
* add docstring
* add ut
* fix lint
---------
Co-authored-by: Ma Zerun <mzr1996@163.com>
2023-06-19 11:27:50 +08:00
bc3c4a35ee
[Refactor] Support to use "split" to specify training set/validation set in the ImageNet dataset ( #1535 )
...
* [Feature]: Add caption
* [Feature]: Update scienceqa
* [CI] Add test mim CI. (#879 )
* refactor imagenet dataset
* refactor imagenet dataset
* refactor imagenet dataset
* update imagenet21k
* update configs
* update mnist
* update dataset_prepare.md
* fix sun397 url and update user_guides/dataset_prepare.md
* update dataset_prepare.md
* fix sun397 dataset
* fix sun397
* update chinese dataset_prepare.md
* update dataset_prepare.md
* [Refactor] update voc dataset
* [Refactor] update voc dataset
* refactor imagenet
* refactor imagenet
* use mmengine.fileio
---------
Co-authored-by: liuyuan <3463423099@qq.com>
Co-authored-by: Ma Zerun <mzr1996@163.com>
Co-authored-by: Ezra-Yu <18586273+Ezra-Yu@users.noreply.github.com>
2023-06-02 11:03:18 +08:00
5bd088ef43
[Fix] Update torchvision transform wrapper ( #1595 )
...
* Update torchvision transform wrapper
* Update requirements
* fix unit tests
---------
Co-authored-by: Ezra-Yu <18586273+Ezra-Yu@users.noreply.github.com>
2023-05-26 17:56:09 +08:00
e4c4a81b56
[Feature] Support iTPN and HiViT ( #1584 )
...
* hivit added
* Update hivit.py
* Update hivit.py
* Add files via upload
* Update __init__.py
* Add files via upload
* Update __init__.py
* Add files via upload
* Update hivit.py
* Add files via upload
* Add files via upload
* Add files via upload
* Add files via upload
* Update itpn.py
* Add files via upload
* Update __init__.py
* Update mae_hivit-base-p16.py
* Delete mim_itpn-base-p16.py
* Add files via upload
* Update itpn_hivit-base-p16.py
* Update itpn.py
* Update hivit.py
* Update __init__.py
* Update mae.py
* Delete hivit.py
* Update __init__.py
* Delete configs/itpn directory
* Add files via upload
* Add files via upload
* Delete configs/hivit directory
* Add files via upload
* refactor and add metafile and readme
* update clip
* add ut
* update ut
* update
* update docstring
* update model.rst
---------
Co-authored-by: 田运杰 <48153283+sunsmarterjie@users.noreply.github.com>
2023-05-26 12:08:34 +08:00
1f07c92ed1
[Feature] Add retrieval mAP metric. ( #1552 )
...
* rebase
* fefine
* fix lint
* update readme
* rebase
* fix lint
* update docstring
* update docstring
* rebase
* rename corespanding names
* rebase
2023-05-26 10:40:08 +08:00
46a523ef63
[Feature] Add GQA dataset. ( #1585 )
...
* [Feature]: Add GQA dataset
* [Feature]: Add GQA
* [Feature]: Add GQA UT
* [Fix]: Fix hint
* [Feature]: Add BLIP2 GQA
* [Fix]: Fix lint
* [Feature]: Update anno link
* [Fix]: Update docstring
* [Feature]: Update all links
2023-05-23 11:25:42 +08:00
be389eb846
[Fix] Fix scienceqa ( #1581 )
2023-05-22 16:10:17 +08:00
13e4d6c512
[Fix]: Fix UT
2023-05-22 11:55:08 +08:00
87f849cbb6
[Feature]: Add scienceqa metric
2023-05-22 11:31:07 +08:00
6847d20d57
[Feature] Support multiple multi-modal algorithms and inferencers. ( #1561 )
...
* [Feat] Migrate blip caption to mmpretrain. (#50 )
* Migrate blip caption to mmpretrain
* minor fix
* support train
* [Feature] Support OFA caption task. (#51 )
* [Feature] Support OFA caption task.
* Remove duplicated files.
* [Feature] Support OFA vqa task. (#58 )
* [Feature] Support OFA vqa task.
* Fix lint.
* [Feat] Add BLIP retrieval to mmpretrain. (#55 )
* init
* minor fix for train
* fix according to comments
* refactor
* Update Blip retrieval. (#62 )
* [Feature] Support OFA visual grounding task. (#59 )
* [Feature] Support OFA visual grounding task.
* minor add TODO
---------
Co-authored-by: yingfhu <yingfhu@gmail.com>
* [Feat] Add flamingos coco caption and vqa. (#60 )
* first init
* init flamingo coco
* add vqa
* minor fix
* remove unnecessary modules
* Update config
* Use `ApplyToList`.
---------
Co-authored-by: mzr1996 <mzr1996@163.com>
* [Feature]: BLIP2 coco retrieval (#53 )
* [Feature]: Add blip2 retriever
* [Feature]: Add blip2 all modules
* [Feature]: Refine model
* [Feature]: x1
* [Feature]: Runnable coco ret
* [Feature]: Runnable version
* [Feature]: Fix lint
* [Fix]: Fix lint
* [Feature]: Use 364 img size
* [Feature]: Refactor blip2
* [Fix]: Fix lint
* refactor files
* minor fix
* minor fix
---------
Co-authored-by: yingfhu <yingfhu@gmail.com>
* Remove
* fix blip caption inputs (#68 )
* [Feat] Add BLIP NLVR support. (#67 )
* first init
* init flamingo coco
* add vqa
* add nlvr
* refactor nlvr
* minor fix
* minor fix
* Update dataset
---------
Co-authored-by: mzr1996 <mzr1996@163.com>
* [Feature]: BLIP2 Caption (#70 )
* [Feature]: Add language model
* [Feature]: blip2 caption forward
* [Feature]: Reproduce the results
* [Feature]: Refactor caption
* refine config
---------
Co-authored-by: yingfhu <yingfhu@gmail.com>
* [Feat] Migrate BLIP VQA to mmpretrain (#69 )
* reformat
* change
* change
* change
* change
* change
* change
* change
* change
* change
* change
* change
* change
* change
* change
* change
* change
* change
* change
* change
* refactor code
---------
Co-authored-by: yingfhu <yingfhu@gmail.com>
* Update RefCOCO dataset
* [Fix] fix lint
* [Feature] Implement inference APIs for multi-modal tasks. (#65 )
* [Feature] Implement inference APIs for multi-modal tasks.
* [Project] Add gradio demo.
* [Improve] Update requirements
* Update flamingo
* Update blip
* Add NLVR inferencer
* Update flamingo
* Update hugging face model register
* Update ofa vqa
* Update BLIP-vqa (#71 )
* Update blip-vqa docstring (#72 )
* Refine flamingo docstring (#73 )
* [Feature]: BLIP2 VQA (#61 )
* [Feature]: VQA forward
* [Feature]: Reproduce accuracy
* [Fix]: Fix lint
* [Fix]: Add blank line
* minor fix
---------
Co-authored-by: yingfhu <yingfhu@gmail.com>
* [Feature]: BLIP2 docstring (#74 )
* [Feature]: Add caption docstring
* [Feature]: Add docstring to blip2 vqa
* [Feature]: Add docstring to retrieval
* Update BLIP-2 metafile and README (#75 )
* [Feature]: Add readme and docstring
* Update blip2 results
---------
Co-authored-by: mzr1996 <mzr1996@163.com>
* [Feature] BLIP Visual Grounding on MMPretrain Branch (#66 )
* blip grounding merge with mmpretrain
* remove commit
* blip grounding test and inference api
* refcoco dataset
* refcoco dataset refine config
* rebasing
* gitignore
* rebasing
* minor edit
* minor edit
* Update blip-vqa docstring (#72 )
* rebasing
* Revert "minor edit"
This reverts commit 639cec757c215e654625ed0979319e60f0be9044.
* blip grounding final
* precommit
* refine config
* refine config
* Update blip visual grounding
---------
Co-authored-by: Yiqin Wang 王逸钦 <wyq1217@outlook.com>
Co-authored-by: mzr1996 <mzr1996@163.com>
* Update visual grounding metric
* Update OFA docstring, README and metafiles. (#76 )
* [Docs] Update installation docs and gradio demo docs. (#77 )
* Update OFA name
* Update Visual Grounding Visualizer
* Integrate accelerate support
* Fix imports.
* Fix timm backbone
* Update imports
* Update README
* Update circle ci
* Update flamingo config
* Add gradio demo README
* [Feature]: Add scienceqa (#1571 )
* [Feature]: Add scienceqa
* [Feature]: Change param name
* Update docs
* Update video
---------
Co-authored-by: Hubert <42952108+yingfhu@users.noreply.github.com>
Co-authored-by: yingfhu <yingfhu@gmail.com>
Co-authored-by: Yuan Liu <30762564+YuanLiuuuuuu@users.noreply.github.com>
Co-authored-by: Yiqin Wang 王逸钦 <wyq1217@outlook.com>
Co-authored-by: Rongjie Li <limo97@163.com>
2023-05-19 16:50:04 +08:00
770eb8e24a
[Fix] Fix ddp bugs caused by `out_type`. ( #1570 )
...
* set out_type to be 'raw'
* update test
2023-05-17 17:32:10 +08:00
034919d032
[Feature] add eva02 backbone ( #1450 )
...
* [CI] Add test mim CI. (#879 )
* [CI] Add test mim CI. (#879 )
* feat: add eva02 backbone
* feat: add eva02 backbone
* feat: add eva02 backbone
* feat: add eva02 backbone
* feat: add eva02 backbone
* feat: add eva02 backbone
* feat: add eva02 backbone
* feat: add eva02 backbone
* update
* update ci
* rebase
* feat: add eva02 backbone
* feat: add eva02 backbone
* feat: add eva02 backbone
* feat: add eva02 backbone
* feat: add eva02 backbone
* feat: add eva02 backbone
* feat: add eva02 backbone
* feat: add eva02 backbone
* update
* update readme and configs
* update readme and configs
* refactore eva02
* [CI] Add test mim CI. (#879 )
* feat: add eva02 backbone
* feat: add eva02 backbone
* feat: add eva02 backbone
* feat: add eva02 backbone
* feat: add eva02 backbone
* feat: add eva02 backbone
* feat: add eva02 backbone
* feat: add eva02 backbone
* update
* update ci
* rebase
* feat: add eva02 backbone
* feat: add eva02 backbone
* feat: add eva02 backbone
* update
* update readme and configs
* refactore eva02
* update readme and metafile
* update readme and metafile
* update readme and metafile
* update
* rename eva02
* rename eva02
* fix uts
* rename configs
---------
Co-authored-by: Ma Zerun <mzr1996@163.com>
Co-authored-by: Ezra-Yu <18586273+Ezra-Yu@users.noreply.github.com>
2023-05-06 19:28:31 +08:00
d9e561a09d
[Feature] Support dinov2 backbone ( #1522 )
...
* support dinov2 backbone
* update metafile and readme
* compatible to use_layer_scale
* update SwiGLUFFN
* add deprecation warning
* update
2023-05-05 16:59:37 +08:00
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
99e48116aa
[Feature] Register torchvision transforms into mmcls ( #1265 )
...
* [Enhance] Add stochastic depth decay rule in resnet. (#1363 )
* add stochastic depth decay rule to drop path rate
* add default value
* update
* pass ut
* update
* pass ut
* remove np
* rebase
* update ToPIL and ToNumpy
* rebase
* rebase
* rebase
* rebase
* add readme
* fix review suggestions
* rebase
* fix conflicts
* fix conflicts
* fix lint
* remove comments
* remove useless code
* update docstring
* update doc API
* update doc
---------
Co-authored-by: Yixiao Fang <36138628+fangyixiao18@users.noreply.github.com>
2023-04-13 18:05:57 +08:00
0826df8963
[Feature] Add ViT of SAM ( #1476 )
...
* add vit of sam
* update
* update
* add ut
* update ut
* remove num_classes
* support dynamic input
* add ut
* add comments
* update ut
2023-04-13 17:03:28 +08:00
47e033c466
[Fix] fix pbn bug ( #1466 )
2023-04-07 15:38:04 +08:00
32c258ff19
update riformer mmpretrain
2023-04-06 21:56:25 +08:00
0b70c108b0
update riformer mmpretrain
2023-04-06 21:56:25 +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
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
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
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
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
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
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
4ce7be17c9
[Enhance] Enable to toggle whether Gem Pooling is trainable or not. ( #1246 )
...
* Enable to toggle whether Gem Pooling is trainable or not.
* Add test case whether Gem Pooling is trainable or not.
* Enable to toggle whether Gem Pooling is trainable or not by requires_grad
---------
Co-authored-by: Yusuke Fujimoto <yusuke.fujimoto@rist.co.jp>
2023-02-09 11:27:05 +08:00
a3f2effb17
[Feature] Add `ImageClassificationInferencer`. ( #1261 )
...
* [Feature] Add ImageClassificationInferencer.
* Update inferencer implementation and add unit tests.
* Update documentations.
* Update pre-commit hook
* Update docs
2023-02-08 14:30:12 +08:00
7e4502b0ac
[Feature] Support InShop Dataset (Image Retrieval). ( #1019 )
...
* rebase
* feat: add inshop dataset (retrieval)
* update fileIO
* update unit tests
* fix windows ci
* fix windows ci
* fix lint
* update unit tests
* update docs
* update docs
Co-authored-by: Ezra-Yu <18586273+Ezra-Yu@users.noreply.github.com>
2023-01-18 17:16:54 +08:00
6b9e2b55dd
[Feature] Support LeViT backbone. ( #1238 )
...
* 网络搭建完成、能正常推理
* 网络搭建完成、能正常推理
* 网络搭建完成、能正常推理
* 添加了模型转换未验证，配置文件 但有无法运行
* 模型转换、结构验证完成，可以推理出正确答案
* 推理精度与原论文一致 已完成转化
* 三个方法改为class 暂存
* 完成推理精度对齐 误差0.04
* 暂时使用的levit2mmcls
* 训练跑通，训练相关参数未对齐
* '训练相关参数对齐'参数'
* '修复训练时验证导致模型结构改变无法复原问题'
* '修复训练时验证导致模型结构改变无法复原问题'
* '添加mixup和labelsmooth'
* '配置文件补齐'
* 添加模型转换
* 添加meta文件
* 添加meta文件
* 删除demo.py测试文件
* 添加模型README文件
* docs文件回滚
* model-index删除末行空格
* 更新模型metafile
* 更新metafile
* 更新metafile
* 更新README和metafile
* 更新模型README
* 更新模型metafile
* Delete the model class and get_LeViT_model methods in the mmcls.models.backone.levit file
* Change the class name to Google Code Style
* use arch to provide default architectures
* use nn.Conv2d
* mmcv.cnn.fuse_conv_bn
* modify some details
* remove down_ops from the architectures.
* remove init_weight function
* Modify ambiguous variable names
* Change the drop_path in config to drop_path_rate
* Add unit test
* remove train function
* add unit test
* modify nn.norm1d to build_norm_layer
* update metafile and readme
* Update configs and LeViT implementations.
* Update README.
* Add docstring and update unit tests.
* Revert irrelative modification.
* Fix unit tests
* minor fix
Co-authored-by: mzr1996 <mzr1996@163.com>
2023-01-17 17:43:42 +08:00
Ma Zerun
Yuan Liu
fangyixiao18
marouane amzil
Yinlei Sun
liyl
fanqiNO1
Wang Xiang
Yixiao Fang
zzc98
Ezra-Yu
liuyuan
techmonsterwang
QINGTIAN
takuoko
fam_taro
aso538