f54e5603fd
[Enchance] Extending `DetDataPreprocessor` flexibility ( #511 )
...
* Extending DetDataPreprocessor flexibility
* update
* add cast_data
2023-02-06 17:47:22 +08:00
1dee9eed6e
[Docs] How to specify specific GPU training and inference ( #503 )
...
* 中文版指定GPU训练
* 删去不必要文件
* typo
* rebase dev
* add test example
* add english version
* fix format
* english typo
* Update docs/zh_cn/advanced_guides/how_to.md
Co-authored-by: Range King <RangeKingHZ@gmail.com>
* Update docs/zh_cn/advanced_guides/how_to.md
Co-authored-by: Range King <RangeKingHZ@gmail.com>
---------
Co-authored-by: Range King <RangeKingHZ@gmail.com>
2023-02-06 10:12:04 +08:00
6acde82ec8
Fix typo for assigner visualization ( #509 )
2023-02-04 13:46:14 +08:00
8237f19bf8
Added the prompt to submit the pull request ( #508 )
2023-02-04 11:49:58 +08:00
5f2b08cf5a
add model structure graph of yolov7 in readme ( #504 )
2023-02-03 20:01:05 +08:00
79f0aae555
Beautify the YOLOv5 configuration ( #501 )
...
* refactor_config
2023-02-03 14:28:35 +08:00
74558aa2f7
Use official isort
2023-01-31 14:50:10 +08:00
77d9ad0c6b
[CI]: fix isort version issue in lint ( #492 )
...
* fix lint
* fix-lint
2023-01-29 17:48:54 +08:00
e62c8c4593
Bump version to 0.4.0
...
Bump version to 0.4.0
2023-01-18 19:54:21 +08:00
d8b13538dc
Add v0.4.0 Changelog ( #477 )
...
* Add v0.4.0 changelog
* update
* fix
* fix
* fix
* update
* update
* add url
2023-01-18 19:43:27 +08:00
ac27f900e3
[Docs] Refine README ( #482 )
...
* add emoji in README
* update
* update
* update1
* update1
* Add banner
* minor fix
* adjust position
* add toc
* update zh_CN
* update zh_CN
* test
* update
* test
* test
* update
* update
* update
* update
* add banner in README_zh-CN
* replace emoji
* minor fix
* remove some videos
* Update README_zh-CN.md
Co-authored-by: huanghaian <huanghaian@sensetime.com>
2023-01-18 19:38:34 +08:00
51d30a2837
[Docs] Update `article.md` and `model_zoo.md` ( #481 )
...
* Update article.md
* Update model_zoo.md
* Update model_zoo.md
2023-01-18 15:23:31 +08:00
6bdfc0da26
[Feature] Add a tool for scheduler visualization ( #479 )
...
* Create vis_scheduler.py
* add docs
* Update .gitignore
* Update useful_tools.md
* add source link
* Add TODO
2023-01-18 15:03:02 +08:00
6f478e1ae0
[Bug] Support RTMDet P6 model ( #480 )
2023-01-18 15:02:39 +08:00
da62ae7283
[Doc] Add PPYOLOE model structure ( #471 )
...
* Update README.md
Add PPYOLOE model structure
* Update README.md
* Update README.md
* Update README.md
* Update README.md
* Update README.md
* Update README.md
2023-01-18 14:32:20 +08:00
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
8eefed39ef
[Doc] Fix the installation step on get_started.md ( #474 )
2023-01-17 15:15:24 +08:00
c32ca4549f
[DOC] Add YOLOv8 docs ( #461 )
...
* Add YOLOv8 docs
* update
* update
* fix some error
* update en
* fix comment
* update
2023-01-17 15:12:02 +08:00
feef44da23
[Fix] Fix requirement about only ONNXRuntime without TensorRT ( #469 )
...
* Fix requirement about only onnxruntime without tensorrt
* Update projects/easydeploy/model/backendwrapper.py
Co-authored-by: HinGwenWoong <peterhuang0323@qq.com>
* Update projects/easydeploy/model/backendwrapper.py
Co-authored-by: HinGwenWoong <peterhuang0323@qq.com>
* Update projects/easydeploy/model/backendwrapper.py
Co-authored-by: HinGwenWoong <peterhuang0323@qq.com>
* Update projects/easydeploy/model/backendwrapper.py
Co-authored-by: HinGwenWoong <peterhuang0323@qq.com>
Co-authored-by: HinGwenWoong <peterhuang0323@qq.com>
2023-01-16 19:36:49 +08:00
e66991fc6e
[Fix] Fix YOLOv8 graph and a lint issue ( #459 )
...
* add missing bg frame
* fix easydeploy lint
2023-01-12 16:00:46 +08:00
8e39252259
Support YOLOv8 deploy ( #456 )
2023-01-12 10:07:44 +08:00
550664d514
[Doc] Add the model structure graph of YOLOv8-P5 ( #455 )
...
* add YOLOv8 model structure figure
* Fix the layer name
2023-01-12 10:07:06 +08:00
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
f1855ca618
[Feature] Add new YOLOX bbox coder for TensorRT-8 and fix TensorRT-7 bug ( #450 )
...
* Add yolox bbox decoder for tensorrt backend
* Fix tensorrt7+ bug
2023-01-11 17:10:04 +08:00
fd0f07b8ac
[Fix] Fix import error in `optimize_anchors.py` ( #452 )
...
* [Fix] Fix bugs in the output annotation file when is set.
* Update labelme2coco.py
* Update labelme2coco.py
* Improve code
* [Fix]Fix import error in optimize_anchors.py
* [Fix]Fix import error in optimize_anchors.py
Co-authored-by: HinGwenWoong <peterhuang0323@qq.com>
2023-01-11 17:09:31 +08:00
c682ae82c6
adjust the minimum supported python version to 3.9 ( #449 )
2023-01-09 18:31:37 +08:00
b0cf073241
Bump version to v0.3.0
...
Bump version to v0.3.0
2023-01-08 15:48:53 +08:00
9c6b4fedc8
Add v0.3.0 changelog ( #440 )
...
* bump version to v0.3.0
* add v7 cat demo config
* update mmdet version
* update
* fix topk out of range error
* update
* add changelog.md
* update
* update
* update
* update
* update
2023-01-08 15:31:58 +08:00
fc107b8997
[Docs] Add an example project to 'projects\' folder ( #401 )
...
* add_example_project
* renew readme
* Update projects/example_project/README.md
Co-authored-by: HinGwenWoong <peterhuang0323@qq.com>
* Update projects/example_project/README.md
Co-authored-by: HinGwenWoong <peterhuang0323@qq.com>
* Update projects/example_project/README.md
Co-authored-by: HinGwenWoong <peterhuang0323@qq.com>
* Update projects/example_project/README.md
Co-authored-by: HinGwenWoong <peterhuang0323@qq.com>
* Update projects/example_project/README.md
Co-authored-by: HinGwenWoong <peterhuang0323@qq.com>
* Update projects/example_project/README.md
Co-authored-by: Haian Huang(深度眸) <1286304229@qq.com>
* Update projects/example_project/dummy/dummy_resnet.py
Co-authored-by: Haian Huang(深度眸) <1286304229@qq.com>
* Update projects/example_project/README.md
Co-authored-by: Haian Huang(深度眸) <1286304229@qq.com>
* Update projects/example_project/dummy/dummy_resnet.py
Co-authored-by: Haian Huang(深度眸) <1286304229@qq.com>
* Update projects/example_project/README.md
Co-authored-by: Haian Huang(深度眸) <1286304229@qq.com>
* renew to yolov5
* Update projects/example_project/dummy/dummy_yolov5cspdarknet.py
Co-authored-by: Range King <RangeKingHZ@gmail.com>
* Update projects/example_project/README.md
Co-authored-by: Range King <RangeKingHZ@gmail.com>
* Update projects/example_project/dummy/dummy_yolov5cspdarknet.py
Co-authored-by: HinGwenWoong <peterhuang0323@qq.com>
* fix issue
* Update projects/example_project/dummy/dummy_yolov5cspdarknet.py
* Update projects/example_project/README.md
* Update projects/example_project/README.md
Co-authored-by: HinGwenWoong <peterhuang0323@qq.com>
Co-authored-by: Haian Huang(深度眸) <1286304229@qq.com>
Co-authored-by: Range King <RangeKingHZ@gmail.com>
2023-01-08 14:51:39 +08:00
a631370cb3
Update model graphs ( #442 )
2023-01-07 17:44:19 +08:00
89275adb29
[CodeCamp] #50 Add module combination doc ( #349 )
...
* the file to verify the installation
* "add module_combination"
* add module_combination
* modified module_combination.md
* modify module_combination.md
* fix module_combination.md
2023-01-06 20:30:48 +08:00
7d07e1a780
Update yolo_detector.py ( #441 )
...
将 YOLOX -> YOLO
2023-01-06 19:20:24 +08:00
07afc3ee78
Fix potential bugs in rtmdet multi-scale training ( #439 )
...
* Distinguish variables when training and testing
* beauty
2023-01-06 19:03:14 +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
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
a20b160f0f
[CodeCamp]Add doc of yolov6_description.md ( #382 )
...
* YOLOv6网络结构文档
* 修正YOLOv6网络结构文档
* 修改模型结构图四周多余的留白
Co-authored-by: yl4203 <wsnbbzwd19960817>
2023-01-06 14:59:34 +08:00
c14ecc336c
[DOC] Update useful_tools.md ( #384 )
...
* Update useful_tools.md
更新了浏览数据部分的文档
* Update useful_tools.md
* Update useful_tools.md
* Update useful_tools.md
* Update docs/zh_cn/user_guides/useful_tools.md
Co-authored-by: HinGwenWoong <peterhuang0323@qq.com>
Co-authored-by: HinGwenWoong <peterhuang0323@qq.com>
2023-01-06 14:48:17 +08:00
2f399df2d6
[Feature] Add the unit test to test the `scale_factor` in LetterResize ( #408 )
...
* 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
Co-authored-by: hha <1286304229@qq.com>
2023-01-06 14:40:21 +08:00
2290713ae6
[Fix] Fix bugs in the output annotation file when `--class-id-txt` is set. ( #430 )
...
* [Fix] Fix bugs in the output annotation file when is set.
* Update labelme2coco.py
* Update labelme2coco.py
* Improve code
Co-authored-by: HinGwenWoong <peterhuang0323@qq.com>
2023-01-06 11:41:17 +08:00
d8cbf216de
[Fix] Fix typo in docstring ( #433 )
...
* fix typo in docstring
* fix typo in docs
* fix typo in docs
2023-01-06 11:40:47 +08:00
5172805dae
fix rtmdet-s cfg and neck ( #434 )
2023-01-06 11:40:19 +08:00
28439f102a
[Deploy] Support YOLOv7 easy deploy ( #427 )
...
* Support YOLOv7 easy deploy
* Remove useless comment
* Fix mkdir
2023-01-04 20:10:58 +08:00
7c5e21c39f
Fix customize dataset training ( #428 )
2023-01-04 20:02:02 +08:00
48f8896e84
[Feature] Implement fast version of RTMDet. ( #425 )
...
* Accelerate RTMDet
* update
* update
* update
* update1
* update2
* update pipeline
* update lr cudnnbenchmark
* revert batchsize
* fix batch inference
* refactor head
* update box
* bs=16
* update
* move reduce mean
* update head
* per img loss
* fix
* fix sum
* concat loss
* batch dsla
* sort topk
* bs 32
* clean code
* update readme
* update ut
* update checkpoint
* num_class
* clean code
* resolve comments
* fix readme
* fix ut
Co-authored-by: huanghaian <huanghaian@sensetime.com>
Co-authored-by: hha <1286304229@qq.com>
2023-01-04 16:06:44 +08:00
c7a9026812
url fix in README ( #426 )
2023-01-03 15:43:23 +08:00
14d9cab354
[CodeCamp] Add docs about how to freeze the weight of backbone or neck ( #418 )
...
* Add docs about how to freeze the weight of backbone or neck
* refine the docs
* refine the docs
* refine the docs
2022-12-31 16:36:40 +08:00
b1f7e80fe3
Fix typehint in Heads ( #416 )
2022-12-31 16:35:18 +08:00
13de22dfd2
Fix typehint in YOLOv6 Head ( #415 )
2022-12-29 21:21:44 +08:00
cdc359c2de
fix batch inference bug ( #413 )
2022-12-29 17:18:54 +08:00
99af3c980f
[Fix] Set metainfo fields to lower case and fix CI ( #412 )
...
* Set metainfo fields to lower case
* Set metainfo fields to lower case
* Fix ci
* Fix ci
2022-12-29 16:23:52 +08:00
Haian Huang(深度眸)
yechenzhi
布鲁瓦丝
Audrey528
tripleMu
Range King
Youfu
Nioolek
LinXiaoZheng
HinGwenWoong
Yijie Zheng
tianlei
Haonan
Seperendity
Jing Tang
wzr-skn
MingJian.L
tang576225574
jason_w
RangiLyu
Xin Li
kitecats