Commit Graph

17 Commits (887d3ddfdc13d7f78b7a7394a17da65ca9c17794)

Author SHA1 Message Date
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
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(深度眸) 1c833eb195
[Feature] Support TTA (#553)
* support TTA

* update note
2023-02-24 21:46:24 +08:00
HinGwenWoong ff3e89809e
[DOC] Fix error link (#537)
* Fix error link

* Fix link

* Fix lint
2023-02-10 10:06:17 +08:00
Range King a631370cb3
Update model graphs (#442) 2023-01-07 17:44:19 +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
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
q.yao 59f3d30fd2
Fix checkpoint link (#63)
* fix checkpoint link

* format docs
2022-09-22 10:48:31 +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
hha cfec074cf6 init commit 2022-09-18 10:11:55 +08:00