mmyolo/tests/test_datasets
JosonChan 600343eb08
[Feature] Support YOLOv5 instance segmentation (#735)
* add

* reproduce map

* add typehint and doc

* format code

* replace key

* add ut

* format

* format

* format code

* fix ut

* fix ut

* fix comment

* fix comment

* fix comment

* [WIP][Feature] Support yolov5-Ins training

* fix comment

* change data flow and fix loss_mask compute

* align the data pipeline

* remove albu gt mask key

* support yolov5 ins inference

* fix multi gpu test

* align the post_process with v8

* support training

* support training

* code formatting

* code formatting

* Support pad_param type (#672)

* add half_pad_param

* fix default fast_test

* fix loss weight compute

* fix mask rescale, add segment merge, fix segment2bbox

* fix clip and fix mask init

* code formatting

* code formatting

* code formatting

* code formatting

* [Fix] fix load image from file

* [Add] Add docs and more config

* [Fix] config type and test_formatting

* [Fix] fix yolov5-ins_m packdetinputs

* update

---------

Co-authored-by: Nioolek <379319054@qq.com>
Co-authored-by: Nioolek <40284075+Nioolek@users.noreply.github.com>
Co-authored-by: huanghaian <huanghaian@sensetime.com>
2023-04-27 14:47:52 +08:00
..
test_transforms [Feature] Support YOLOv5 instance segmentation (#735) 2023-04-27 14:47:52 +08:00
__init__.py Upgrade pre commit hooks (#129) 2022-10-13 17:39:52 +08:00
test_utils.py [Enchance] Extending `DetDataPreprocessor` flexibility (#511) 2023-02-06 17:47:22 +08:00
test_yolov5_coco.py [Feature]Add YOLOv5CocoDataset; BatchShapePolicy (#10) 2022-09-18 11:18:31 +08:00
test_yolov5_voc.py [DOC] Add new video link for `custom dataset` (#365) 2022-12-14 10:07:17 +08:00