9 Commits

Author SHA1 Message Date
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
Haian Huang(深度眸)
573ff033e4
[Feature] Support YOLOv7 P5 training (#243)
* support yolov7

* update dev

* update headmodule and convert

* update

* fix pipeline

* fix loss

* fix optimizer parameter groups

* refactor mosaic9

* refactor optim

* refactor loss

* refactor

* refactor

* refactor new

* support yolov7x

* refine

* support tiny

* refactor model

* refactor model

* support yolov7x inference

* support yolov7-tiny inference

* support yolov7-e inference

* refactor

* support yolov7-tiny train

* add docstr

* fix merge error

* fix merge error

* fix merge error

* fix lint

* fix lint

* fix lint

* fix UT

* update

* update
2022-11-21 17:30:19 +08:00
wanghonglie
ab4e7c5158
[Feature] Support YOLOv6 ML model (#265)
* align larger infer

* align large test

* update docstr

* add ut

* Add yolov6m pipeline

* Add yolov6m pipeline

* Improve coding

* update l config

* align m test

* rename config

* support training

* update

* block_cfg -> stage_block_cfg

* update

* Add docstr

* Fix lint

* update convert script

* update

* rename param

* update readme&metafile

* rm duplicate docde

* fix lint

* update

* update

* update

Co-authored-by: HinGwenWoong <peterhuang0323@qq.com>
2022-11-11 19:13:51 +08:00
Nioolek
e4080f8870 [Feature] Support PPYOLOE inference (#145)
* add ppyoloe backbone, neck

* add ppyoloe test

* add docstring

* add ppyoloe m/l/x configfile

* add ppyoloe_coco.py

* rename config

* add typehint

* format repvgg

* format code; add ut

* add doc; use mmdet.DropBlock

* add doc, format code, add non-plus config

* format code

* format code

* format code

* format code

* format code

* format code
2022-11-03 19:03:06 +08:00
JiayuXu
ba4a496562 Upgrade pre commit hooks (#129)
* Fix the version conflict in CircleCI

* Update pre-commit config and add zh-cn version.

* Add codespell ignore list.

* Add codespell ignore list.

* Add check copyright config.
2022-10-13 17:39:52 +08:00
wanghonglie
d705f1c57b [Feature]Support plugin layers for backbone (#75)
* add plugin layer

* add docstring. add config

* clean the config

* add how_to-doc

* add ut

* update

* update

* update lint

* fix yoloxbug

* update

* update

* update doc

* update

* update

* update

* update

* update
2022-09-29 19:09:56 +08:00
huanghaian
66dcc4d3d2 fix import order 2022-09-20 10:57:33 +08:00
hha
cc6ac864c9 update featmap_vis_demo and format code 2022-09-18 17:04:14 +08:00
wanghonglie
85e504fe67
[Feature] Add Backbone;Neck (#2)
* add backbone neck

* update
2022-09-18 10:47:44 +08:00