Commit Graph

25 Commits (6ecebdbbd5b4456557f9a010559a590ec1474afd)

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
tang576225574 b4bf125d15
fix the browse_dataset for visualization of test and val (#641)
* correct the scale_factor and pad_param value in LetterResize

* correct the scale_factor and pad_param value in LetterResize

* correct the scale_factor and pad_param value in LetterResize

* tang

* update

* tang

* tang

* fix linting

* tang

* tang

* tang

* add print registers

* add print registers

* add print registers

* add print registers

* add print registers

* tang

* tang v1 finished

* tang

* print registers v1.2

* update

* tang

* update version

* update

* update

* remove exclude_prefix

* swap the val pipeline for visualization

* swap the val pipeline for visualization

* swap the val pipeline for visualization

* swap the val pipeline for visualization

* swap the val pipeline for visualization

---------

Co-authored-by: hha <1286304229@qq.com>
Co-authored-by: huanghaian <huanghaian@sensetime.com>
2023-03-08 11:04:26 +08:00
Haian Huang(深度眸) cdc885550a
[Feature] Add model zoo (#607)
* Update model_zoo.md

* fix

* update

* update

* update
2023-03-01 16:34:17 +08:00
tianlei d06de6d36f
[Feature] Support Calculate FLOPs (#603)
* support compute flops

* fix yolov8 error

* Add FLOPs Doc

* Update docs/zh_cn/recommended_topics/complexity_analysis.md

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

* Update get_flops.py

* Update complexity_analysis.md

* Update complexity_analysis.md

* Add EN `complexity_analysis.md`

* Update complexity_analysis.md

* Add `complexity_analysis.md` index

* Add `complexity_analysis.md` index

* Update README_zh-CN.md

* Update README.md

* update

---------

Co-authored-by: huanghaian <huanghaian@sensetime.com>
Co-authored-by: Haian Huang(深度眸) <1286304229@qq.com>
2023-03-01 10:06:09 +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
leling 87a2672c00
[Feature] Add confusion matrix drawing tool (#572)
* Add files via upload

* cat_single_channel_test

* [Feature] Add confusion matrix drawing tool

* [Feature] Add confusion matrix drawing tool

* [Feature] Add confusion matrix drawing tool

* [Feature] Add confusion matrix drawing tool
2023-02-23 10:08:54 +08:00
Range King 6bdfc0da26
[Feature] Add a tool for scheduler visualization (#479)
* Create vis_scheduler.py

* add docs

* Update .gitignore

* Update useful_tools.md

* add source link

* Add TODO
2023-01-18 15:03:02 +08:00
Yijie Zheng fd0f07b8ac
[Fix] Fix import error in `optimize_anchors.py` (#452)
* [Fix] Fix bugs in the output annotation file when  is set.

* Update labelme2coco.py

* Update labelme2coco.py

* Improve code

* [Fix]Fix import error in optimize_anchors.py

* [Fix]Fix import error in optimize_anchors.py

Co-authored-by: HinGwenWoong <peterhuang0323@qq.com>
2023-01-11 17:09:31 +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
Range King e00c3439f8
deprecating np.bool type alias (#396) 2022-12-21 10:11:13 +08:00
Range King 051d30a41a
[Fix] Fix CI error due to `np.int` and legacy builder.py (#389)
* Update utils.py

* fix docs

* Remove legacy build_detector

* Update test.yml

* fix error
2022-12-19 14:15:58 +08:00
HinGwenWoong 78a23ca8b9
[Fix] Fix `dataset_analysis` ProgressBar incorrect when `RepeatDataset` (#338)
* Fix ProgressBar incorrect when RepeatDataset

* Fix lint

* Improve doc

* Improve doc

* Improve import

* Fix hard code

* Fix hard code
2022-12-06 10:21:41 +08:00
HinGwenWoong 46b2494453
[Improvement] Improve args for `browse_dataset` and supprot `RepeatDataset` (#340)
* Improve doc and Fix bug

* Fix hard code

* Fix hard code

* Update browse_dataset.py
2022-12-06 10:19:49 +08:00
HinGwenWoong e3d95069ef
[Improvement] Improve args for `tools/analysis_tools/optimize_anchors.py` (#339)
* Improve args

* Improve doc

* Improve doc

* Improve doc
2022-12-04 17:18:33 +08:00
MingJian.L 1f06f4f594
[Improvement] `browse_dataset.py` (#304)
* Create browse_transform.py

Upgrate the brow_transform
对pipeline中的transform过程实现了可视化,以及可以将transform中key参数的变化打印出来。

* Update browse_dataset.py

* Delete browse_transform.py

* Update browse_dataset.py

* Update browse_dataset.py

* Update browse_dataset.py

* Update browse_dataset.py

* Update browse_dataset.py

* Update browse_dataset.py

* Update browse_dataset.py

* Update browse_dataset.py

* Update browse_dataset.py

修改了215行result_i = [result['dataset_sample'] for result in intermediate_imgs]通过lint

* fix some error

Co-authored-by: huanghaian <huanghaian@sensetime.com>
2022-12-01 10:26:00 +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
kitecats 748f151886
[Refactor] Refactor some of the functionality of `dataset_analysis` (#294)
* Refactor some of the functionality of data_analysis

* Remove redundant code in dataset_analysis.py

* Update tools/analysis_tools/dataset_analysis.py

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

* simplified code

* add docstring and simplify code

Co-authored-by: Haian Huang(深度眸) <1286304229@qq.com>
2022-11-22 11:21:35 +08:00
HinGwenWoong 47bb3ce408
[Improvement] Improve browse coco json script args (#309)
* Improve browse coco json script

* Improve doc

* Improve script

* Improve doc

* Improve doc

* Improve doc
2022-11-22 10:51:42 +08:00
LinXiaoZheng 4c1878ce8b [Improvement] dataset_analysis (#257)
* add ConcatDataset judgment

* Update dataset_analysis.py
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
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
huanghaian 66dcc4d3d2 fix import order 2022-09-20 10:57:33 +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
hha cfec074cf6 init commit 2022-09-18 10:11:55 +08:00