6b52f81c07
[FEATURE] Support YOLOv6 v3.0 face detection ( #812 )
...
* support the usage of WIDERFace dataset
* add YOLOv6FaceHead
* add YOLOv6 face detection configs
* add a checkpoint convertion script
* add a face visualizer
* fix a bug of YOLOv6CSPBep initialization
2023-08-23 15:17:25 +08:00
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
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
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
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
cdc885550a
[Feature] Add model zoo ( #607 )
...
* Update model_zoo.md
* fix
* update
* update
* update
2023-03-01 16:34:17 +08:00
1c833eb195
[Feature] Support TTA ( #553 )
...
* support TTA
* update note
2023-02-24 21:46:24 +08:00
d8b1353691
[Improve] Beautify YOLOv6 all configs ( #539 )
...
* Beauty YOLOv6 config
* Beauty YOLOv6 config
* Beauty config
* Beauty config
2023-02-10 10:13:47 +08:00
a631370cb3
Update model graphs ( #442 )
2023-01-07 17:44:19 +08:00
884330108d
[BC Breaking] Keep the same input shape order as mmdet 3.0.0rc5 ( #436 )
...
* Keep the same input shape order as mmdet
* add desc
* update
* update
2023-01-06 19:00:30 +08:00
2bf502b13c
[Improvement] Improve doc `custom_dataset` ( #336 )
...
* Improve doc
* Improve doc
* Add config
* Add config
* Add config
* Improve doc
* Improve doc
* Improve doc
* Improve doc
* Improve doc index
* Improve doc
* Improve doc
* Improve doc
* Improve doc
* Improve doc
* Improve doc
* Improve doc
* Improve doc
* Improve doc
* Improve doc
* Improve doc
* Improve doc
* Improve doc
* Improve doc
* Improve doc
* Improve doc
* Improve doc
* Improve doc
* Improve doc
* Improve doc
* Improve doc
* Improve doc
* Improve doc
* Improve doc
* Improve doc
* Add deploy demo
* Improve doc
* Fix lint
* Improve doc
* Improve doc
* Improve doc
* Improve doc
* Improve doc
* Improve code
* Improve code
* Improve code
* Improve code
* Improve doc
* Improve doc
* Improve doc
* Improve doc
* Improve doc
* Improve doc
* Improve doc
* Improve doc
* Improve doc
* Improve doc
* Improve doc
* Improve doc
* Improve doc
* Improve doc
* Improve doc
* Improve doc
* Improve doc
* Improve doc
* Improve doc
* Improve doc
* Improve doc
* Improve doc
* Improve doc
* Improve doc
* Improve doc
* Improve doc
* Improve doc
* Improve doc
* Improve doc
* Improve doc
* Improve doc
* Improve doc
* Improve doc
* Improve doc
* Improve doc
* Improve doc
* Improve doc
* Improve doc
* Improve doc
* Improve doc
* Improve doc
* Improve doc
* Improve doc
* Improve doc
* Improve doc
2022-12-12 10:16:03 +08:00
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
f045ecf213
[Feature] Add YOLOv6 metafile ( #227 )
...
* add v6 meta file
* update v6s
2022-11-03 19:03:06 +08:00
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
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#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
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
cfec074cf6
init commit
2022-09-18 10:11:55 +08:00
Qingren
Haian Huang(深度眸)
Range King
HinGwenWoong
wanghonglie
古月闻星