Commit Graph

21 Commits (yolov6_face)

Author SHA1 Message Date
yechenzhi 6ecebdbbd5
Support yolox-pose based on mmpose (#694)
* 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

* add models

* add dataset1

* add dataset2

* add dataset3

* add configs

* re commit __init__

* re commit __init__

* re commit

* del local

* add typo

* del PoseToDetConverter and BBoxKeypoints

* del local changes

* fix mask rescale, add segment merge, fix segment2bbox

* fix pipeline

* add dataset

* fix typo

* add resize in mmyolo

* fix typo

* del local

* del local changes

* del local changes

* fix dir name

* fix dir name

* add FilterAnnotations

* fix typo

* new config for yolox-pose

* fix typo

* fix typo

* fix clip and fix mask init

* del pose dataset changes

* fix YOLOv5DetDataPreprocessor

* del local file

* fix typo

* del init_cfg

* simplify config

* fix batch size

* fix batch size

* fix typo

* code formatting

* code formatting

* code formatting

* code formatting

* fix bug for FilterAnnotations

* simpler way for FilterAnnotations

* update config

* [Fix] fix load image from file

* shorten eval time

* fix typo

* add large model

* [Add] Add docs and more config

* [Fix] config type and test_formatting

* [Fix] fix yolov5-ins_m packdetinputs

* hand rebase from yolov5-ins

* use new PackDetInputs

* rebase fix typo

* add mapping table

* fix typo

* add weight

* del typo

* del typo

* add results

* install mmpose, Keypoints note, context manager, predict, ota rename

* fix test

* add unittest for pose_sim_ota_assigner and yolox_head

* add unittest for pose_sim_ota_assigner and yolox_head

* fix typo

---------

Co-authored-by: Nioolek <379319054@qq.com>
Co-authored-by: josonchan <josonchan1998@163.com>
Co-authored-by: Nioolek <40284075+Nioolek@users.noreply.github.com>
Co-authored-by: huanghaian <huanghaian@sensetime.com>
2023-05-15 10:58:25 +08:00
Range King 887d3ddfdc
Update docs.txt (#766) 2023-05-10 11:35:15 +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
Yanyi Liu 4b996f10a5
Rotated object detection and RTMDet-R (#513)
* init

* add cfg

* update align

* update

* fix regularize box

* fix comment

* update config

* remove ckpt

* update

* make mmrotate optional

* fix doc

* add mmrotate req

* support large_demo with rbbox

* add ut

* update

* add doc v01

* update doc

* fix doc

* update

* update

* update readme

* update comments

* fix

* fix doc

* fix doc

* fix

* update

* update

* fix large

* update doc

* update readme

* fix config

* fix configs

* inprove

* update doc

* update assigner

* update ut

* remove rdsl assigner

* rename aug config

* speedup ut

* add comment

* fix data root

* remove doc

* remove empty folder

* add docs

* rename configs

* fix readme

* fix readme

* fix configs

* revert

* fix name

* fix table

* fix doc link

* fix doc link

* update

* update

* update

* Refactor dota splits

* add shapely

* fix typo

* fix ci

* change

* fix type

* uppdata link

* uppdata link

* add some comment

* update

---------

Co-authored-by: huanghaian <huanghaian@sensetime.com>
2023-03-02 10:27:46 +08:00
Haian Huang(深度眸) e9e007973e
[Feature] Support auto registry based on mmdet 3.0.0rc6 (#597)
* support auto registry

* update version

* update
2023-02-28 11:01:13 +08:00
Range King f2a576d153
[Docs] Refactor zh-CN docs (#568)
* refactor_docs_cn

* update

* update

* update

* add dataset_preparation

---------

Co-authored-by: huanghaian <huanghaian@sensetime.com>
2023-02-20 15:51:37 +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
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
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
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
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
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
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
Haian Huang(深度眸) 89588b60fb update mmengine/mmdet version and article (#201)
* update mmengine and mmdet version

* update
2022-11-03 19:03:06 +08:00
Nioolek d75fbabcb0 Add visualization of the base model to the docstring. (#196)
* add yolov5 voc training

* format code

* [Feature] Support YOLOv7 inference (#149)

* update

* update

* update

* update

* update

* add docstr

* fix comments

* update

* add video of featmap vis demo (#169)

* [Enhance] YOLOv6 model convert (#165)

* add yolov5 voc training

* fix bug

* fix bug

* fix yolov6 model convert

* update yolov5 modelconvert

* rollback yolov5 model convert

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

* add neck vis docstr

* add neck vis docstr

* add backbone model structure diagram

* fix readthedocs.txt

* fix doc vis

* fix vis

* fix vis

* fix vis

* fix vis

Co-authored-by: Haian Huang(深度眸) <huanghaian@pjlab.org.cn>
2022-11-03 19:03:06 +08:00
Haian Huang(深度眸) b18d5cf586 [Fix] Remove Albu version limit (#187)
* remove albu version limit

* fix lint
2022-11-03 19:03:06 +08:00
Haian Huang(深度眸) 2b644dab29 fix ci due to mmengine update to 0.2.0 (#155) 2022-10-13 17:39:52 +08:00
huanghaian 6882e09084 follow office guide of albumentations 2022-09-21 12:18:31 +08:00
huanghaian a93cf7c3e4 Update albumentations requirements 2022-09-21 12:14:04 +08:00
huanghaian 66dcc4d3d2 fix import order 2022-09-20 10:57:33 +08:00
hha cfec074cf6 init commit 2022-09-18 10:11:55 +08:00