Commit Graph

150 Commits (178b0bf3ffec1353923460670bcbafccc15121cf)
 

Author SHA1 Message Date
Haian Huang(深度眸) 178b0bf3ff
Bump version to v0.1.3
Bump version to v0.1.3
2022-11-10 11:22:03 +08:00
Haian Huang(深度眸) abcde7679a add v0.1.3 changelog (#275) 2022-11-10 11:12:31 +08:00
Range King 7f14183082 [Docs] Add graph of P6 model (#273)
* Update yolov5_description.md

* Update model_design.md

* Fix typo in yolov5_description.md

* Update model_design.md

* Update yolov5_description.md

* Update model_design.md

* Update README.md

* Update README_zh-CN.md

* Fix format

* Update en docs

* Refine zh_cn docs

* Update README_zh-CN.md

* Fix links of model_design.md

* specify the config file
2022-11-10 11:12:31 +08:00
kitecats f1279eb3de [Feature] Add attention module of CBAM (#246)
* Add Attention Modules

* Adde tutorials on the use of the attention module in How_to

* Update how_to.md

Added tutorials on the use of the attention module

* Update attention_layers.py

* Rename attention_layers.py to cbam_layer.py

* Update __init__.py

* Update how_to.md

* Update how_to.md

* Update how_to.md

* Update cbam_layer.py

* Update cbam_layer.py

* Update cbam_layer.py

* Update how_to.md

* update

* add docstring typehint

* add unit test

* refine unit test

* updata how_to

* add plugins directory

* refine plugin.md

* refine cbam.py and plugins.md

* refine cbam.py and plugins.md

* fix error in test_cbam.py

* refine cbam.py and fix error in test_cbam.py

* refine cbam.py and plugins.md

* refine cbam.py and docs
2022-11-10 11:12:31 +08:00
HinGwenWoong 596ffd6617 [Feature] `featmap_vis_demo` supports input image folder and url reference (#248)
* [Feature] supports input image folder and url reference

* [Feature] supports input image folder and url reference

* Fix col bug

* Fix filename bug

* Fix filename bug

* Fix filename bug

* Fix filename bug

* Improve suffix

* Add utils.py for common func

* Fix lint

* improve coding

* improve coding

* Fix typo

* Fix typo

* Fix typo

* Improve coding

* Fix coding

* Improve coding
2022-11-10 11:12:31 +08:00
HinGwenWoong 8626fa5f02 [DOC] Update mmengine to `0.3.1`, mmdet to `3.0.0rc3` (#269)
* Update mmengine version

* Update mmengine version

* Update mmengine version

* Update mmdet version

* Improve test.yml

* Fix version

* Roll back test.yaml
2022-11-10 11:12:31 +08:00
LinXiaoZheng fd73c32750 [Docs] Fix useful_tools.md #274 2022-11-10 11:12:31 +08:00
Xiaolin Wang 71e5c793e5 [Doc] Fix typo (#270)
* fix typo

* fix ci
2022-11-10 11:12:31 +08:00
魏凯 1eaa260625 updata backbone load init_cfg (#272) 2022-11-10 11:12:31 +08:00
tripleMu ee122ec5bd [Update] Change the typing import method based on mmdet (#261)
* Fix import from mmdet

* Fix pre-commit known_third_party
2022-11-10 11:12:31 +08:00
LinXiaoZheng 4c1878ce8b [Improvement] dataset_analysis (#257)
* add ConcatDataset judgment

* Update dataset_analysis.py
2022-11-10 11:12:31 +08:00
Yue Zhou 1e0cab7371 Update yolov5_description.md (#254) 2022-11-10 11:12:31 +08:00
wanghonglie 7ed550b6fc update dump results (#253) 2022-11-10 11:12:31 +08:00
Range King b99bbd1971 [Fix] Wrong layer number in model graph (#238)
* Fix base module graph in README.md

* Fix base module graph in README_zh-CN.md

* Update yolov5_description.md

Fix YOLOv5 structure graph in yolov5_description.md

* Update rtmdet_description.md

Fix RTMDet structure graph

* Fix base module graph in model_design.md

* Update yolov5_description.md

Fix YOLOv5 structure graph

* Reorder structure in base_neck docstring
2022-11-10 11:12:31 +08:00
tripleMu 307cb42d30 [BUG] Fix opencv new version bug (#242)
* Install opencv old version

* Replace mmcv-full with source code install

* Fix env

* Fix env

* Fix env

* Fix typo
2022-11-10 11:12:31 +08:00
LinXiaoZheng e510543442 add voc2012 download link (#251) 2022-11-10 11:12:31 +08:00
HinGwenWoong 0b48313dc7 [Fxi] Fix ci bug of BatchATSSAssigner (#232)
* Fix ci bug

* Fix ci bug

* Fix ci bug

* Fix ci bug

* Fix ci bug
2022-11-03 19:03:06 +08:00
Haian Huang(深度眸) 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
wanghonglie f045ecf213 [Feature] Add YOLOv6 metafile (#227)
* add v6 meta file

* update v6s
2022-11-03 19:03:06 +08:00
tripleMu 8e0f18be6d [Feautre] Add deploy dockerfile (#224)
* Add dockerfile for deploy

* Fix

* Update docker/Dockerfile_deployment

Co-authored-by: HinGwenWoong <peterhuang0323@qq.com>

* Update docker/Dockerfile_deployment

Co-authored-by: HinGwenWoong <peterhuang0323@qq.com>

* Add opencv old version

* Add dockerfile and fix some typo

* Remove repeat packages

* Fix undefined symbol bug

* Update docker/Dockerfile_deployment

Co-authored-by: Haian Huang(深度眸) <1286304229@qq.com>

* Update docker/Dockerfile_deployment

Co-authored-by: HinGwenWoong <peterhuang0323@qq.com>

* Add docs for deploy dockerfile

* Fix typo

* Update docs/zh_cn/advanced_guides/yolov5_deploy.md

Co-authored-by: HinGwenWoong <peterhuang0323@qq.com>

* Add profiler

* Update docs/zh_cn/advanced_guides/yolov5_deploy.md

Co-authored-by: HinGwenWoong <peterhuang0323@qq.com>

* Update docker/Dockerfile_deployment

Co-authored-by: Range King <RangeKingHZ@gmail.com>

* Remove mmcv-full

Co-authored-by: HinGwenWoong <peterhuang0323@qq.com>
Co-authored-by: Haian Huang(深度眸) <1286304229@qq.com>
Co-authored-by: Range King <RangeKingHZ@gmail.com>
2022-11-03 19:03:06 +08:00
Nioolek 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
LinXiaoZheng e6e6f736e6 [Feature] Add dataset analysis script (#172)
* messages

* again

* again1

* again_2

* again_3

* again_4

* again_5

* Update docs/zh_cn/user_guides/useful_tools.md

Co-authored-by: HinGwenWoong <peterhuang0323@qq.com>

* Update docs/zh_cn/user_guides/useful_tools.md

Co-authored-by: HinGwenWoong <peterhuang0323@qq.com>

* Update docs/en/user_guides/useful_tools.md

Co-authored-by: HinGwenWoong <peterhuang0323@qq.com>

* Update docs/en/user_guides/useful_tools.md

Co-authored-by: HinGwenWoong <peterhuang0323@qq.com>

* Update docs/en/user_guides/useful_tools.md

Co-authored-by: HinGwenWoong <peterhuang0323@qq.com>

* Update docs/zh_cn/user_guides/useful_tools.md

Co-authored-by: HinGwenWoong <peterhuang0323@qq.com>

* Update tools/analysis_tools/dataset_analysis.py

Co-authored-by: HinGwenWoong <peterhuang0323@qq.com>

* Update tools/analysis_tools/dataset_analysis.py

Co-authored-by: HinGwenWoong <peterhuang0323@qq.com>

* Update tools/analysis_tools/dataset_analysis.py

Co-authored-by: HinGwenWoong <peterhuang0323@qq.com>

* Update tools/analysis_tools/dataset_analysis.py

Co-authored-by: HinGwenWoong <peterhuang0323@qq.com>

* modify code

* Update docs/en/user_guides/useful_tools.md

Co-authored-by: HinGwenWoong <peterhuang0323@qq.com>

* Modify document

* Modify document

* new code

* revise decuments and codes

* Revise datails

* Update tools/analysis_tools/dataset_analysis.py

Co-authored-by: HinGwenWoong <peterhuang0323@qq.com>

* modify func name

* code

* Documentation and code

* modify error meaasge

* deleted height,

Co-authored-by: HinGwenWoong <peterhuang0323@qq.com>
2022-11-03 19:03:06 +08:00
Xin Li 93e802b0bf [Docs] yolov5_description.md in English (#198)
* draft 1 yolov5_descrip_eng

* Update docs/en/algorithm_descriptions/yolov5_description.md

Co-authored-by: HinGwenWoong <peterhuang0323@qq.com>

* Update docs/en/algorithm_descriptions/yolov5_description.md

Co-authored-by: HinGwenWoong <peterhuang0323@qq.com>

* Update docs/en/algorithm_descriptions/yolov5_description.md

Co-authored-by: HinGwenWoong <peterhuang0323@qq.com>

* add section 1.3

* draft finished

* update docs

* Update docs/en/algorithm_descriptions/yolov5_description.md

Co-authored-by: HinGwenWoong <peterhuang0323@qq.com>

* remove unnecessary parts

* Update docs/zh_cn/algorithm_descriptions/yolov5_description.md

Co-authored-by: Range King <RangeKingHZ@gmail.com>

* Update docs/en/algorithm_descriptions/yolov5_description.md

Co-authored-by: Range King <RangeKingHZ@gmail.com>

* Update docs/en/algorithm_descriptions/yolov5_description.md

Co-authored-by: Range King <RangeKingHZ@gmail.com>

* Update docs/en/algorithm_descriptions/yolov5_description.md

Co-authored-by: Range King <RangeKingHZ@gmail.com>

* Update docs/en/algorithm_descriptions/yolov5_description.md

Co-authored-by: Range King <RangeKingHZ@gmail.com>

* Update docs/en/algorithm_descriptions/yolov5_description.md

Co-authored-by: Range King <RangeKingHZ@gmail.com>

* Update docs/en/algorithm_descriptions/yolov5_description.md

Co-authored-by: Range King <RangeKingHZ@gmail.com>

* Update docs/en/algorithm_descriptions/yolov5_description.md

Co-authored-by: Range King <RangeKingHZ@gmail.com>

* open-sourced -> open-source

* merge&update

* assignment strategy&merge dev branch

Co-authored-by: HinGwenWoong <peterhuang0323@qq.com>
Co-authored-by: Range King <RangeKingHZ@gmail.com>
2022-11-03 19:03:06 +08:00
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 47939e99da [DOC] Add deploy guide doc (#220)
* Add deploy doc-0

* Update 部署必备指南.md

Co-authored-by: HinGwenWoong <peterhuang0323@qq.com>

* Update 部署必备指南.md

Co-authored-by: HinGwenWoong <peterhuang0323@qq.com>

* Update 部署必备指南.md

Co-authored-by: HinGwenWoong <peterhuang0323@qq.com>

* Update 部署必备指南.md

Co-authored-by: HinGwenWoong <peterhuang0323@qq.com>

* Update 部署必备指南.md

Co-authored-by: HinGwenWoong <peterhuang0323@qq.com>

* Update 部署必备指南.md

Co-authored-by: HinGwenWoong <peterhuang0323@qq.com>

* Update 部署必备指南.md

Co-authored-by: HinGwenWoong <peterhuang0323@qq.com>

* Update 部署必备指南.md

Co-authored-by: HinGwenWoong <peterhuang0323@qq.com>

* Update 部署必备指南.md

Co-authored-by: HinGwenWoong <peterhuang0323@qq.com>

* Del unused line

* Add docs

* Fix

* Fix

* Rename docs

* Rename docs

Co-authored-by: HinGwenWoong <peterhuang0323@qq.com>
2022-11-03 19:03:06 +08:00
AZhou 188377c62b [Fix] Fix resume is not set correctly bug when `cfg.resume` is True (#221)
* add optimize_anchors for mmyolo

* update optimize_anchors

* optimize anchors script

* add parameter in args

* add optimize anchor related docs

* fix lint error

* fix docs

* change anchors output format

* fix docs error

* fix resume set

* fix resume mode set

* add mmeval relate docs

Co-authored-by: imAzhou <lyzhou@creams.io>
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
AZhou 190ee5aaa7 [Feature] Add `optimize_anchors` script (#174)
* add optimize_anchors for mmyolo

* update optimize_anchors

* optimize anchors script

* add parameter in args

* add optimize anchor related docs

* fix lint error

* fix docs

* change anchors output format

* fix docs error

Co-authored-by: imAzhou <lyzhou@creams.io>
2022-11-03 19:03:06 +08:00
Range King 2fdbe3c4e7 [Docs] Optimize Readthedocs page display (#209)
* Adjust the logo size

* Fix the language type
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
Xin Li 14d3f294ef [Docs] Translate visualization.md (#206)
* visualization.md translate draft

* visualization.md translate draft

* Update docs/en/user_guides/visualization.md

Co-authored-by: HinGwenWoong <peterhuang0323@qq.com>

* Update docs/en/user_guides/visualization.md

Co-authored-by: Range King <RangeKingHZ@gmail.com>

* Update docs/en/user_guides/visualization.md

Co-authored-by: Range King <RangeKingHZ@gmail.com>

* visualization.md translate draft 2

* Update docs/en/user_guides/visualization.md

Co-authored-by: Range King <RangeKingHZ@gmail.com>

* Update docs/zh_cn/user_guides/visualization.md

Co-authored-by: Range King <RangeKingHZ@gmail.com>

* Update docs/en/user_guides/visualization.md

Co-authored-by: Range King <RangeKingHZ@gmail.com>

* removed 'identify' in the params explanation

Co-authored-by: HinGwenWoong <peterhuang0323@qq.com>
Co-authored-by: Range King <RangeKingHZ@gmail.com>
2022-11-03 19:03:06 +08:00
Range King 38173ccc5c [Docs] Add examples of how to use the backbone from other repos in `how_to.md` and update `article.md` (#207)
* Update article.md

* Update article.md

* Update how_to.md

* Update how_to.md

* Refine zh_CN how_to.md

* Update EN how_to.md

* Refine comments in how_to.md

* Adjust the position of notes
2022-11-03 19:03:06 +08:00
Haian Huang(深度眸) 42bb4fd14a fix pypi (#205) 2022-11-03 19:03:06 +08:00
Haian Huang(深度眸) 4e4615c835 Fix not showing box bug and update the visualization tutorial (#204)
* Fix not showing box bug and update the visualization tutorial

* update

* update

* update
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
Haian Huang(深度眸) 89588b60fb update mmengine/mmdet version and article (#201)
* update mmengine and mmdet version

* update
2022-11-03 19:03:06 +08:00
kitecats a381c0f932 [Docs] Fix some errors in `config.md` and `how_to.md` (#200)
* Update config.md

* Update docs/zh_cn/user_guides/config.md

Co-authored-by: Haian Huang(深度眸) <1286304229@qq.com>

* [Docs] Fix some error in how_to.md

* 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>

* 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: HinGwenWoong <peterhuang0323@qq.com>

* Update yolov5_head.py

* Update yolox_head.py

* Update yolov5_head.py

* Update yolox_head.py

* Update docs/zh_cn/advanced_guides/how_to.md

Co-authored-by: Haian Huang(深度眸) <1286304229@qq.com>

* Update how_to.md

* Update yolov5_head.py

* Update yolox_head.py

Co-authored-by: Haian Huang(深度眸) <1286304229@qq.com>
Co-authored-by: Range King <RangeKingHZ@gmail.com>
Co-authored-by: HinGwenWoong <peterhuang0323@qq.com>
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
Haian Huang(深度眸) ce73b03b72 [Feature] Add `extract_subcoco` script (#186)
* add extract_subcoco.py

* update

* fix comment

* fix comment
2022-11-03 19:03:06 +08:00
Nioolek d75fbabcb0 Add visualization of the base model to the docstring. (#196)
* add yolov5 voc training

* format code

* [Feature] Support YOLOv7 inference (#149)

* update

* update

* update

* update

* update

* add docstr

* fix comments

* update

* add video of featmap vis demo (#169)

* [Enhance] YOLOv6 model convert (#165)

* add yolov5 voc training

* fix bug

* fix bug

* fix yolov6 model convert

* update yolov5 modelconvert

* rollback yolov5 model convert

* [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

* add neck vis docstr

* add neck vis docstr

* add backbone model structure diagram

* fix readthedocs.txt

* fix doc vis

* fix vis

* fix vis

* fix vis

* fix vis

Co-authored-by: Haian Huang(深度眸) <huanghaian@pjlab.org.cn>
2022-11-03 19:03:06 +08:00
Range King 75997fcd60 [Docs] Refine config.md (#191)
* Refine __delete__ example in config.md

* Add missing docstring in cspnext.py

* Refine docstring of cspnext

* Correct the description of param_scheduler

* Refine description param_scheduler in config.md

* Update docs/zh_cn/user_guides/config.md

Co-authored-by: Haian Huang(深度眸) <1286304229@qq.com>

Co-authored-by: Haian Huang(深度眸) <1286304229@qq.com>
2022-11-03 19:03:06 +08:00
Range King 2f5d16f5f1 [Docs] Update config.md, faq.md and pull_request_template.md (#190)
* Update Chinese faq.md

add the description of the difference between MMYOLO
 and MMDet

* Update English faq.md

add the description of the difference between MMYOLO
and MMDet

* Update Chinese config.md

* Update English config.md

* Update pull_request_template.md

* Delete the unnecessary description in config.md

* Delete the unnecessary description in config.md
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
Xin Li 714a7238cf [Feature] Add script converts yolo-style to coco format (#161)
* the script converts yolo-style to coco format draft 1

* Update tools/dataset_converters/yolo2coco.py

Co-authored-by: HinGwenWoong <peterhuang0323@qq.com>

* Update tools/dataset_converters/yolo2coco.py

Co-authored-by: HinGwenWoong <peterhuang0323@qq.com>

* Update tools/dataset_converters/yolo2coco.py

Co-authored-by: HinGwenWoong <peterhuang0323@qq.com>

* Update tools/dataset_converters/yolo2coco.py

Co-authored-by: HinGwenWoong <peterhuang0323@qq.com>

* Update tools/dataset_converters/yolo2coco.py

Co-authored-by: HinGwenWoong <peterhuang0323@qq.com>

* Update tools/dataset_converters/yolo2coco.py

Co-authored-by: HinGwenWoong <peterhuang0323@qq.com>

* Update tools/dataset_converters/yolo2coco.py

Co-authored-by: HinGwenWoong <peterhuang0323@qq.com>

* typo fixed and update

* encaped and modified yolo2coco

* more usage instructions of yolo2coco

* Update docs/zh_cn/user_guides/useful_tools.md

Co-authored-by: HinGwenWoong <peterhuang0323@qq.com>

* Update docs/en/user_guides/useful_tools.md

Co-authored-by: HinGwenWoong <peterhuang0323@qq.com>

* Update tools/dataset_converters/yolo2coco.py

Co-authored-by: HinGwenWoong <peterhuang0323@qq.com>

* Update tools/dataset_converters/yolo2coco.py

Co-authored-by: HinGwenWoong <peterhuang0323@qq.com>

* Update docs/en/user_guides/useful_tools.md

Co-authored-by: HinGwenWoong <peterhuang0323@qq.com>

* Update docs/zh_cn/user_guides/useful_tools.md

Co-authored-by: HinGwenWoong <peterhuang0323@qq.com>

* Update docs/zh_cn/user_guides/useful_tools.md

Co-authored-by: HinGwenWoong <peterhuang0323@qq.com>

* Update docs/en/user_guides/useful_tools.md

Co-authored-by: HinGwenWoong <peterhuang0323@qq.com>

* final draft of yolo2coco

* final draft of yolo2coco

* final draft of yolo2coco

* revise the script and remove -split indicator

* Update docs/en/user_guides/useful_tools.md

Co-authored-by: HinGwenWoong <peterhuang0323@qq.com>

* update the structural example of the root path

* Update tools/dataset_converters/yolo2coco.py

Co-authored-by: HinGwenWoong <peterhuang0323@qq.com>

* fix typo on check_existence

* Update docs/zh_cn/user_guides/useful_tools.md

Co-authored-by: HinGwenWoong <peterhuang0323@qq.com>

* Update tools/dataset_converters/yolo2coco.py

Co-authored-by: HinGwenWoong <peterhuang0323@qq.com>

* Update tools/dataset_converters/yolo2coco.py

Co-authored-by: HinGwenWoong <peterhuang0323@qq.com>

* Update tools/dataset_converters/yolo2coco.py

Co-authored-by: HinGwenWoong <peterhuang0323@qq.com>

* Update tools/dataset_converters/yolo2coco.py

Co-authored-by: HinGwenWoong <peterhuang0323@qq.com>

* update the commits

* rewrite split_by_files part and add copy_image part

* add instruction on using absolute image paths

* add instruction on using absolute image paths

* Update docs/en/user_guides/useful_tools.md

* Update tools/dataset_converters/yolo2coco.py

Co-authored-by: Range King <RangeKingHZ@gmail.com>

* Update docs/en/user_guides/useful_tools.md

Co-authored-by: Range King <RangeKingHZ@gmail.com>

* Update docs/en/user_guides/useful_tools.md

Co-authored-by: Range King <RangeKingHZ@gmail.com>

* Update tools/dataset_converters/yolo2coco.py

Co-authored-by: Range King <RangeKingHZ@gmail.com>

* Update tools/dataset_converters/yolo2coco.py

Co-authored-by: Range King <RangeKingHZ@gmail.com>

* Update tools/dataset_converters/yolo2coco.py

Co-authored-by: Range King <RangeKingHZ@gmail.com>

* Update tools/dataset_converters/yolo2coco.py

Co-authored-by: Range King <RangeKingHZ@gmail.com>

* add basic statistic func

* add basic statistic func

* revise the script

* remove copy&move image func

* Update docs/zh_cn/user_guides/useful_tools.md

Co-authored-by: Haian Huang(深度眸) <1286304229@qq.com>

* Update docs/en/user_guides/useful_tools.md

Co-authored-by: Haian Huang(深度眸) <1286304229@qq.com>

* Update docs/en/user_guides/useful_tools.md

Co-authored-by: Haian Huang(深度眸) <1286304229@qq.com>

* Update docs/en/user_guides/useful_tools.md

Co-authored-by: Haian Huang(深度眸) <1286304229@qq.com>

* Update docs/en/user_guides/useful_tools.md

Co-authored-by: Haian Huang(深度眸) <1286304229@qq.com>

* Update docs/en/user_guides/useful_tools.md

Co-authored-by: Haian Huang(深度眸) <1286304229@qq.com>

* Update tools/dataset_converters/yolo2coco.py

Co-authored-by: Haian Huang(深度眸) <1286304229@qq.com>

* revise image_dir

Co-authored-by: HinGwenWoong <peterhuang0323@qq.com>
Co-authored-by: Range King <RangeKingHZ@gmail.com>
Co-authored-by: Haian Huang(深度眸) <1286304229@qq.com>
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
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(深度眸) 936bcf8d86 add video of featmap vis demo (#169) 2022-11-03 19:03:06 +08:00