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
600343eb08
[Feature] Support YOLOv5 instance segmentation ( #735 )
...
* 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
* fix mask rescale, add segment merge, fix segment2bbox
* fix clip and fix mask init
* code formatting
* code formatting
* code formatting
* code formatting
* [Fix] fix load image from file
* [Add] Add docs and more config
* [Fix] config type and test_formatting
* [Fix] fix yolov5-ins_m packdetinputs
* update
---------
Co-authored-by: Nioolek <379319054@qq.com>
Co-authored-by: Nioolek <40284075+Nioolek@users.noreply.github.com>
Co-authored-by: huanghaian <huanghaian@sensetime.com>
2023-04-27 14:47:52 +08:00
9f3adc426f
[FEATURE] Support YOLOv6 3.0 inference ( #744 )
...
* [FEATURE] Support YOLOv6 3.0 inference
* add CSPSPPFBottleneck module for YOLOv6 3.0 backbone
* add BiFusion module, YOLOv6RepBiPAFPN module for YOLOv6 3.0 neck
* modify YOLOv6HeadModule to support YOLOv6 3.0 head
* add a yolov6v3 l/m/s/t/n conifgs
* [Fix] Modify YOLOv6 3.0 neck
* Modify YOLOv6RepBiPAFPN
* Add unit tests
* [Fix] Modify configs
* Rename yolov6_v3 configs
* Fix a bug in building BiFusion Module
* Add a checkpoint convert script
2023-04-25 10:22:48 +08:00
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
dab054875d
optimize the convert_kd_ckpt_to_student.py file ( #647 )
...
* 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
* optimize the convert_kd_ckpt_to_student file
* optimize the convert_kd_ckpt_to_student file
* optimize the tools/model_converters/convert_kd_ckpt_to_student.py file
---------
Co-authored-by: hha <1286304229@qq.com>
Co-authored-by: huanghaian <huanghaian@sensetime.com>
2023-03-09 19:07:29 +08:00
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
69b43e696d
[Feature] Support YOLOv5u inference ( #624 )
...
* support yolov5u inference
* minor fix
* Update README.md
* Add all configs
* unify config links
2023-03-07 10:23:13 +08:00
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
cdc885550a
[Feature] Add model zoo ( #607 )
...
* Update model_zoo.md
* fix
* update
* update
* update
2023-03-01 16:34:17 +08:00
6f38b781bd
[Feature] Add RTMDet distillation cfg ( #544 )
...
* add rtm distillation cfg
* rename the cfg file
* use norm connector
* fix cfg
* fix cfg
* support rtm distillation
* fix readme and cfgs
* fix readme
* add docstring
* add links of ckpts and logs
* Update configs/rtmdet/README.md
Co-authored-by: RangiLyu <lyuchqi@gmail.com>
* fix cfgs
* rename stop distillation hook
* rename stop_epoch
* fix cfg
* add model converter
* add metafile and tta results
* fix metafile
* fix readme
* mv distillation/metafile to metafile
---------
Co-authored-by: RangiLyu <lyuchqi@gmail.com>
2023-03-01 16:20:45 +08:00
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
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
1c833eb195
[Feature] Support TTA ( #553 )
...
* support TTA
* update note
2023-02-24 21:46:24 +08:00
e08e9895b0
[Docs] Add FAQ and Troubleshooting steps ( #579 )
...
* add FAQ
* update
* update
* update
* update
* update
* update
2023-02-23 16:38:08 +08:00
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
cbadd3abe4
add yolov7-e6e weight converter ( #570 )
...
* add yolov7-e6e weight converter
* add yolov7-e6e weight converter
* add yolov7-e6e weight converter
* fix format
* fix format
2023-02-19 21:48:32 +08:00
ccb8315ff3
[Enchance] judge `metainfo` is lower ( #535 )
...
* judge metainfo is lower
* update
* update
* update
2023-02-09 09:46:59 +08:00
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
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
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
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
2290713ae6
[Fix] Fix bugs in the output annotation file when `--class-id-txt` is set. ( #430 )
...
* [Fix] Fix bugs in the output annotation file when is set.
* Update labelme2coco.py
* Update labelme2coco.py
* Improve code
Co-authored-by: HinGwenWoong <peterhuang0323@qq.com>
2023-01-06 11:41:17 +08:00
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
e00c3439f8
deprecating np.bool type alias ( #396 )
2022-12-21 10:11:13 +08:00
1a0dfb0203
[Enhance]: Add resume from specific checkpoint in CLI ( #393 )
2022-12-20 10:25:19 +08:00
446dfbd4b9
Add raise when class is blank or space between words ( #391 )
2022-12-19 16:55:33 +08:00
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
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
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
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
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
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
60a515f41f
[Enhance] Add option of json output in `test.py` ( #316 )
...
* add json args
* Update how_to.md
* Update test.py
* Update how_to.md
* Update how_to.md
2022-11-26 19:59:37 +08:00
4b79e4f983
[Feature] Add condition for extract coco datasets ( #286 )
...
* add condition for extract coco datasets
* fix lint error
* guarantee images be extracted definitely has annotation info
* reset num_img default value
2022-11-26 16:19:04 +08:00
daae790075
[Feature] Add split COCO dataset script ( #311 )
...
* Add coco_sqlit script
* rename file
* Improve code
* Improve code
* Improve coding
* Improve file structure
* ratios normalize
2022-11-26 09:27:30 +08:00
285c395066
[Improvement] Add output `class_with_id.txt` after labelme2coco ( #313 )
...
* Add output class_with_id.json
* Add `--class-id-json`
* Improve coding, using txt and saving txt
* Improve coding
2022-11-25 16:50:50 +08:00
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
b1e478a8bd
[Feature] Add `labelme2coco` script ( #308 )
...
* Add label2coco script
* Add label2coco script
* Add label2coco script
* Add label2coco script
* Improve script
* Improve script
2022-11-23 10:32:52 +08:00
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
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
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
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
4c1878ce8b
[Improvement] dataset_analysis ( #257 )
...
* add ConcatDataset judgment
* Update dataset_analysis.py
2022-11-10 11:12:31 +08:00
7ed550b6fc
update dump results ( #253 )
2022-11-10 11:12:31 +08:00
e510543442
add voc2012 download link ( #251 )
2022-11-10 11:12:31 +08:00
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
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
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
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
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
yechenzhi
JosonChan
Qingren
Haian Huang(深度眸)
tang576225574
Range King
Yanyi Liu
whcao
tianlei
leling
HinGwenWoong
Yijie Zheng
Nioolek
Yue WANG 王跃
MingJian.L
AZhou
kitecats
Haian Huang(深度眸)
wanghonglie
LinXiaoZheng