Commit Graph

2 Commits (887d3ddfdc13d7f78b7a7394a17da65ca9c17794)

Author SHA1 Message Date
Nioolek 75fc8fc2a3
[Feature] YOLOv8 supports using mask annotation to optimize bbox (#484)
* add cfg

* add copypaste

* add todo

* 在mosaic和mixup中处理gt_masks,改config

* fix cat bug

* add finetune box in affine

* add repr

* del albu config in l

* add doc

* add config

* format code

* fix loadmask

* addconfig,fix mask

* fix loadann

* fix tra

* update LoadAnnotations

* update

* support mask

* fix error

* fix error

* fix config and no maskrefine bug

* fix

* fix

* update config

* format code

* beauty config

* add yolov5 config and readme

* beauty yolov5 config

* add ut

* fix ut. bitmap 2 poly

* fix ut and add mix transform ut.

* fix bool

* fix loadann

* rollback yolov5

* rollback yolov5

* format

* 提高速度

* update

---------

Co-authored-by: huanghaian <huanghaian@sensetime.com>
2023-02-20 11:11:13 +08:00
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