mmyolo/tools/model_converters
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
..
convert_kd_ckpt_to_student.py optimize the convert_kd_ckpt_to_student.py file (#647) 2023-03-09 19:07:29 +08:00
ppyoloe_to_mmyolo.py
rtmdet_to_mmyolo.py Rotated object detection and RTMDet-R (#513) 2023-03-02 10:27:46 +08:00
yolov5_to_mmyolo.py [Feature] Support YOLOv5 instance segmentation (#735) 2023-04-27 14:47:52 +08:00
yolov5u_to_mmyolo.py [Feature] Support YOLOv5u inference (#624) 2023-03-07 10:23:13 +08:00
yolov6_to_mmyolo.py
yolov6_v3_to_mmyolo.py [FEATURE] Support YOLOv6 3.0 inference (#744) 2023-04-25 10:22:48 +08:00
yolov7_to_mmyolo.py
yolov8_to_mmyolo.py [Feature] Support YOLOv5 instance segmentation (#735) 2023-04-27 14:47:52 +08:00
yolox_to_mmyolo.py