Commit Graph

75 Commits (cdc885550acd73b826b45caf066e482024194a70)

Author SHA1 Message Date
wanghonglie 980e908618 [Feature] Support YOLOv6 training (#183)
* init v6 loss

* init v6s train

* Add train pipeline

* Add lr scheduler

* update

* update

* update

* update

* update

* update

* update

* update

* update

* fix detach bug

* fix detach bug

* update

* Add stop aug hook

* Add save best ckpt

* update

* Add PipelineSwitchHook

* Fix train pipeline stage 2

* update

* Fix train pipeline

* update

* fix stage2 randomaffine bug

update

update

clean

clean

* update letterResize param

* add v6affine config

* add v6 randomaffine

* update v6 config

* update

* update

* update

* update

* update config param

* update

* update

* refactor iou loss % rm v6affine

* update

* rm dfl

* add v6 300 epoch config

* Factor batch atss assigner

* Format code

* Format code

* Roll back

* Refactor dist_calculator

* Refactor select_candidates_in_gts

* Refactor select_highest_overlaps

* Refactor iou_calculator

* Refactor all code

* Improve docstr

* Improve code

* clean config

* add nano tiny config

* pre-commit

* Refactor

* Improve code

* Improve naming and link

* Add UT

* pre commit

* Add UT

* Add UT

* Improve code, using mmdet.BboxOverlaps2D for all iou calculation

* Improve code, using mmdet.BboxOverlaps2D for all iou calculation

* Improve code

* pre commit

* pre commit

* Add UT

* fix config

* pre commit

* Improve code

* Improve code

* Improve code

* Improve code

* [Refactor] YOLOv6 BatchATSSAssigner (#179)

* Factor batch atss assigner

* Format code

* Format code

* Roll back

* Refactor dist_calculator

* Refactor select_candidates_in_gts

* Refactor select_highest_overlaps

* Refactor iou_calculator

* Refactor all code

* Improve docstr

* Improve code

* Improve code

* Improve naming and link

* Add UT

* pre commit

* Add UT

* Add UT

* Improve code, using mmdet.BboxOverlaps2D for all iou calculation

* Improve code, using mmdet.BboxOverlaps2D for all iou calculation

* Improve code

* pre commit

* Fix conflicts

* Improve code

* Improve code

* Improve code

* Improve code

* Improve code

* Improve code

* add utils.py, order the input param

* Improve docstr

* Fix lint

* Improve param mapping

* Improve param mapping

* Improve naming

* assigner return dict

* update

* update config

* update config

* Fix

* Fix UT

* Improve UT

* Improve naming

* Improve coding

* pre commit

* pre commit

* pre commit

* Fix ci

* Improve naming

* Improve coding

* Fix training iou calculate error

* Improve naming

* Improve naming

* Improve type hint

* fix lint

* fix conflicts

* fix UT

* Improve type hint

* Improve naming

* Improve coding

* Improve coding

* Fix UT

* Refactor SIoU

* Pre commit

* Fix

* Improve ciou

* Improve ciou

* refactor varifocal

* Improve ciou

* Improve ciou

* Improve siou

* Improve type hint

* Improve siou

* Improve siou

* Fix lint

* refactor varifocal

* fix iou bug

* fix siou and loss_cls bug

* update

* update

* add scope

* update

* update

* Improve func `gt_instances_preprocess`

* support deploy mode

* Improve func `gt_instances_preprocess`

* Improve func `gt_instances_preprocess`

* Improve func `gt_instances_preprocess`

* Improve func `bbox_overlaps`

* Improve coding

* Improve bbox_overlaps

* Delete useless code

* add yolov6 deploy mode hook

* fix lint

* Add common attributes to reduce calculation

* Improve code

* Improve code

* Fix bug

* Fix bug

* update

* add readme

* update readme

* update readme url

Co-authored-by: HinGwenWoong <peterhuang0323@qq.com>
2022-11-03 19:03:06 +08:00
tripleMu 177eb4ea13 [Feature] Add static shape model config (#214)
* Add static model config

* Rename config name

* Rename file

* Add yolov6s static config
2022-11-03 19:03:06 +08:00
tripleMu 275beec782 [Feature] Support YOLOv5 YOLOv6 YOLOX Deploy in mmdeploy (#199)
* Support YOLOv5 YOLOv6 YOLOX Deploy in mmdeploy

* Fix lint

* Rename _class to detector_type

* Add some common

* fix lint

Co-authored-by: huanghaian <huanghaian@sensetime.com>
2022-11-03 19:03:06 +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
hanrui1sensetime 5874e4108e [Feature] support mmyolo deployment (#79)
* support mmyolo deployment

* mv deploy place

* remove unused configs

* add deploy code

* fix new register

* fix comments

* fix dependent codebase register

* remove unused initialize

* refact deploy config

* credit return to triplemu

* Add yolov5 head rewrite

* refactor deploy

* refactor deploy

* Add yolov5 head rewrite

* fix configs

* refact config

* fix comment

* sync name after mmdeploy 1088

* fix mmyolo

* fix yapf

* fix deploy config

* try to fix flake8 importlib-metadata

* add mmyolo models ut

* add deploy uts

* add deploy uts

* fix trt dynamic error

* fix multi-batch for dynamic batch value

* fix mode

* fix lint

* sync model.py

* add ci for deploy test

* fix ci

* fix ci

* fix ci

* extract script to command for fixing CI

* fix cmake for CI

* sudo ln

* move ort position

* remove unused sdk compile

* cd mmdeploy

* simplify build

* add missing make

* change order

* add -v

* add setuptools

* get locate

* get locate

* upgrade torch

* change torchvision  version

* fix config

* fix ci

* fix ci

* fix lint

Co-authored-by: tripleMu <gpu@163.com>
Co-authored-by: RunningLeon <mnsheng@yeah.net>
2022-11-03 19:03:06 +08:00
bin ze 15e0cfa4b5 [Fix] Update metafile for RTMDet (#188)
* update metafile

* update metafile

* update metafile
2022-11-03 19:03:06 +08:00
Haian Huang(深度眸) b18d5cf586 [Fix] Remove Albu version limit (#187)
* remove albu version limit

* fix lint
2022-11-03 19:03:06 +08:00
Nioolek 45e23ae4c8 Fix voc training error (#178)
* add doc; add model log url

* add doc; add model log url
2022-11-03 19:03:06 +08:00
Nioolek 5ef3606482 [Feature] Support VOC Dataset in YOLOv5 (#134)
* add yolov5 voc training

* fix mosaic bug

* fix mosaic bug and temp config

* fix mosaic bug

* update config

* support training on voc dataset

* format code

* format code

* Optimize Code. Change `RandomTransform` to `OneOf`

* Change `OneOf` to `mmcv.RandomChoice`

* fix yolov5coco dataset

* fix yolov5coco dataset

* fix bug, format code

* format config

* format code

* add yolov5 voc training

* rebase

* fix mosaic bug

* update config

* support training on voc dataset

* format code

* format code

* Optimize Code. Change `RandomTransform` to `OneOf`

* Change `OneOf` to `mmcv.RandomChoice`

* fix yolov5coco dataset

* fix yolov5coco dataset

* fix bug, format code

* format code

* add yolov5 voc training

* fix mosaic bug and temp config

* fix mosaic bug

* update config

* support training on voc dataset

* format code

* format code

* Optimize Code. Change `RandomTransform` to `OneOf`

* Change `OneOf` to `mmcv.RandomChoice`

* fix yolov5coco dataset

* fix yolov5coco dataset

* fix bug, format code

* format code

* add yolov5 voc training

* rebase

* fix mosaic bug

* update config

* support training on voc dataset

* format code

* format code

* Optimize Code. Change `RandomTransform` to `OneOf`

* Change `OneOf` to `mmcv.RandomChoice`

* fix yolov5coco dataset

* fix yolov5coco dataset

* fix bug, format code

* format code

* format code

* fix lint

* add unittest

* add auto loss_weight

* add doc; add model log url

* add doc; add model log url

* add doc; add model log url
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(深度眸) 3ea6edfeac Fix visualization bug in test_pipeline (#166) 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
q.yao 59f3d30fd2
Fix checkpoint link (#63)
* fix checkpoint link

* format docs
2022-09-22 10:48:31 +08:00
huanghaian 2aad4fde4f Fix the finetune command in zh_CN yolov5_tutorial.md 2022-09-21 19:07:33 +08:00
Haian Huang(深度眸) c068dcdacc
Add Circleci project (#57)
* Add .circleci/config.yml

* Add circleci

* Add circleci

* Updated config.yml

* format code

* format code

* fix
2022-09-21 16:52:34 +08:00
HinGwenWoong 26165b4ff2
[Add & Fix] Add docstr and typehint and fix config `data_root` (#53)
* [Add] Add docstr

* [Add] Add docstr
2022-09-21 15:43:48 +08:00
huanghaian a93cf7c3e4 Update albumentations requirements 2022-09-21 12:14:04 +08:00
huanghaian c683d1f94d Fix some config and add description 2022-09-21 10:12:00 +08:00
Range King 98a0f6b9c9
[Docs] RE: Fix typos and add citations (#45)
* update logo (#37)

* [Docs] Update images with new logo (#38)

* Update model_design.md

Update logo

* Revert "Update model_design.md"

This reverts commit 207f529522.

* Update yolov5_description.md

Update logo

* Update model_design.md

Update logo

* Update model_design.md

Update logo

* Update README.md

Update logo

* Update README_zh-CN.md

Update logo

* Add GIF Link (#40)

* Update README.md

* Update README_zh-CN.md

* Update README.md

* Update README_zh-CN.md

* Fix typos in config.md

* Add citation of yolov6

* Add citation of yolov5

* Fix links in README.md

* Fix typos in data_flow.md in en

Co-authored-by: wanghonglie <wanghonglie@zju.edu.cn>
Co-authored-by: MingJian.L <45811724+matrixgame2018@users.noreply.github.com>
2022-09-20 10:29:01 +08:00
huanghaian 98712bb898 upload yolox_tiny and yolov5_n model 2022-09-20 10:07:57 +08:00
huanghaian 8b13aed8c3 update model 2022-09-19 13:01:47 +08:00
古月闻星 bc4df98973
[Fix] replace pre to per (#24)
* [Fix] change pre to per

* [Fix] batch_size_pre_gpu to batch_size_per_gpu
2022-09-18 15:40:29 +08:00
古月闻星 d3fed9f2cd
[Fix] fix the num_workers setting (#25) 2022-09-18 15:38:36 +08:00
hha cfec074cf6 init commit 2022-09-18 10:11:55 +08:00