mmyolo/tests/test_models
Nioolek 07ab35222c
[Feature] Support YOLOv8 training (#458)
* add v8 training head

* add training config

* update

* update1

* update all config

* update all config

* update metafile

* add head loss

* update readme

* iou2ciou in tal, add config

* change init layer

* update

* update

* add clip_grad

* update max_aspect_ratio

* add init

* fix aug config

* update forward

* add official v8 loss

* add official v8 loss

* 同步预测方式

* fix loss bug

* delete useless code

* 替换iou,但还有问题

* fix bug

* fix empty

* add UT

* add UT

* update readme

* replace iou_cal

* add use_ciou docstr

* update metafile

* update metafile

* update metafile

Co-authored-by: huanghaian <huanghaian@sensetime.com>
Co-authored-by: hha <1286304229@qq.com>
2023-01-18 14:16:37 +08:00
..
test_backbone [Feature] Support YOLOv8 training (#458) 2023-01-18 14:16:37 +08:00
test_data_preprocessor [Feature] Support PPYOLOE training (#259) 2023-01-06 15:54:39 +08:00
test_dense_heads [Feature] Support YOLOv8 training (#458) 2023-01-18 14:16:37 +08:00
test_detectors [Feature] Support YOLOv8 training (#458) 2023-01-18 14:16:37 +08:00
test_layers
test_necks [Feature] Support YOLOv8 inference (#445) 2023-01-11 19:01:06 +08:00
test_plugins [Feature] Add attention module of CBAM (#246) 2022-11-10 11:12:31 +08:00
test_task_modules [Fxi] Fix ci bug of BatchATSSAssigner (#232) 2022-11-03 19:03:06 +08:00
test_utils
__init__.py