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
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
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
b8b31e9343
[Improve] Multiple fixes. ( #957 )
...
* Rename some files in unit tests.
* Rename `pipelines` folder to `transforms`.
* Fix version requirement range.
* Remove `init_random_seed` and `set_random_seed`.
* Fix unit tests
* Rename `convert_models` to `model_converters`.
2022-08-09 15:58:08 +08:00
43e60ad5a6
[Improve] Rename `mmcls.data` to `mmcls.structures`. ( #941 )
2022-07-29 14:18:13 +08:00
35cba87f82
refactor version and utils
2022-07-18 11:11:13 +08:00
9118b97239
refactor evaluation
2022-07-18 11:11:13 +08:00
317216efc6
fix visualizer registry bug
2022-07-18 11:11:13 +08:00
1c82863449
fix lint
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
efaa93aae1
[Refactor] Refactor entry scripts `train.py` and `test.py`.
2022-07-18 10:53:56 +08:00
de002e455f
[Refactor] Use `MMLogger` from MMEngine in `get_logger` and `print_log`.
2022-07-18 10:53:27 +08:00
702c196514
[Docs] Refine API reference. ( #774 )
...
* [Docs] Refine API reference
* Add PoolFormer
2022-04-15 20:19:20 +08:00
f552419e45
[Enhance] Add setup multi-processing both in train and test. ( #671 )
2022-01-27 10:18:36 +08:00
894a82ea4f
[Refactor] Change `load_json_logs` to `load_json_log`. ( #602 )
2021-12-14 18:03:27 +08:00
b486bb23f8
[Feature] Add a tool to visualize learning rate in each iterations ( #498 )
...
* add vis lr
* add doc
* Update doc and rm load_json_log in analysis
* Update docs
* add unit tests and update docstring
* fix unit-tests
* Use DummyIterBasedRunner and DummyEpochBasedRunner to simulate training
process.
* rm function val and add judge
* update docs
* improve english expression
* Update docs/tools/visualization.md
Co-authored-by: Ma Zerun <mzr1996@163.com>
* fix typo and ImageNet datasize
* fix typo
* update example
Co-authored-by: mzr1996 <mzr1996@163.com>
2021-11-03 10:48:56 +08:00
f9eb9b409b
[Docs] Add Copyright information. ( #413 )
2021-08-17 19:52:42 +08:00
5de2d2c0e9
Change to use mmcv.utils.collect_env to avoid duplicate codes ( #103 )
2020-12-01 14:24:08 +08:00
a99e6eb10d
init commit
2020-05-21 21:21:43 +08:00
Ezra-Yu
zzc98
Ma Zerun
Ezra-Yu
yingfhu
LXXXXR
chenkai