mmyolo/tests/test_models/test_backbone
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
..
__init__.py Upgrade pre commit hooks (#129) 2022-10-13 17:39:52 +08:00
test_csp_darknet.py [Feature] Support YOLOv8 training (#458) 2023-01-18 14:16:37 +08:00
test_csp_resnet.py [Feature] Support PPYOLOE inference (#145) 2022-11-03 19:03:06 +08:00
test_efficient_rep.py [Feature] Support YOLOv6 ML model (#265) 2022-11-11 19:13:51 +08:00
test_yolov7_backbone.py [Feature] Support YOLOv7 P5 training (#243) 2022-11-21 17:30:19 +08:00
utils.py fix import order 2022-09-20 10:57:33 +08:00