Commit Graph

64 Commits (887d3ddfdc13d7f78b7a7394a17da65ca9c17794)

Author SHA1 Message Date
JosonChan 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
Qingren 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
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
tang576225574 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
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
Range King 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
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(深度眸) cdc885550a
[Feature] Add model zoo (#607)
* Update model_zoo.md

* fix

* update

* update

* update
2023-03-01 16:34:17 +08:00
whcao 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
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
Haian Huang(深度眸) 1c833eb195
[Feature] Support TTA (#553)
* support TTA

* update note
2023-02-24 21:46:24 +08:00
Haian Huang(深度眸) e08e9895b0
[Docs] Add FAQ and Troubleshooting steps (#579)
* add FAQ

* update

* update

* update

* update

* update

* update
2023-02-23 16:38:08 +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
yechenzhi 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
Haian Huang(深度眸) ccb8315ff3
[Enchance] judge `metainfo` is lower (#535)
* judge metainfo is lower

* update

* update

* update
2023-02-09 09:46:59 +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
HinGwenWoong 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
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
Nioolek 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
Yijie Zheng 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
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
Yue WANG 王跃 1a0dfb0203
[Enhance]: Add resume from specific checkpoint in CLI (#393) 2022-12-20 10:25:19 +08:00
HinGwenWoong 446dfbd4b9
Add raise when class is blank or space between words (#391) 2022-12-19 16:55:33 +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 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
Range King 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
AZhou 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
HinGwenWoong 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
HinGwenWoong 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
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
HinGwenWoong 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
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
Haian Huang(深度眸) 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
wanghonglie 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
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