Commit Graph

16 Commits (0b48313dc757a26ad3087aa17f0991833bd7cc91)

Author SHA1 Message Date
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