Commit Graph

37 Commits (729bdacecf3461223e1ce66d761c10619b1fed2c)

Author SHA1 Message Date
Lum 729bdacecf
[Docs] Create 15 minutes instance segmentation tutorial. (#765)
* Add balloon support of yolov5s instance segmentation.

* Fixed typo.

* [Docs] Update 15_minutes_instance_segmentation.md Chinese version.

* [Docs] Update 15_minutes_object_detection.md.

* [Docs] Update 15_minutes_instance_segmentation.md

* Update picture.

* Update comment.

* Fixed typo.

* [Docs] Create 15 min ins seg toturial.

* Create jupyter notebook.

* Fixed typo.

* Update colab url.
2023-08-15 11:36:32 +08:00
JosonChan e8203150f5
Fix copypaste in yolov5-ins l/x config (#756)
* [Fix] fix copypaste in yolov5-ins l/x config

* fix

* format

* update

* update

---------

Co-authored-by: huanghaian <huanghaian@sensetime.com>
2023-05-10 12:16:38 +08:00
JosonChan 600343eb08
[Feature] Support YOLOv5 instance segmentation (#735)
* add

* reproduce map

* add typehint and doc

* format code

* replace key

* add ut

* format

* format

* format code

* fix ut

* fix ut

* fix comment

* fix comment

* fix comment

* [WIP][Feature] Support yolov5-Ins training

* fix comment

* change data flow and fix loss_mask compute

* align the data pipeline

* remove albu gt mask key

* support yolov5 ins inference

* fix multi gpu test

* align the post_process with v8

* support training

* support training

* code formatting

* code formatting

* Support pad_param type (#672)

* add half_pad_param

* fix default fast_test

* fix loss weight compute

* fix mask rescale, add segment merge, fix segment2bbox

* fix clip and fix mask init

* code formatting

* code formatting

* code formatting

* code formatting

* [Fix] fix load image from file

* [Add] Add docs and more config

* [Fix] config type and test_formatting

* [Fix] fix yolov5-ins_m packdetinputs

* update

---------

Co-authored-by: Nioolek <379319054@qq.com>
Co-authored-by: Nioolek <40284075+Nioolek@users.noreply.github.com>
Co-authored-by: huanghaian <huanghaian@sensetime.com>
2023-04-27 14:47:52 +08:00
Haian Huang(深度眸) 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
Haian Huang(深度眸) 421be537e4
[Docs] Add Multi-scale training and testing (#630)
* add ms docs

* fix

* fix

* add en

* update

* update

* update

* update
2023-03-07 16:18:43 +08:00
Range King 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
Nioolek bda1050118
[Feature] YOLOv5 supports using mask annotation to optimize bbox (#565)
* add v5 config and readme

* fix config

* update config

* add remove mask

* update

* update

* fix

* update

---------

Co-authored-by: huanghaian <huanghaian@sensetime.com>
2023-03-06 17:39:32 +08:00
Haian Huang(深度眸) 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
Haian Huang(深度眸) a321cfae80
Fix box cam error (#601)
* fix box cam

* fix box cam
2023-02-28 15:22:56 +08:00
Haian Huang(深度眸) 1c833eb195
[Feature] Support TTA (#553)
* support TTA

* update note
2023-02-24 21:46:24 +08:00
Haian Huang(深度眸) 260509a574
[Update] Refactor Docs (#573)
* update get_started

* update

* update

* update

* update

* update

* update

* fix

* fix

* refactor en

* update

* update

* update
2023-02-21 17:19:01 +08:00
HinGwenWoong ff3e89809e
[DOC] Fix error link (#537)
* Fix error link

* Fix link

* Fix lint
2023-02-10 10:06:17 +08:00
Nioolek e966ce4e60
[Improve] Beauty RTMDet config (#531)
* beauty rtmdet config

* format
2023-02-09 19:08:21 +08:00
Range King c3acf42db4
Beautify the YOLOv8 configuration (#516)
* Update yolov5_s-v61_syncbn_8xb16-300e_coco.py

* Update yolov8_s_syncbn_fast_8xb16-500e_coco.py

* Update yolov8_m_syncbn_fast_8xb16-500e_coco.py

* Update yolov8_l_syncbn_fast_8xb16-500e_coco.py

* Update yolov8_s_syncbn_fast_8xb16-500e_coco.py

* Add todo

* Update yolov8_s_syncbn_fast_8xb16-500e_coco.py

* Update transforms.py
2023-02-06 19:39:39 +08:00
Haian Huang(深度眸) 79f0aae555
Beautify the YOLOv5 configuration (#501)
* refactor_config
2023-02-03 14:28:35 +08:00
Range King a631370cb3
Update model graphs (#442) 2023-01-07 17:44:19 +08:00
Haian Huang(深度眸) 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
Range King bb4aea90da
[Refactor] Refactor dataset metainfo to lowercase (#362)
* [Refactor] Refactor dataset metainfo to lowercase.

* Add docs

* Refactor metainfo to lowercase in new files

* Fix dataset meta compatibility
2022-12-27 21:19:39 +08:00
tripleMu 81b7942de1
[Feature] Support `iscrowd` attribute training in YOLOv5 (#260)
* [Feature] Support YOLOv7 inference (#149)

* update

* update

* update

* update

* update

* add docstr

* fix comments

* update

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

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

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

* Add changelog of v0.1.2 (#226)

* Add changelog of v0.1.2

* update version

* fix comments

* fix comments

* update

* update version

* update version

* Support ignore

Cal loss with ignore

Typo

Fix import from mmdet

Fix train_cfg is None

* Fix judgement

* Fix bug and add a demo config

* Add new config

* remove

* rm

* remove loadanno

* Try 0

* Try-2

* fix

* add avg_factor

* update

* support rectangle training

* add ignore cfg

* update

* delete

* Fix

* Add unitest

* Add ignore_iof_thr

* Add test

Co-authored-by: Haian Huang(深度眸) <huanghaian@pjlab.org.cn>
Co-authored-by: hanrui1sensetime <83800577+hanrui1sensetime@users.noreply.github.com>
Co-authored-by: RunningLeon <mnsheng@yeah.net>
Co-authored-by: huanghaian <huanghaian@sensetime.com>
Co-authored-by: HinGwenWoong <peterhuang0323@qq.com>
Co-authored-by: Haian Huang(深度眸) <1286304229@qq.com>
Co-authored-by: Range King <RangeKingHZ@gmail.com>
2022-12-27 14:18:10 +08:00
Haian Huang(深度眸) 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
HinGwenWoong 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
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
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
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
Haian Huang(深度眸) 3ea6edfeac Fix visualization bug in test_pipeline (#166) 2022-11-03 19:03:06 +08:00
q.yao 59f3d30fd2
Fix checkpoint link (#63)
* fix checkpoint link

* format docs
2022-09-22 10:48:31 +08:00
Haian Huang(深度眸) 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
HinGwenWoong 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
huanghaian a93cf7c3e4 Update albumentations requirements 2022-09-21 12:14:04 +08:00
huanghaian c683d1f94d Fix some config and add description 2022-09-21 10:12:00 +08:00
Range King 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.

* Update yolov5_description.md

Update logo

* Update model_design.md

Update logo

* Update model_design.md

Update logo

* Update README.md

Update logo

* Update README_zh-CN.md

Update logo

* Add GIF Link (#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
huanghaian 98712bb898 upload yolox_tiny and yolov5_n model 2022-09-20 10:07:57 +08:00
huanghaian 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
hha cfec074cf6 init commit 2022-09-18 10:11:55 +08:00