Commit Graph

17 Commits (600343eb083e148e9c8da1301a1559ef105d2e0f)

Author SHA1 Message Date
JosonChan 600343eb08
[Feature] Support YOLOv5 instance segmentation (#735)
* add

* reproduce map

* add typehint and doc

* format code

* replace key

* add ut

* format

* format

* format code

* fix ut

* fix ut

* fix comment

* fix comment

* fix comment

* [WIP][Feature] Support yolov5-Ins training

* fix comment

* change data flow and fix loss_mask compute

* align the data pipeline

* remove albu gt mask key

* support yolov5 ins inference

* fix multi gpu test

* align the post_process with v8

* support training

* support training

* code formatting

* code formatting

* Support pad_param type (#672)

* add half_pad_param

* fix default fast_test

* fix loss weight compute

* fix mask rescale, add segment merge, fix segment2bbox

* fix clip and fix mask init

* code formatting

* code formatting

* code formatting

* code formatting

* [Fix] fix load image from file

* [Add] Add docs and more config

* [Fix] config type and test_formatting

* [Fix] fix yolov5-ins_m packdetinputs

* update

---------

Co-authored-by: Nioolek <379319054@qq.com>
Co-authored-by: Nioolek <40284075+Nioolek@users.noreply.github.com>
Co-authored-by: huanghaian <huanghaian@sensetime.com>
2023-04-27 14:47:52 +08:00
Qingren 9f3adc426f
[FEATURE] Support YOLOv6 3.0 inference (#744)
* [FEATURE] Support YOLOv6 3.0 inference

* add CSPSPPFBottleneck module for YOLOv6 3.0 backbone

* add BiFusion module, YOLOv6RepBiPAFPN module for YOLOv6 3.0 neck

* modify YOLOv6HeadModule to support YOLOv6 3.0 head

* add a yolov6v3 l/m/s/t/n conifgs

* [Fix] Modify YOLOv6 3.0 neck

* Modify YOLOv6RepBiPAFPN

* Add unit tests

* [Fix] Modify configs

* Rename yolov6_v3 configs

* Fix a bug in building BiFusion Module

* Add a checkpoint convert script
2023-04-25 10:22:48 +08:00
tang576225574 dab054875d
optimize the convert_kd_ckpt_to_student.py file (#647)
* correct the scale_factor and pad_param value in LetterResize

* correct the scale_factor and pad_param value in LetterResize

* correct the scale_factor and pad_param value in LetterResize

* tang

* update

* tang

* tang

* fix linting

* tang

* tang

* tang

* add print registers

* add print registers

* add print registers

* add print registers

* add print registers

* tang

* tang v1 finished

* tang

* print registers v1.2

* update

* tang

* update version

* update

* update

* remove exclude_prefix

* swap the val pipeline for visualization

* swap the val pipeline for visualization

* swap the val pipeline for visualization

* swap the val pipeline for visualization

* swap the val pipeline for visualization

* optimize the convert_kd_ckpt_to_student file

* optimize the convert_kd_ckpt_to_student file

* optimize the tools/model_converters/convert_kd_ckpt_to_student.py file

---------

Co-authored-by: hha <1286304229@qq.com>
Co-authored-by: huanghaian <huanghaian@sensetime.com>
2023-03-09 19:07:29 +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
Yanyi Liu 4b996f10a5
Rotated object detection and RTMDet-R (#513)
* init

* add cfg

* update align

* update

* fix regularize box

* fix comment

* update config

* remove ckpt

* update

* make mmrotate optional

* fix doc

* add mmrotate req

* support large_demo with rbbox

* add ut

* update

* add doc v01

* update doc

* fix doc

* update

* update

* update readme

* update comments

* fix

* fix doc

* fix doc

* fix

* update

* update

* fix large

* update doc

* update readme

* fix config

* fix configs

* inprove

* update doc

* update assigner

* update ut

* remove rdsl assigner

* rename aug config

* speedup ut

* add comment

* fix data root

* remove doc

* remove empty folder

* add docs

* rename configs

* fix readme

* fix readme

* fix configs

* revert

* fix name

* fix table

* fix doc link

* fix doc link

* update

* update

* update

* Refactor dota splits

* add shapely

* fix typo

* fix ci

* change

* fix type

* uppdata link

* uppdata link

* add some comment

* update

---------

Co-authored-by: huanghaian <huanghaian@sensetime.com>
2023-03-02 10:27:46 +08:00
whcao 6f38b781bd
[Feature] Add RTMDet distillation cfg (#544)
* add rtm distillation cfg

* rename the cfg file

* use norm connector

* fix cfg

* fix cfg

* support rtm distillation

* fix readme and cfgs

* fix readme

* add docstring

* add links of ckpts and logs

* Update configs/rtmdet/README.md

Co-authored-by: RangiLyu <lyuchqi@gmail.com>

* fix cfgs

* rename stop distillation hook

* rename stop_epoch

* fix cfg

* add model converter

* add metafile and tta results

* fix metafile

* fix readme

* mv distillation/metafile to metafile

---------

Co-authored-by: RangiLyu <lyuchqi@gmail.com>
2023-03-01 16:20:45 +08:00
yechenzhi cbadd3abe4
add yolov7-e6e weight converter (#570)
* add yolov7-e6e weight converter

* add yolov7-e6e weight converter

* add yolov7-e6e weight converter

* fix format

* fix format
2023-02-19 21:48:32 +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
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(深度眸) 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
Haian Huang(深度眸) b95bf63916 [Feature] Support P6 YOLOv5 (#168)
* add yolov5s config

* support train

* fix some comment

* add config

* add link

* UPDATE

* UPDATE

* remove
2022-11-03 19:03:06 +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
Nioolek 5708864d5b [Enhance] YOLOv6 model convert (#165)
* add yolov5 voc training

* fix bug

* fix bug

* fix yolov6 model convert

* update yolov5 modelconvert

* rollback yolov5 model convert
2022-11-03 19:03:06 +08:00
Haian Huang(深度眸) 95f60784a5 [Feature] Support YOLOv7 inference (#149)
* update

* update

* update

* update

* update

* add docstr

* fix comments

* update
2022-11-03 19:03:06 +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
wanghonglie b3d405aa4c
add model convert (#8) 2022-09-18 11:12:08 +08:00