Commit Graph

70 Commits (260509a5741fce3ee83646906b533497b3b8ba05)

Author SHA1 Message Date
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
Nioolek 75fc8fc2a3
[Feature] YOLOv8 supports using mask annotation to optimize bbox (#484)
* add cfg

* add copypaste

* add todo

* 在mosaic和mixup中处理gt_masks,改config

* fix cat bug

* add finetune box in affine

* add repr

* del albu config in l

* add doc

* add config

* format code

* fix loadmask

* addconfig,fix mask

* fix loadann

* fix tra

* update LoadAnnotations

* update

* support mask

* fix error

* fix error

* fix config and no maskrefine bug

* fix

* fix

* update config

* format code

* beauty config

* add yolov5 config and readme

* beauty yolov5 config

* add ut

* fix ut. bitmap 2 poly

* fix ut and add mix transform ut.

* fix bool

* fix loadann

* rollback yolov5

* rollback yolov5

* format

* 提高速度

* update

---------

Co-authored-by: huanghaian <huanghaian@sensetime.com>
2023-02-20 11:11:13 +08:00
Haian Huang(深度眸) 6400fba1af
[Enchance] Optimize and accelerate YOLOX with RTMDet hyps (#542)
* enchance yolox

* update

* update

* fix

* fix

* fix lint
2023-02-17 11:27:03 +08:00
Youfu 6f0a765ea8
Update README.md (#552)
modify yolox_loss part
2023-02-15 10:57:22 +08:00
Range King 8d2a7d1ab2
Update rtmdet_tiny_ofa_lat31_syncbn_16xb16-300e_coco.py (#551) 2023-02-13 19:42:49 +08:00
Yue Sun 164c319493
[Feature] Support MMRazor searchable backbone (#453)
* update subnet cfg

* add docs

* update model link

* fix lint

* mdformat

* update readme

* fix lint

* update link

* rename folder

* fix readme

* update readme

* make lint

* rename

* update readme

* sync mmrazor cfg

* fix cfg

* install issue

* require mmcls

* fix yolo cfg

---------

Co-authored-by: aptsunny <aptsunny@tongji.edu.cn>
Co-authored-by: sunyue1 <sunyue1@sensetime.com>
2023-02-10 16:24:47 +08:00
Youfu 3a6899e232
[Improve] Beautify the YOLOX configuration (#529)
* Beautify the YOLOX configuration

* fix checks

* Update configs/yolox/yolox_s_fast_8xb8-300e_coco.py

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

* fix letter case problem

* beauty yolox configs except yolox_s's config

* fix lint

* Update configs/yolox/yolox_s_fast_8xb8-300e_coco.py

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

* fix yolox_s yolox_tiny

* fix tiny

* fix tiny

* simple tiny

---------

Co-authored-by: HinGwenWoong <peterhuang0323@qq.com>
Co-authored-by: Haian Huang(深度眸) <1286304229@qq.com>
2023-02-10 10:50:59 +08:00
tianlei 4e8bf17c90
[Improve] Beautify the YOLOv7 configuration (#506)
* Beautify the YOLOv7 configuration

* yolov7 config

* Update configs/yolov7/yolov7_l_syncbn_fast_8x16b-300e_coco.py

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

* Update configs/yolov7/yolov7_tiny_syncbn_fast_8x16b-300e_coco.py

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

* Update configs/yolov7/yolov7_w-p6_syncbn_fast_8x16b-300e_coco.py

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

* Update configs/yolov7/yolov7_w-p6_syncbn_fast_8x16b-300e_coco.py

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

* Update configs/yolov7/yolov7_w-p6_syncbn_fast_8x16b-300e_coco.py

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

* Update configs/yolov7/yolov7_w-p6_syncbn_fast_8x16b-300e_coco.py

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

* Beautify the YOLOv7 configuration

---------

Co-authored-by: Haian Huang(深度眸) <1286304229@qq.com>
2023-02-10 10:18:57 +08:00
HinGwenWoong d8b1353691
[Improve] Beautify YOLOv6 all configs (#539)
* Beauty YOLOv6 config

* Beauty YOLOv6 config

* Beauty config

* Beauty config
2023-02-10 10:13:47 +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
Youfu 2813e89f44
[Feature] Implement fast version of YOLOX (#518)
* Implement fast version of YOLOX

* config change

* Update yolox_head.py

* Update mmyolo/models/data_preprocessors/data_preprocessor.py

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

* Update mmyolo/models/data_preprocessors/data_preprocessor.py

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

* add test and modify faults

* fix lint

* fix lint

* modify metafile and README

* modify metafile and readme

* fix

* fix

* fix

* fix

* fix

* fix test

---------

Co-authored-by: Haian Huang(深度眸) <1286304229@qq.com>
2023-02-08 20:10:03 +08:00
tripleMu e90c369f9d
[Docs] Fix links in YOLOv8 README (#517)
* Update link

* Fix
2023-02-07 10:09:51 +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
Audrey528 5f2b08cf5a
add model structure graph of yolov7 in readme (#504) 2023-02-03 20:01:05 +08:00
Haian Huang(深度眸) 79f0aae555
Beautify the YOLOv5 configuration (#501)
* refactor_config
2023-02-03 14:28:35 +08:00
Youfu da62ae7283
[Doc] Add PPYOLOE model structure (#471)
* Update README.md

Add PPYOLOE model structure

* Update README.md

* Update README.md

* Update README.md

* Update README.md

* Update README.md

* Update README.md
2023-01-18 14:32:20 +08:00
Nioolek 07ab35222c
[Feature] Support YOLOv8 training (#458)
* add v8 training head

* add training config

* update

* update1

* update all config

* update all config

* update metafile

* add head loss

* update readme

* iou2ciou in tal, add config

* change init layer

* update

* update

* add clip_grad

* update max_aspect_ratio

* add init

* fix aug config

* update forward

* add official v8 loss

* add official v8 loss

* 同步预测方式

* fix loss bug

* delete useless code

* 替换iou,但还有问题

* fix bug

* fix empty

* add UT

* add UT

* update readme

* replace iou_cal

* add use_ciou docstr

* update metafile

* update metafile

* update metafile

Co-authored-by: huanghaian <huanghaian@sensetime.com>
Co-authored-by: hha <1286304229@qq.com>
2023-01-18 14:16:37 +08:00
Range King e66991fc6e
[Fix] Fix YOLOv8 graph and a lint issue (#459)
* add missing bg frame

* fix easydeploy lint
2023-01-12 16:00:46 +08:00
Range King 550664d514
[Doc] Add the model structure graph of YOLOv8-P5 (#455)
* add YOLOv8 model structure figure

* Fix the layer name
2023-01-12 10:07:06 +08:00
HinGwenWoong 935d710f79
[Feature] Support YOLOv8 inference (#445)
* Add backbone

* Improve code

* fix

* Add neck

* Add config

* Fix layer param

* Fix layer param

* Add head

* Add head

* Add model converter

* Add model converter

* Add model converter

* Improve code

* update

* update

* align test

* Improve code

* Improve code

* Improve code

* Fix lint

* Fix lint

* Fix lint

* Fix lint

* Improve code

* Improve code

* Improve code

* Add configs

* Improve doc

* Improve doc

* update

* Fix config

* update

* Fix config

* Fix config

* update

* Fix config epoch

* update

* Fix docstr

* Add UT

* Add UT

* Fix doc

* update

* Fix lint

* Fix doc

* Fix config name

* Improve config

* Improve default

* Add docstr

* Improve code

* Improve code

* Drop `bbox_head.head_module.dfl.conv.weight` when convert to mmyolo weight

* Delete useless code

* Add batch_shapes_cfg but not enable it.

* update

Co-authored-by: huanghaian <huanghaian@sensetime.com>
2023-01-11 19:01:06 +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
Nioolek 8127805dd3
[Feature] Support PPYOLOE training (#259)
* add ppyoloe backbone, neck

* add ppyoloe test

* add docstring

* add ppyoloe m/l/x configfile

* add ppyoloe_coco.py

* rename config

* add typehint

* format code; add ut

* add datapre

* add datapre

* add ppyoloe datapre

* add ppyoloe datapre

* add ppyoloe datapre

* reproduce coco v0.1

* add ut

* add ut, docstring

* fix transforms bug

* use mmdet dfloss

* add non plus model config

* add non plus model config

* fix

* add ut

* produce coco v0.2

* fix config

* fix config

* fix eps and transforms bug

* add ema

* fix resize

* fix transforms.py

* fix transforms.py

* fix transforms.py

* old version

* old version

* old version

* old version

* old version

* old version

* fix stride loss error

* add INTER_LANCZOS4

* fix crop bug

* init commit

* format code

* format code

* bgr transforms.py

* add typehint and doc in transforms.py

* 继承新版yolov6head写法,删除不必要的注释

* fix transforms var name bug

* bbox decode use stridetensor insted of priors

* add headmodule todo

* add ppyoloe README.md

* add ppyoloe README.md

* Update tests/test_datasets/test_transforms/test_transforms.py

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

* Update tests/test_datasets/test_transforms/test_transforms.py

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

* save ckpt last 10 epochs

* save_best ckpt

* del ppyoloe collate

* change name of ppyoloebatchrandomresize

* add the reason for rewritten PPYOLOEDetDataPreprocessor

* solve ppyoloerandomresize name error

* rm PPYOLOERandomExpand

* rm l1 loss

* rm yolov6 loss_obj

* add difference between yolov6 and ppyoloe

* add reason for rewrite paramscheduler

* change proj init way

* fix error

* rm proj_conv in pth

* format code

* add load_from

* update

* support fast training

* add pretrained model url

* update

* add pretrained model url

* fix error

* add imagenet model convert and use init_cfg to init backbone

* add plus model pretrain model

* add ut

* add ut

* fix ut

* fix withstride bug

* cat in yolov5_collate

* merge

* fix typehint

* update readme

* add reason for gap

* fix log in README.md

* rollback yolov6

* change inherit

* fix ut

* fix ut

Co-authored-by: Range King <RangeKingHZ@gmail.com>
Co-authored-by: hha <1286304229@qq.com>
Co-authored-by: huanghaian <huanghaian@sensetime.com>
2023-01-06 15:54:39 +08:00
RangiLyu 5172805dae
fix rtmdet-s cfg and neck (#434) 2023-01-06 11:40:19 +08:00
RangiLyu 48f8896e84
[Feature] Implement fast version of RTMDet. (#425)
* Accelerate RTMDet

* update

* update

* update

* update1

* update2

* update pipeline

* update lr cudnnbenchmark

* revert batchsize

* fix batch inference

* refactor head

* update box

* bs=16

* update

* move reduce mean

* update head

* per img loss

* fix

* fix sum

* concat loss

* batch dsla

* sort topk

* bs 32

* clean code

* update readme

* update ut

* update checkpoint

* num_class

* clean code

* resolve comments

* fix readme

* fix ut

Co-authored-by: huanghaian <huanghaian@sensetime.com>
Co-authored-by: hha <1286304229@qq.com>
2023-01-04 16:06:44 +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
Xin Li ab5665e840
update base v6 config name (#398) 2022-12-27 11:01:17 +08:00
Youfu b012154c10
CodeCamp #1519 [Docs] add yolox structure description (#402)
* add yolox structure description

* Update configs/yolox/README.md

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

Co-authored-by: Range King <RangeKingHZ@gmail.com>
2022-12-26 14:15:23 +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
RunningLeon dcf73091e5
add mmyolo regression test yml (#359) 2022-12-12 10:27:21 +08:00
HinGwenWoong 2bf502b13c
[Improvement] Improve doc `custom_dataset` (#336)
* Improve doc

* Improve doc

* Add config

* Add config

* Add config

* Improve doc

* Improve doc

* Improve doc

* Improve doc

* Improve doc index

* 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

* Add deploy demo

* Improve doc

* Fix lint

* Improve doc

* Improve doc

* Improve doc

* Improve doc

* Improve doc

* Improve code

* Improve code

* Improve code

* Improve code

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

* Improve doc

* Improve doc

* Improve doc

* Improve doc

* Improve doc

* Improve doc

* Improve doc

* Improve doc

* Improve doc

* Improve doc

* Improve doc
2022-12-12 10:16:03 +08:00
CaiLiYuan 9985e0bc08
Update yolov7_l_syncbn_fast_8x16b-300e_coco.py (#333)
change:model->bbox_head->head_module->dict(num_classes=num_classes)
2022-12-02 16:31:27 +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
Haian Huang(深度眸) e85ddeac0d
[Feature] Support YOLOv7 P6 training (#310)
* support p6 train

* fix bug

* add readme

* add link
2022-11-30 18:45:08 +08:00
hanrui1sensetime 4a8699d6fe
[Fix] Enable switch to deploy when create pytorch model in deployment. (#324)
* support switch_to_deploy when deploy

* fix docformatter

* fix README for yolox shape
2022-11-29 18:23:12 +08:00
AllentDan 68c9fd4745
[Feature] Support YOLOv5 deployment at RV1126 device (#262)
* add yolov5 head rewrite and rknn config

* type hint

* remove useless

* refactor config
2022-11-29 16:54:08 +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
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
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
Xiaolin Wang 71e5c793e5 [Doc] Fix typo (#270)
* fix typo

* fix ci
2022-11-10 11:12:31 +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
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
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 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
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 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