327 Commits

Author SHA1 Message Date
Range King
55da55eb09
Update mix_img_transforms.py (#319) 2022-11-28 09:58:00 +08:00
Range King
60a515f41f
[Enhance] Add option of json output in test.py (#316)
* add json args

* Update how_to.md

* Update test.py

* Update how_to.md

* Update how_to.md
2022-11-26 19:59:37 +08:00
Range King
5efa56b3b5
[Docs] Update video tutorials in article.md (#318)
* Update article.md

* Update article.md

* Update article.md
2022-11-26 19:58:41 +08:00
AZhou
4b79e4f983
[Feature] Add condition for extract coco datasets (#286)
* add condition for extract coco datasets

* fix lint error

* guarantee images be extracted definitely has annotation info

* reset num_img default value
2022-11-26 16:19:04 +08:00
Xin Li
7440a68e51
[Docs] Deployment translation (#289)
* translate deployment docs

* translate deployment docs part 2

* Update docs/zh_cn/deploy/yolov5_deployment.md

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

* Update docs/zh_cn/deploy/yolov5_deployment.md

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

* Update docs/zh_cn/deploy/yolov5_deployment.md

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

* Update docs/zh_cn/deploy/yolov5_deployment.md

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

* Update docs/zh_cn/deploy/yolov5_deployment.md

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

* Update docs/zh_cn/deploy/yolov5_deployment.md

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

* Update docs/zh_cn/deploy/yolov5_deployment.md

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

* Update docs/zh_cn/deploy/yolov5_deployment.md

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

* Update docs/en/deploy/basic_deployment_guide.md

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

* Update docs/en/deploy/basic_deployment_guide.md

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

* Update docs/en/deploy/basic_deployment_guide.md

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

* debug link representation in readthedocs

* debug link representation in readthedocs

* debug link representation in readthedocs

Co-authored-by: Range King <RangeKingHZ@gmail.com>
Co-authored-by: HinGwenWoong <peterhuang0323@qq.com>
2022-11-26 10:07:50 +08:00
HinGwenWoong
daae790075
[Feature] Add split COCO dataset script (#311)
* Add coco_sqlit script

* rename file

* Improve code

* Improve code

* Improve coding

* Improve file structure

* ratios normalize
2022-11-26 09:27:30 +08:00
hanrui1sensetime
15224b7816
[Fix] Fix mmyolo int8 deploy config (#315)
* fix mmyolo int8 deploy config

* fix lint
2022-11-26 09:26:40 +08:00
HinGwenWoong
285c395066
[Improvement] Add output class_with_id.txt after labelme2coco (#313)
* Add output class_with_id.json

* Add `--class-id-json`

* Improve coding, using txt and saving txt

* Improve coding
2022-11-25 16:50:50 +08:00
HinGwenWoong
e3f1cf93a6
[Improvement] Add --class-name for filter when save to labelme label. (#314)
* Add `--class-name` for filter when save to labelme label.

* Improve code

* Add check class

* Improve coding

* Improve coding
2022-11-25 16:48:59 +08:00
Haian Huang(深度眸)
9221499af4
[Feature] Support grad-based cam and grad-free cam (#234)
* support cam

* update

* update

* done

* fix lint

* add docstr

* add doc

* update

* update

* FIX
2022-11-25 16:45:40 +08:00
tripleMu
1045b41b68
[Deploy] MMYOLO model convert to onnx for deployment. (#279)
* Fromat code

* Support ONNXRUNTIME

* Support mmyolo model convert to onnx for deploy.

* Same as dev branch

* Support yolox focus rewrite

* Support GConv Focus

* Update mmyolo/easydeploy/backbone/focus.py

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

* Add TensorRT build/infer Wrapper

* Add image_demo for deploy model

* Fix

* Merge dev

* Remove image-demo

* Roll back to dev

* Support model switch to deploy

* Remove --deploy

* Add new deploy method

* Format code and add doc

* Move md to project

* add readme and readme_zh to easy_deploy

* Update projects/easydeploy/README_zh-CN.md

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

* Update projects/easydeploy/README.md

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

Co-authored-by: HinGwenWoong <peterhuang0323@qq.com>
Co-authored-by: xin-li-67 <williamlee.xin@gmail.com>
Co-authored-by: Xin Li <7219519+xin-li-67@users.noreply.github.com>
Co-authored-by: Haian Huang(深度眸) <1286304229@qq.com>
2022-11-23 10:43:41 +08:00
HinGwenWoong
b1e478a8bd
[Feature] Add labelme2coco script (#308)
* Add label2coco script

* Add label2coco script

* Add label2coco script

* Add label2coco script

* Improve script

* Improve script
2022-11-23 10:32:52 +08:00
kitecats
748f151886
[Refactor] Refactor some of the functionality of dataset_analysis (#294)
* Refactor some of the functionality of data_analysis

* Remove redundant code in dataset_analysis.py

* Update tools/analysis_tools/dataset_analysis.py

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

* simplified code

* add docstring and simplify code

Co-authored-by: Haian Huang(深度眸) <1286304229@qq.com>
2022-11-22 11:21:35 +08:00
HinGwenWoong
47bb3ce408
[Improvement] Improve browse coco json script args (#309)
* Improve browse coco json script

* Improve doc

* Improve script

* Improve doc

* Improve doc

* Improve doc
2022-11-22 10:51:42 +08:00
HinGwenWoong
1773439b83
[Fix] Fix scale factor order of LetterResize and YOLOv5KeepRatioResize (#305)
* Fix bug

* Fix bug

* Fix lint

* Fix scale from hw -> wh

* Fix scale from hw -> wh
2022-11-21 19:44:33 +08:00
Range King
82d1680b4d
[Improvement] Add mosaic9 graph in docstring (#307)
* Add mosaic9 graph in docstring

* fix typos
2022-11-21 19:13:52 +08:00
Haian Huang(深度眸)
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
HinGwenWoong
5e0599c825
[Feature] Add flag for output labelme label file in image_demo (#288)
* Add flag for labelme

* Add labelme format code

* Add labelme format code

* Add labelme format code

* Add labelme format code

* Add labelme format code

* Add labelme format code

* Add labelme format code

* Add labelme format code

* Add labelme format code

* Add labelme format code

* Add labelme format code

* Not save `imageData`

* preprocess score

* preprocess score

* Fix lint
2022-11-20 15:37:58 +08:00
LinXiaoZheng
12a1e8a984
[Docs] Fix config.md (#297)
* fix_config.md

* Delete yolo_bricks.py

* fix

* Update yolo_bricks.py
2022-11-20 09:13:35 +08:00
Range King
cb93fb6754
fix links in config.md (#303) 2022-11-20 09:12:49 +08:00
HinGwenWoong
90494b44f2
[DOC] Add YOLOv6 description doc - overview (#252)
* [DOC] Add YOLOv6 description doc

* [DOC] Add YOLOv6 description 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-11-19 10:22:26 +08:00
fatih
0fd6444c00
[Feature] Add large image demo with sahi (#284)
* add large image demo with sahi

* fix some typos

* restructure based on reviews

* update default patch size

* add docstring and update docs

* updates based on reviews

* print information

* add debug, update docs, add large image sample

* update docs

* update docs

* update docs

* direct user to install sahi
2022-11-18 20:30:43 +08:00
jason_w
5cee9c977d
fix make_stage_plugins doc in basebackbone (#296)
* fix 'make_stage_plugins' doc

* Update the parameter name to be consistent with the document
2022-11-18 20:25:36 +08:00
Range King
1c8e581d2c
[Docs] Add two examples of backbone replacement and update plugin.md in EN (#291)
* Update how_to.md

* delete toc

* Update how_to.md

* Update plugin.md

* Update how_to.md

* Update how_to.md

* replace todolist

* fix pretrained model link

* Update how_to.md

* refine configs
2022-11-18 09:55:50 +08:00
hanrui1sensetime
5ecaf18eda
fix rtmdet deploy error (#287) 2022-11-17 15:53:03 +08:00
Range King
c043181149
[Fix] training errors of yolox nano (#285)
* Update yolox_pafpn.py

* Update csp_darknet.py
2022-11-16 11:13:35 +08:00
wanghonglie
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
Haian Huang(深度眸)
178b0bf3ff
Bump version to v0.1.3
Bump version to v0.1.3
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
v0.1.2
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