Commit Graph

5 Commits (8c4d9dc503dc8e327bec8147e8dc97124052f693)

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
Nioolek 8127805dd3
[Feature] Support PPYOLOE training (#259)
* add ppyoloe backbone, neck

* add ppyoloe test

* add docstring

* add ppyoloe m/l/x configfile

* add ppyoloe_coco.py

* rename config

* add typehint

* format code; add ut

* add datapre

* add datapre

* add ppyoloe datapre

* add ppyoloe datapre

* add ppyoloe datapre

* reproduce coco v0.1

* add ut

* add ut, docstring

* fix transforms bug

* use mmdet dfloss

* add non plus model config

* add non plus model config

* fix

* add ut

* produce coco v0.2

* fix config

* fix config

* fix eps and transforms bug

* add ema

* fix resize

* fix transforms.py

* fix transforms.py

* fix transforms.py

* old version

* old version

* old version

* old version

* old version

* old version

* fix stride loss error

* add INTER_LANCZOS4

* fix crop bug

* init commit

* format code

* format code

* bgr transforms.py

* add typehint and doc in transforms.py

* 继承新版yolov6head写法,删除不必要的注释

* fix transforms var name bug

* bbox decode use stridetensor insted of priors

* add headmodule todo

* add ppyoloe README.md

* add ppyoloe README.md

* Update tests/test_datasets/test_transforms/test_transforms.py

Co-authored-by: Range King <RangeKingHZ@gmail.com>

* Update tests/test_datasets/test_transforms/test_transforms.py

Co-authored-by: Range King <RangeKingHZ@gmail.com>

* save ckpt last 10 epochs

* save_best ckpt

* del ppyoloe collate

* change name of ppyoloebatchrandomresize

* add the reason for rewritten PPYOLOEDetDataPreprocessor

* solve ppyoloerandomresize name error

* rm PPYOLOERandomExpand

* rm l1 loss

* rm yolov6 loss_obj

* add difference between yolov6 and ppyoloe

* add reason for rewrite paramscheduler

* change proj init way

* fix error

* rm proj_conv in pth

* format code

* add load_from

* update

* support fast training

* add pretrained model url

* update

* add pretrained model url

* fix error

* add imagenet model convert and use init_cfg to init backbone

* add plus model pretrain model

* add ut

* add ut

* fix ut

* fix withstride bug

* cat in yolov5_collate

* merge

* fix typehint

* update readme

* add reason for gap

* fix log in README.md

* rollback yolov6

* change inherit

* fix ut

* fix ut

Co-authored-by: Range King <RangeKingHZ@gmail.com>
Co-authored-by: hha <1286304229@qq.com>
Co-authored-by: huanghaian <huanghaian@sensetime.com>
2023-01-06 15:54:39 +08:00
Haian Huang(深度眸) 4e8dac5fff
add changelog of v0.2.0 (#323)
* add changelog v0.2.0

* update

* update

* update

* update

* update
2022-12-01 10:54:44 +08:00
Haian Huang(深度眸) bb3aa48689 [Feature] Support RTMDet (#85)
* add cspnext

* add neck


* update

* align s,tiny test

* update

* update

* fix bug of UT

* fix typehint

* fix resize bug

* update s and tiny

* update s and tiny link

* update convert

* add cls config with rtmdet

* update config


* fix ci error

Co-authored-by: wanghonglie <wanghonglie@pjlab.org.cn>
2022-09-29 19:09:56 +08:00
hha cfec074cf6 init commit 2022-09-18 10:11:55 +08:00