Commit Graph

10 Commits (8c4d9dc503dc8e327bec8147e8dc97124052f693)

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(深度眸) 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 e00c3439f8
deprecating np.bool type alias (#396) 2022-12-21 10:11:13 +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
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
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