ab5665e840
update base v6 config name ( #398 )
2022-12-27 11:01:17 +08:00
b012154c10
CodeCamp #1519 [Docs] add yolox structure description ( #402 )
...
* add yolox structure description
* Update configs/yolox/README.md
Co-authored-by: Range King <RangeKingHZ@gmail.com>
Co-authored-by: Range King <RangeKingHZ@gmail.com>
2022-12-26 14:15:23 +08:00
8cf8ab9ca3
[Feature] Add `crowdhuman` dataset ( #368 )
...
* add crowdhuman dataset
* update
* update anhchor
* add pretrain link
* fix name
* remove batch_shapes_cfg
* add readme
* update readme
* update readme
2022-12-15 16:37:09 +08:00
dcf73091e5
add mmyolo regression test yml ( #359 )
2022-12-12 10:27:21 +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
9985e0bc08
Update yolov7_l_syncbn_fast_8x16b-300e_coco.py ( #333 )
...
change:model->bbox_head->head_module->dict(num_classes=num_classes)
2022-12-02 16:31:27 +08:00
b8afd0055d
[DOC] Add custom dataset guide ( #306 )
...
* Add doc
* Add doc
* Add doc
* Add 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 config
* 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 config
* Improve doc config
* Improve doc
* Improve doc
* Improve doc
* Improve doc
* Improve doc
* Improve doc
* Improve doc
* Improve doc
* Improve doc
* Improve doc
2022-11-30 18:58:43 +08:00
e85ddeac0d
[Feature] Support YOLOv7 P6 training ( #310 )
...
* support p6 train
* fix bug
* add readme
* add link
2022-11-30 18:45:08 +08:00
4a8699d6fe
[Fix] Enable switch to deploy when create pytorch model in deployment. ( #324 )
...
* support switch_to_deploy when deploy
* fix docformatter
* fix README for yolox shape
2022-11-29 18:23:12 +08:00
68c9fd4745
[Feature] Support YOLOv5 deployment at RV1126 device ( #262 )
...
* add yolov5 head rewrite and rknn config
* type hint
* remove useless
* refactor config
2022-11-29 16:54:08 +08:00
15224b7816
[Fix] Fix mmyolo int8 deploy config ( #315 )
...
* fix mmyolo int8 deploy config
* fix lint
2022-11-26 09:26:40 +08:00
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
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
71e5c793e5
[Doc] Fix typo ( #270 )
...
* fix typo
* fix ci
2022-11-10 11:12:31 +08:00
8fc8066b1b
Add changelog of v0.1.2 ( #226 )
...
* Add changelog of v0.1.2
* update version
* fix comments
* fix comments
* update
* update version
* update version
2022-11-03 19:03:06 +08:00
f045ecf213
[Feature] Add YOLOv6 metafile ( #227 )
...
* add v6 meta file
* update v6s
2022-11-03 19:03:06 +08:00
9c250571ca
[Fix] `concatdatasets` for voc train ( #228 )
...
* add yolov5 voc training
* format code
* [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
* [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>
* [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>
* fix vocdatasets
* fix vocdatasets
Co-authored-by: hanrui1sensetime <83800577+hanrui1sensetime@users.noreply.github.com>
Co-authored-by: tripleMu <gpu@163.com>
Co-authored-by: RunningLeon <mnsheng@yeah.net>
Co-authored-by: huanghaian <huanghaian@sensetime.com>
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
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
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
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
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
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
15e0cfa4b5
[Fix] Update metafile for RTMDet ( #188 )
...
* update metafile
* update metafile
* update metafile
2022-11-03 19:03:06 +08:00
b18d5cf586
[Fix] Remove Albu version limit ( #187 )
...
* remove albu version limit
* fix lint
2022-11-03 19:03:06 +08:00
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
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
95f60784a5
[Feature] Support YOLOv7 inference ( #149 )
...
* update
* update
* update
* update
* update
* add docstr
* fix comments
* update
2022-11-03 19:03:06 +08:00
3ea6edfeac
Fix visualization bug in test_pipeline ( #166 )
2022-11-03 19:03:06 +08:00
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
59f3d30fd2
Fix checkpoint link ( #63 )
...
* fix checkpoint link
* format docs
2022-09-22 10:48:31 +08:00
2aad4fde4f
Fix the finetune command in zh_CN yolov5_tutorial.md
2022-09-21 19:07:33 +08:00
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
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
a93cf7c3e4
Update albumentations requirements
2022-09-21 12:14:04 +08:00
c683d1f94d
Fix some config and add description
2022-09-21 10:12:00 +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
98712bb898
upload yolox_tiny and yolov5_n model
2022-09-20 10:07:57 +08:00
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
cfec074cf6
init commit
2022-09-18 10:11:55 +08:00
Xin Li
Youfu
Haian Huang(深度眸)
RunningLeon
HinGwenWoong
CaiLiYuan
hanrui1sensetime
AllentDan
Haian Huang(深度眸)
wanghonglie
Xiaolin Wang
Nioolek
tripleMu
bin ze
q.yao
huanghaian
Range King
古月闻星