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
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
bac181f393
[Feature] Support Multi-task. ( #1229 )
...
* unit test for multi_task_head
* [Feature] MultiTaskHead (#628 , #481 )
* [Fix] lint for multi_task_head
* [Feature] Add `MultiTaskDataset` to support multi-task training.
* Update MultiTaskClsHead
* Update docs
* [CI] Add test mim CI. (#879 )
* [Fix] Remove duplicated wide-resnet metafile.
* [Feature] Support MPS device. (#894 )
* [Feature] Support MPS device.
* Add `auto_select_device`
* Add unit tests
* [Fix] Fix Albu crash bug. (#918 )
* Fix albu BUG: using albu will cause the label from array(x) to array([x]) and crash the trainning
* Fix common
* Using copy incase potential bug in multi-label tasks
* Improve coding
* Improve code logic
* Add unit test
* Fix typo
* Fix yapf
* Bump version to 0.23.2. (#937 )
* [Improve] Use `forward_dummy` to calculate FLOPS. (#953 )
* Update README
* [Docs] Fix typo for wrong reference. (#1036 )
* [Doc] Fix typo in tutorial 2 (#1043 )
* [Docs] Fix a typo in ImageClassifier (#1050 )
* add mask to loss
* add another pipeline
* adpat the pipeline if there is no mask
* switch mask and task
* first version of multi data smaple
* fix problem with attribut by getattr
* rm img_label suffix, fix 'LabelData' object has no attribute 'gt_label'
* training without evaluation
* first version work
* add others metrics
* delete evaluation from dataset
* fix linter
* fix linter
* multi metrics
* first version of test
* change evaluate metric
* Update tests/test_models/test_heads.py
Co-authored-by: Colle <piercus@users.noreply.github.com>
* Update tests/test_models/test_heads.py
Co-authored-by: Colle <piercus@users.noreply.github.com>
* add tests
* add test for multidatasample
* create a generic test
* create a generic test
* create a generic test
* change multi data sample
* correct test
* test
* add new test
* add test for dataset
* correct test
* correct test
* correct test
* correct test
* fix : #5
* run yapf
* fix linter
* fix linter
* fix linter
* fix isort
* fix isort
* fix docformmater
* fix docformmater
* fix linter
* fix linter
* fix data sample
* Update mmcls/structures/multi_task_data_sample.py
Co-authored-by: Colle <piercus@users.noreply.github.com>
* Update mmcls/structures/multi_task_data_sample.py
Co-authored-by: Colle <piercus@users.noreply.github.com>
* Update mmcls/structures/multi_task_data_sample.py
Co-authored-by: Colle <piercus@users.noreply.github.com>
* Update mmcls/structures/multi_task_data_sample.py
Co-authored-by: Colle <piercus@users.noreply.github.com>
* Update mmcls/structures/multi_task_data_sample.py
Co-authored-by: Colle <piercus@users.noreply.github.com>
* Update mmcls/structures/multi_task_data_sample.py
Co-authored-by: Colle <piercus@users.noreply.github.com>
* Update tests/test_structures/test_datasample.py
Co-authored-by: Colle <piercus@users.noreply.github.com>
* Update mmcls/structures/multi_task_data_sample.py
Co-authored-by: Colle <piercus@users.noreply.github.com>
* Update tests/test_structures/test_datasample.py
Co-authored-by: Colle <piercus@users.noreply.github.com>
* Update tests/test_structures/test_datasample.py
Co-authored-by: Colle <piercus@users.noreply.github.com>
* update data sample
* update head
* update head
* update multi data sample
* fix linter
* fix linter
* fix linter
* fix linter
* fix linter
* fix linter
* update head
* fix problem we don't set pred or gt
* fix problem we don't set pred or gt
* fix problem we don't set pred or gt
* fix linter
* fix : #2
* fix : linter
* update multi head
* fix linter
* fix linter
* update data sample
* update data sample
* fix ; linter
* update test
* test pipeline
* update pipeline
* update test
* update dataset
* update dataset
* fix linter
* fix linter
* update formatting
* add test for multi-task-eval
* update formatting
* fix linter
* update test
* update
* add test
* update metrics
* update metrics
* add doc for functions
* fix linter
* training for multitask 1.x
* fix linter
* run flake8
* run linter
* update test
* add mask in evaluation
* update metric doc
* update metric doc
* Update mmcls/evaluation/metrics/multi_task.py
Co-authored-by: Colle <piercus@users.noreply.github.com>
* Update mmcls/evaluation/metrics/multi_task.py
Co-authored-by: Colle <piercus@users.noreply.github.com>
* Update mmcls/evaluation/metrics/multi_task.py
Co-authored-by: Colle <piercus@users.noreply.github.com>
* Update mmcls/evaluation/metrics/multi_task.py
Co-authored-by: Colle <piercus@users.noreply.github.com>
* Update mmcls/evaluation/metrics/multi_task.py
Co-authored-by: Colle <piercus@users.noreply.github.com>
* Update mmcls/evaluation/metrics/multi_task.py
Co-authored-by: Colle <piercus@users.noreply.github.com>
* update metric doc
* update metric doc
* Fix cannot import name MultiTaskDataSample
* fix test_datasets
* fix test_datasets
* fix linter
* add an example of multitask
* change name of configs dataset
* Refactor the multi-task support
* correct test and metric
* add test to multidatasample
* add test to multidatasample
* correct test
* correct metrics and clshead
* Update mmcls/models/heads/cls_head.py
Co-authored-by: Colle <piercus@users.noreply.github.com>
* update cls_head.py documentation
* lint
* lint
* fix: lint
* fix linter
* add eval mask
* fix documentation
* fix: single_label.py back to 1.x
* Update mmcls/models/heads/multi_task_head.py
Co-authored-by: Ma Zerun <mzr1996@163.com>
* Remove multi-task configs.
Co-authored-by: mzr1996 <mzr1996@163.com>
Co-authored-by: HinGwenWoong <peterhuang0323@qq.com>
Co-authored-by: Ming-Hsuan-Tu <alec.tu@acer.com>
Co-authored-by: Lei Lei <18294546+Crescent-Saturn@users.noreply.github.com>
Co-authored-by: WRH <12756472+wangruohui@users.noreply.github.com>
Co-authored-by: marouaneamz <maroineamil99@gmail.com>
Co-authored-by: marouane amzil <53240092+marouaneamz@users.noreply.github.com>
2022-12-30 10:36:00 +08:00
c127c474b9
[Feature] Support getting model from the name defined in the model-index file. ( #1236 )
...
* [Feature] Support getting model from the name defined in the model-index file.
* Add unit tests.
* Prevent import `timm` if the `TIMMBackbone` is not used.
* Fix Windows CI.
* Move `init_model` to `mmcls.apis.hub`, and support pass nn.Module to all
model components.
* Fix requirements
* Rename `hub.py` to `model.py` and add unit tests.
2022-12-06 17:00:22 +08:00
d6fa480915
[Refactor] Add VOC dataset and mutil_label dataset
2022-07-18 11:11:13 +08:00
27e685fe10
[Feature] Add `BaseDataset`, `CustomDataset`, `ImageNet` and `ImageNet21k`
2022-07-18 10:53:27 +08:00
28e3bc8fd9
[CI] Add Windows CI. ( #708 )
...
* Update workflow to add Windows CI
* Add some optional requirements to requirements/optional.txt
* Update setup.py
* Update PyTorch version in Windows CI
* Update CI and requirement
* Replace `tempfile.NamedTemporaryFile` to avoid unit test error on
Windows
* Update tests
* Update CI
* Add OpenCV installation
* Update CI
2022-02-28 10:06:08 +08:00
fcd57913ae
[Enhance] Upgrade isort pre-commit hooks. ( #687 )
...
* Use new version flake8 and isort hooks
* Fix missing copyright
2022-02-17 02:17:20 +08:00
16864c7495
[Feature] Support features_only in TIMMBackbone ( #668 )
...
* Support features_only in TIMMBackbone
based on https://github.com/open-mmlab/mmsegmentation/pull/998
* update test for mmdet
* fix unit test for build_without_timm
* Update docstring
Co-authored-by: mzr1996 <mzr1996@163.com>
2022-01-25 11:22:23 +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
2ce5825ef1
[Feature] Support ImageNet21k dataset. ( #461 )
...
* add imagnet21k
* Update unit test
* update imaenet21k
* use slots
* use slots
* rename Data_item to ImageInfo
* add unit tests
* Update unit tests
* rm some print
* update unit tests
* fix lint
* remove default value of pipeline
2021-10-28 15:22:08 +08:00
a8f4f82b8e
[Enhance] Improve downstream repositories compatibility ( #421 )
...
* Defaults to return tuple in all backbones.
* Compat downstream of swin transformer.
* Support tuple input for multi label head and stacked head.
* Fix backbone unit tests for tuple output.
* Add downstream inference unit tests for mmdet.
* Update gitignore
* Add unit tests for `return_tuple` option
* Add unit tests for head input tuple.
* Add warning in `simple_test`
* Add TIMMBackbone return tuple.
* Modify timm backbone unit test.
2021-09-08 10:38:57 +08:00
06f415c28a
support to test mmdet inference with mmcls backbone ( #343 )
...
* support to test mmdet inference with mmcls backbone
* update config
* update
* add mmdet dependency in tests
2021-07-06 14:34:12 +08:00
f6260b33bf
Add RandomCrop, RandomResizedCrop, RandomGrayscale, impad
2020-07-12 00:06:55 +08:00
ba789da5c8
Add commonly used datasets
2020-07-01 16:09:06 +08:00
Ma Zerun
Yixiao Fang
Colle
Ezra-Yu
Yosuke Shinya
Wenwei Zhang
lixiaojie
yanglei