Commit Graph

23 Commits (178b0bf3ffec1353923460670bcbafccc15121cf)

Author SHA1 Message Date
LinXiaoZheng 4c1878ce8b [Improvement] dataset_analysis (#257)
* add ConcatDataset judgment

* Update dataset_analysis.py
2022-11-10 11:12:31 +08:00
wanghonglie 7ed550b6fc update dump results (#253) 2022-11-10 11:12:31 +08:00
LinXiaoZheng e510543442 add voc2012 download link (#251) 2022-11-10 11:12:31 +08:00
LinXiaoZheng e6e6f736e6 [Feature] Add dataset analysis script (#172)
* messages

* again

* again1

* again_2

* again_3

* again_4

* again_5

* Update docs/zh_cn/user_guides/useful_tools.md

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

* Update docs/zh_cn/user_guides/useful_tools.md

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

* Update docs/en/user_guides/useful_tools.md

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

* Update docs/en/user_guides/useful_tools.md

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

* Update docs/en/user_guides/useful_tools.md

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

* Update docs/zh_cn/user_guides/useful_tools.md

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

* Update tools/analysis_tools/dataset_analysis.py

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

* Update tools/analysis_tools/dataset_analysis.py

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

* Update tools/analysis_tools/dataset_analysis.py

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

* Update tools/analysis_tools/dataset_analysis.py

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

* modify code

* Update docs/en/user_guides/useful_tools.md

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

* Modify document

* Modify document

* new code

* revise decuments and codes

* Revise datails

* Update tools/analysis_tools/dataset_analysis.py

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

* modify func name

* code

* Documentation and code

* modify error meaasge

* deleted height,

Co-authored-by: HinGwenWoong <peterhuang0323@qq.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
AZhou 188377c62b [Fix] Fix resume is not set correctly bug when `cfg.resume` is True (#221)
* add optimize_anchors for mmyolo

* update optimize_anchors

* optimize anchors script

* add parameter in args

* add optimize anchor related docs

* fix lint error

* fix docs

* change anchors output format

* fix docs error

* fix resume set

* fix resume mode set

* add mmeval relate docs

Co-authored-by: imAzhou <lyzhou@creams.io>
2022-11-03 19:03:06 +08:00
AZhou 190ee5aaa7 [Feature] Add `optimize_anchors` script (#174)
* add optimize_anchors for mmyolo

* update optimize_anchors

* optimize anchors script

* add parameter in args

* add optimize anchor related docs

* fix lint error

* fix docs

* change anchors output format

* fix docs error

Co-authored-by: imAzhou <lyzhou@creams.io>
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
Haian Huang(深度眸) ce73b03b72 [Feature] Add `extract_subcoco` script (#186)
* add extract_subcoco.py

* update

* fix comment

* fix comment
2022-11-03 19:03:06 +08:00
Xin Li 714a7238cf [Feature] Add script converts yolo-style to coco format (#161)
* the script converts yolo-style to coco format draft 1

* Update tools/dataset_converters/yolo2coco.py

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

* Update tools/dataset_converters/yolo2coco.py

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

* Update tools/dataset_converters/yolo2coco.py

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

* Update tools/dataset_converters/yolo2coco.py

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

* Update tools/dataset_converters/yolo2coco.py

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

* Update tools/dataset_converters/yolo2coco.py

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

* Update tools/dataset_converters/yolo2coco.py

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

* typo fixed and update

* encaped and modified yolo2coco

* more usage instructions of yolo2coco

* Update docs/zh_cn/user_guides/useful_tools.md

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

* Update docs/en/user_guides/useful_tools.md

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

* Update tools/dataset_converters/yolo2coco.py

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

* Update tools/dataset_converters/yolo2coco.py

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

* Update docs/en/user_guides/useful_tools.md

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

* Update docs/zh_cn/user_guides/useful_tools.md

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

* Update docs/zh_cn/user_guides/useful_tools.md

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

* Update docs/en/user_guides/useful_tools.md

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

* final draft of yolo2coco

* final draft of yolo2coco

* final draft of yolo2coco

* revise the script and remove -split indicator

* Update docs/en/user_guides/useful_tools.md

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

* update the structural example of the root path

* Update tools/dataset_converters/yolo2coco.py

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

* fix typo on check_existence

* Update docs/zh_cn/user_guides/useful_tools.md

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

* Update tools/dataset_converters/yolo2coco.py

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

* Update tools/dataset_converters/yolo2coco.py

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

* Update tools/dataset_converters/yolo2coco.py

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

* Update tools/dataset_converters/yolo2coco.py

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

* update the commits

* rewrite split_by_files part and add copy_image part

* add instruction on using absolute image paths

* add instruction on using absolute image paths

* Update docs/en/user_guides/useful_tools.md

* Update tools/dataset_converters/yolo2coco.py

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

* Update docs/en/user_guides/useful_tools.md

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

* Update docs/en/user_guides/useful_tools.md

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

* Update tools/dataset_converters/yolo2coco.py

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

* Update tools/dataset_converters/yolo2coco.py

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

* Update tools/dataset_converters/yolo2coco.py

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

* Update tools/dataset_converters/yolo2coco.py

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

* add basic statistic func

* add basic statistic func

* revise the script

* remove copy&move image func

* Update docs/zh_cn/user_guides/useful_tools.md

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

* Update docs/en/user_guides/useful_tools.md

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

* Update docs/en/user_guides/useful_tools.md

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

* Update docs/en/user_guides/useful_tools.md

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

* Update docs/en/user_guides/useful_tools.md

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

* Update docs/en/user_guides/useful_tools.md

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

* Update tools/dataset_converters/yolo2coco.py

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

* revise image_dir

Co-authored-by: HinGwenWoong <peterhuang0323@qq.com>
Co-authored-by: Range King <RangeKingHZ@gmail.com>
Co-authored-by: Haian Huang(深度眸) <1286304229@qq.com>
2022-11-03 19:03:06 +08:00
Nioolek 5708864d5b [Enhance] YOLOv6 model convert (#165)
* add yolov5 voc training

* fix bug

* fix bug

* fix yolov6 model convert

* update yolov5 modelconvert

* rollback yolov5 model convert
2022-11-03 19:03:06 +08:00
Haian Huang(深度眸) 95f60784a5 [Feature] Support YOLOv7 inference (#149)
* update

* update

* update

* update

* update

* add docstr

* fix comments

* update
2022-11-03 19:03:06 +08:00
Haian Huang(深度眸) bb3aa48689 [Feature] Support RTMDet (#85)
* add cspnext

* add neck


* update

* align s,tiny test

* update

* update

* fix bug of UT

* fix typehint

* fix resize bug

* update s and tiny

* update s and tiny link

* update convert

* add cls config with rtmdet

* update config


* fix ci error

Co-authored-by: wanghonglie <wanghonglie@pjlab.org.cn>
2022-09-29 19:09:56 +08:00
wanghonglie f97941b4e8 support dump results (#84) 2022-09-29 19:09:56 +08:00
HinGwenWoong 1eb496cd46
[DOCS] Improve docs (#48)
* Improve docs

* Improve zh_cn docs

* Improve zh_cn docs

* Improve en docs

* Improve docs
2022-09-20 15:45:55 +08:00
huanghaian 66dcc4d3d2 fix import order 2022-09-20 10:57:33 +08:00
huanghaian a7d805d2b4 update version 2022-09-19 13:07:53 +08:00
hha cc6ac864c9 update featmap_vis_demo and format code 2022-09-18 17:04:14 +08:00
HinGwenWoong 1915c04e5d
[Fix] Improve type hint (#22)
* Fix type hint

* Delete `-> None` when no return val

* format code
2022-09-18 17:02:00 +08:00
古月闻星 71dfeb335f
[Feature] add download and convert script of dataset (#11) 2022-09-18 11:43:35 +08:00
wanghonglie b3d405aa4c
add model convert (#8) 2022-09-18 11:12:08 +08:00
hha cfec074cf6 init commit 2022-09-18 10:11:55 +08:00