Commit Graph

12 Commits (main)

Author SHA1 Message Date
Haian Huang(深度眸) 927e0a46af
Support setting the cache_size_limit parameter and support mmdet 3.0.0 (#707)
* Support setting the cache_size_limit parameter of dynamo in PyTorch 2.0

* update

* update

* fix

* fix

* update

* update

* update version
2023-04-18 10:59:52 +08:00
Range King 69b43e696d
[Feature] Support YOLOv5u inference (#624)
* support yolov5u inference

* minor fix

* Update README.md

* Add all configs

* unify config links
2023-03-07 10:23:13 +08:00
Range King 557f304574
[Fix] Fix typos in the YOLOv8 diagram (#621)
* fix typo

* fix typo in no.17 output
2023-03-06 10:20:07 +08:00
Haian Huang(深度眸) e32838abe1
Add change log of v0.5.0 (#612)
* update

* update

* update

* update

* add configs

* update

* add tta

* update
2023-03-02 17:09:50 +08:00
Haian Huang(深度眸) 1c833eb195
[Feature] Support TTA (#553)
* support TTA

* update note
2023-02-24 21:46:24 +08:00
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
tripleMu e90c369f9d
[Docs] Fix links in YOLOv8 README (#517)
* Update link

* Fix
2023-02-07 10:09:51 +08:00
Range King c3acf42db4
Beautify the YOLOv8 configuration (#516)
* Update yolov5_s-v61_syncbn_8xb16-300e_coco.py

* Update yolov8_s_syncbn_fast_8xb16-500e_coco.py

* Update yolov8_m_syncbn_fast_8xb16-500e_coco.py

* Update yolov8_l_syncbn_fast_8xb16-500e_coco.py

* Update yolov8_s_syncbn_fast_8xb16-500e_coco.py

* Add todo

* Update yolov8_s_syncbn_fast_8xb16-500e_coco.py

* Update transforms.py
2023-02-06 19:39:39 +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
Range King e66991fc6e
[Fix] Fix YOLOv8 graph and a lint issue (#459)
* add missing bg frame

* fix easydeploy lint
2023-01-12 16:00:46 +08:00
Range King 550664d514
[Doc] Add the model structure graph of YOLOv8-P5 (#455)
* add YOLOv8 model structure figure

* Fix the layer name
2023-01-12 10:07:06 +08:00
HinGwenWoong 935d710f79
[Feature] Support YOLOv8 inference (#445)
* Add backbone

* Improve code

* fix

* Add neck

* Add config

* Fix layer param

* Fix layer param

* Add head

* Add head

* Add model converter

* Add model converter

* Add model converter

* Improve code

* update

* update

* align test

* Improve code

* Improve code

* Improve code

* Fix lint

* Fix lint

* Fix lint

* Fix lint

* Improve code

* Improve code

* Improve code

* Add configs

* Improve doc

* Improve doc

* update

* Fix config

* update

* Fix config

* Fix config

* update

* Fix config epoch

* update

* Fix docstr

* Add UT

* Add UT

* Fix doc

* update

* Fix lint

* Fix doc

* Fix config name

* Improve config

* Improve default

* Add docstr

* Improve code

* Improve code

* Drop `bbox_head.head_module.dfl.conv.weight` when convert to mmyolo weight

* Delete useless code

* Add batch_shapes_cfg but not enable it.

* update

Co-authored-by: huanghaian <huanghaian@sensetime.com>
2023-01-11 19:01:06 +08:00