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
30cc772524
Razor backbone cfg & log update ( #610 )
...
* razor cfg update
* formatter
* update readme
* readme format
* fix test ci
* remove log
---------
Co-authored-by: sunyue1 <sunyue1@sensetime.com>
2023-03-02 17:01:41 +08:00
146cd930c5
[Feature] Support RTMDet Ins Segmentation Inference ( #583 )
...
* update config
* reproduce map in mmyolo
* reproduce map in mmyolo
* collate fn、process加mask
* reproduce result
* beauty code
* beauty code
* beauty code
* del yolov5_seg_head.py
* beauty config
* add doc and typehint
* del objectness
* fix ut; add empty res
2023-03-02 15:11:09 +08:00
4faea33ab5
change name and remove loss_obj ( #613 )
2023-03-02 12:44:17 +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
75fc8fc2a3
[Feature] YOLOv8 supports using mask annotation to optimize bbox ( #484 )
...
* add cfg
* add copypaste
* add todo
* 在mosaic和mixup中处理gt_masks,改config
* fix cat bug
* add finetune box in affine
* add repr
* del albu config in l
* add doc
* add config
* format code
* fix loadmask
* addconfig,fix mask
* fix loadann
* fix tra
* update LoadAnnotations
* update
* support mask
* fix error
* fix error
* fix config and no maskrefine bug
* fix
* fix
* update config
* format code
* beauty config
* add yolov5 config and readme
* beauty yolov5 config
* add ut
* fix ut. bitmap 2 poly
* fix ut and add mix transform ut.
* fix bool
* fix loadann
* rollback yolov5
* rollback yolov5
* format
* 提高速度
* update
---------
Co-authored-by: huanghaian <huanghaian@sensetime.com>
2023-02-20 11:11:13 +08:00
164c319493
[Feature] Support MMRazor searchable backbone ( #453 )
...
* update subnet cfg
* add docs
* update model link
* fix lint
* mdformat
* update readme
* fix lint
* update link
* rename folder
* fix readme
* update readme
* make lint
* rename
* update readme
* sync mmrazor cfg
* fix cfg
* install issue
* require mmcls
* fix yolo cfg
---------
Co-authored-by: aptsunny <aptsunny@tongji.edu.cn>
Co-authored-by: sunyue1 <sunyue1@sensetime.com>
2023-02-10 16:24:47 +08:00
82d288f5d4
[Improve] Beauty YOLOv6 head `gt_instances_preprocess` ( #532 )
...
* beauty yolov6 head preprocess
* add >0
* format
* add ut
* move path
* format
2023-02-09 19:05:28 +08:00
2813e89f44
[Feature] Implement fast version of YOLOX ( #518 )
...
* Implement fast version of YOLOX
* config change
* Update yolox_head.py
* Update mmyolo/models/data_preprocessors/data_preprocessor.py
Co-authored-by: Haian Huang(深度眸) <1286304229@qq.com>
* Update mmyolo/models/data_preprocessors/data_preprocessor.py
Co-authored-by: Haian Huang(深度眸) <1286304229@qq.com>
* add test and modify faults
* fix lint
* fix lint
* modify metafile and README
* modify metafile and readme
* fix
* fix
* fix
* fix
* fix
* fix test
---------
Co-authored-by: Haian Huang(深度眸) <1286304229@qq.com>
2023-02-08 20:10:03 +08:00
f54e5603fd
[Enchance] Extending `DetDataPreprocessor` flexibility ( #511 )
...
* Extending DetDataPreprocessor flexibility
* update
* add cast_data
2023-02-06 17:47:22 +08:00
07ab35222c
[Feature] Support YOLOv8 training ( #458 )
...
* add v8 training head
* add training config
* update
* update1
* update all config
* update all config
* update metafile
* add head loss
* update readme
* iou2ciou in tal, add config
* change init layer
* update
* update
* add clip_grad
* update max_aspect_ratio
* add init
* fix aug config
* update forward
* add official v8 loss
* add official v8 loss
* 同步预测方式
* fix loss bug
* delete useless code
* 替换iou，但还有问题
* fix bug
* fix empty
* add UT
* add UT
* update readme
* replace iou_cal
* add use_ciou docstr
* update metafile
* update metafile
* update metafile
Co-authored-by: huanghaian <huanghaian@sensetime.com>
Co-authored-by: hha <1286304229@qq.com>
2023-01-18 14:16:37 +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
884330108d
[BC Breaking] Keep the same input shape order as mmdet 3.0.0rc5 ( #436 )
...
* Keep the same input shape order as mmdet
* add desc
* update
* update
2023-01-06 19:00:30 +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
2f399df2d6
[Feature] Add the unit test to test the `scale_factor` in LetterResize ( #408 )
...
* 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
Co-authored-by: hha <1286304229@qq.com>
2023-01-06 14:40:21 +08:00
48f8896e84
[Feature] Implement fast version of RTMDet. ( #425 )
...
* Accelerate RTMDet
* update
* update
* update
* update1
* update2
* update pipeline
* update lr cudnnbenchmark
* revert batchsize
* fix batch inference
* refactor head
* update box
* bs=16
* update
* move reduce mean
* update head
* per img loss
* fix
* fix sum
* concat loss
* batch dsla
* sort topk
* bs 32
* clean code
* update readme
* update ut
* update checkpoint
* num_class
* clean code
* resolve comments
* fix readme
* fix ut
Co-authored-by: huanghaian <huanghaian@sensetime.com>
Co-authored-by: hha <1286304229@qq.com>
2023-01-04 16:06:44 +08:00
81b7942de1
[Feature] Support `iscrowd` attribute training in YOLOv5 ( #260 )
...
* [Feature] Support YOLOv7 inference (#149 )
* update
* update
* update
* update
* update
* add docstr
* fix comments
* update
* [Feature] support mmyolo deployment (#79 )
* support mmyolo deployment
* mv deploy place
* remove unused configs
* add deploy code
* fix new register
* fix comments
* fix dependent codebase register
* remove unused initialize
* refact deploy config
* credit return to triplemu
* Add yolov5 head rewrite
* refactor deploy
* refactor deploy
* Add yolov5 head rewrite
* fix configs
* refact config
* fix comment
* sync name after mmdeploy 1088
* fix mmyolo
* fix yapf
* fix deploy config
* try to fix flake8 importlib-metadata
* add mmyolo models ut
* add deploy uts
* add deploy uts
* fix trt dynamic error
* fix multi-batch for dynamic batch value
* fix mode
* fix lint
* sync model.py
* add ci for deploy test
* fix ci
* fix ci
* fix ci
* extract script to command for fixing CI
* fix cmake for CI
* sudo ln
* move ort position
* remove unused sdk compile
* cd mmdeploy
* simplify build
* add missing make
* change order
* add -v
* add setuptools
* get locate
* get locate
* upgrade torch
* change torchvision version
* fix config
* fix ci
* fix ci
* fix lint
Co-authored-by: tripleMu <gpu@163.com>
Co-authored-by: RunningLeon <mnsheng@yeah.net>
* [Feature] Support YOLOv5 YOLOv6 YOLOX Deploy in mmdeploy (#199 )
* Support YOLOv5 YOLOv6 YOLOX Deploy in mmdeploy
* Fix lint
* Rename _class to detector_type
* Add some common
* fix lint
Co-authored-by: huanghaian <huanghaian@sensetime.com>
* [DOC] Add deploy guide doc (#220 )
* Add deploy doc-0
* Update 部署必备指南.md
Co-authored-by: HinGwenWoong <peterhuang0323@qq.com>
* Update 部署必备指南.md
Co-authored-by: HinGwenWoong <peterhuang0323@qq.com>
* Update 部署必备指南.md
Co-authored-by: HinGwenWoong <peterhuang0323@qq.com>
* Update 部署必备指南.md
Co-authored-by: HinGwenWoong <peterhuang0323@qq.com>
* Update 部署必备指南.md
Co-authored-by: HinGwenWoong <peterhuang0323@qq.com>
* Update 部署必备指南.md
Co-authored-by: HinGwenWoong <peterhuang0323@qq.com>
* Update 部署必备指南.md
Co-authored-by: HinGwenWoong <peterhuang0323@qq.com>
* Update 部署必备指南.md
Co-authored-by: HinGwenWoong <peterhuang0323@qq.com>
* Update 部署必备指南.md
Co-authored-by: HinGwenWoong <peterhuang0323@qq.com>
* Del unused line
* Add docs
* Fix
* Fix
* Rename docs
* Rename docs
Co-authored-by: HinGwenWoong <peterhuang0323@qq.com>
* [Feautre] Add deploy dockerfile (#224 )
* Add dockerfile for deploy
* Fix
* Update docker/Dockerfile_deployment
Co-authored-by: HinGwenWoong <peterhuang0323@qq.com>
* Update docker/Dockerfile_deployment
Co-authored-by: HinGwenWoong <peterhuang0323@qq.com>
* Add opencv old version
* Add dockerfile and fix some typo
* Remove repeat packages
* Fix undefined symbol bug
* Update docker/Dockerfile_deployment
Co-authored-by: Haian Huang(深度眸) <1286304229@qq.com>
* Update docker/Dockerfile_deployment
Co-authored-by: HinGwenWoong <peterhuang0323@qq.com>
* Add docs for deploy dockerfile
* Fix typo
* Update docs/zh_cn/advanced_guides/yolov5_deploy.md
Co-authored-by: HinGwenWoong <peterhuang0323@qq.com>
* Add profiler
* Update docs/zh_cn/advanced_guides/yolov5_deploy.md
Co-authored-by: HinGwenWoong <peterhuang0323@qq.com>
* Update docker/Dockerfile_deployment
Co-authored-by: Range King <RangeKingHZ@gmail.com>
* Remove mmcv-full
Co-authored-by: HinGwenWoong <peterhuang0323@qq.com>
Co-authored-by: Haian Huang(深度眸) <1286304229@qq.com>
Co-authored-by: Range King <RangeKingHZ@gmail.com>
* Add changelog of v0.1.2 (#226 )
* Add changelog of v0.1.2
* update version
* fix comments
* fix comments
* update
* update version
* update version
* Support ignore
Cal loss with ignore
Typo
Fix import from mmdet
Fix train_cfg is None
* Fix judgement
* Fix bug and add a demo config
* Add new config
* remove
* rm
* remove loadanno
* Try 0
* Try-2
* fix
* add avg_factor
* update
* support rectangle training
* add ignore cfg
* update
* delete
* Fix
* Add unitest
* Add ignore_iof_thr
* Add test
Co-authored-by: Haian Huang(深度眸) <huanghaian@pjlab.org.cn>
Co-authored-by: hanrui1sensetime <83800577+hanrui1sensetime@users.noreply.github.com>
Co-authored-by: RunningLeon <mnsheng@yeah.net>
Co-authored-by: huanghaian <huanghaian@sensetime.com>
Co-authored-by: HinGwenWoong <peterhuang0323@qq.com>
Co-authored-by: Haian Huang(深度眸) <1286304229@qq.com>
Co-authored-by: Range King <RangeKingHZ@gmail.com>
2022-12-27 14:18:10 +08:00
46e7f6a477
[BUG] Fix the scale_factor and pad_param value in LetterResize ( #387 )
...
* 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
Co-authored-by: hha <1286304229@qq.com>
2022-12-27 11:45:23 +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
00b138a184
[DOC] Add new video link for `custom dataset` ( #365 )
...
* Add new video link
* Add new video link
* Add new video link
* Add new video link
* Add new video link
* Add new video link
* Add new video link
* Add new video link
* Add new video link
* Add new video link
* Add new video link
* Add new video link
* Add new video link
* Add new video link
* Add new video link
* Add new video link
* Add new video link
* Add new video link
* Add new video link
* Add new video link
* Add new video link
2022-12-14 10:07:17 +08:00
dcf73091e5
add mmyolo regression test yml ( #359 )
2022-12-12 10:27:21 +08:00
a20d3c14fc
fix mmyolo unittest scope bug ( #351 )
2022-12-08 11:45:54 +08:00
1773439b83
[Fix] Fix scale factor order of `LetterResize` and `YOLOv5KeepRatioResize` ( #305 )
...
* Fix bug
* Fix bug
* Fix lint
* Fix scale from hw -> wh
* Fix scale from hw -> wh
2022-11-21 19:44:33 +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
f1279eb3de
[Feature] Add attention module of CBAM ( #246 )
...
* Add Attention Modules
* Adde tutorials on the use of the attention module in How_to
* Update how_to.md
Added tutorials on the use of the attention module
* Update attention_layers.py
* Rename attention_layers.py to cbam_layer.py
* Update __init__.py
* Update how_to.md
* Update how_to.md
* Update how_to.md
* Update cbam_layer.py
* Update cbam_layer.py
* Update cbam_layer.py
* Update how_to.md
* update
* add docstring typehint
* add unit test
* refine unit test
* updata how_to
* add plugins directory
* refine plugin.md
* refine cbam.py and plugins.md
* refine cbam.py and plugins.md
* fix error in test_cbam.py
* refine cbam.py and fix error in test_cbam.py
* refine cbam.py and plugins.md
* refine cbam.py and docs
2022-11-10 11:12:31 +08:00
0b48313dc7
[Fxi] Fix ci bug of BatchATSSAssigner ( #232 )
...
* Fix ci bug
* Fix ci bug
* Fix ci bug
* Fix ci bug
* Fix ci bug
2022-11-03 19:03:06 +08:00
9c250571ca
[Fix] `concatdatasets` for voc train ( #228 )
...
* add yolov5 voc training
* format code
* [Feature] Support VOC Dataset in YOLOv5 (#134 )
* add yolov5 voc training
* fix mosaic bug
* fix mosaic bug and temp config
* fix mosaic bug
* update config
* support training on voc dataset
* format code
* format code
* Optimize Code. Change `RandomTransform` to `OneOf`
* Change `OneOf` to `mmcv.RandomChoice`
* fix yolov5coco dataset
* fix yolov5coco dataset
* fix bug, format code
* format config
* format code
* add yolov5 voc training
* rebase
* fix mosaic bug
* update config
* support training on voc dataset
* format code
* format code
* Optimize Code. Change `RandomTransform` to `OneOf`
* Change `OneOf` to `mmcv.RandomChoice`
* fix yolov5coco dataset
* fix yolov5coco dataset
* fix bug, format code
* format code
* add yolov5 voc training
* fix mosaic bug and temp config
* fix mosaic bug
* update config
* support training on voc dataset
* format code
* format code
* Optimize Code. Change `RandomTransform` to `OneOf`
* Change `OneOf` to `mmcv.RandomChoice`
* fix yolov5coco dataset
* fix yolov5coco dataset
* fix bug, format code
* format code
* add yolov5 voc training
* rebase
* fix mosaic bug
* update config
* support training on voc dataset
* format code
* format code
* Optimize Code. Change `RandomTransform` to `OneOf`
* Change `OneOf` to `mmcv.RandomChoice`
* fix yolov5coco dataset
* fix yolov5coco dataset
* fix bug, format code
* format code
* format code
* fix lint
* add unittest
* add auto loss_weight
* add doc; add model log url
* add doc; add model log url
* add doc; add model log url
* [Feature] support mmyolo deployment (#79 )
* support mmyolo deployment
* mv deploy place
* remove unused configs
* add deploy code
* fix new register
* fix comments
* fix dependent codebase register
* remove unused initialize
* refact deploy config
* credit return to triplemu
* Add yolov5 head rewrite
* refactor deploy
* refactor deploy
* Add yolov5 head rewrite
* fix configs
* refact config
* fix comment
* sync name after mmdeploy 1088
* fix mmyolo
* fix yapf
* fix deploy config
* try to fix flake8 importlib-metadata
* add mmyolo models ut
* add deploy uts
* add deploy uts
* fix trt dynamic error
* fix multi-batch for dynamic batch value
* fix mode
* fix lint
* sync model.py
* add ci for deploy test
* fix ci
* fix ci
* fix ci
* extract script to command for fixing CI
* fix cmake for CI
* sudo ln
* move ort position
* remove unused sdk compile
* cd mmdeploy
* simplify build
* add missing make
* change order
* add -v
* add setuptools
* get locate
* get locate
* upgrade torch
* change torchvision version
* fix config
* fix ci
* fix ci
* fix lint
Co-authored-by: tripleMu <gpu@163.com>
Co-authored-by: RunningLeon <mnsheng@yeah.net>
* [Feature] Support YOLOv5 YOLOv6 YOLOX Deploy in mmdeploy (#199 )
* Support YOLOv5 YOLOv6 YOLOX Deploy in mmdeploy
* Fix lint
* Rename _class to detector_type
* Add some common
* fix lint
Co-authored-by: huanghaian <huanghaian@sensetime.com>
* fix vocdatasets
* fix vocdatasets
Co-authored-by: hanrui1sensetime <83800577+hanrui1sensetime@users.noreply.github.com>
Co-authored-by: tripleMu <gpu@163.com>
Co-authored-by: RunningLeon <mnsheng@yeah.net>
Co-authored-by: huanghaian <huanghaian@sensetime.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
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
5874e4108e
[Feature] support mmyolo deployment ( #79 )
...
* support mmyolo deployment
* mv deploy place
* remove unused configs
* add deploy code
* fix new register
* fix comments
* fix dependent codebase register
* remove unused initialize
* refact deploy config
* credit return to triplemu
* Add yolov5 head rewrite
* refactor deploy
* refactor deploy
* Add yolov5 head rewrite
* fix configs
* refact config
* fix comment
* sync name after mmdeploy 1088
* fix mmyolo
* fix yapf
* fix deploy config
* try to fix flake8 importlib-metadata
* add mmyolo models ut
* add deploy uts
* add deploy uts
* fix trt dynamic error
* fix multi-batch for dynamic batch value
* fix mode
* fix lint
* sync model.py
* add ci for deploy test
* fix ci
* fix ci
* fix ci
* extract script to command for fixing CI
* fix cmake for CI
* sudo ln
* move ort position
* remove unused sdk compile
* cd mmdeploy
* simplify build
* add missing make
* change order
* add -v
* add setuptools
* get locate
* get locate
* upgrade torch
* change torchvision version
* fix config
* fix ci
* fix ci
* fix lint
Co-authored-by: tripleMu <gpu@163.com>
Co-authored-by: RunningLeon <mnsheng@yeah.net>
2022-11-03 19:03:06 +08:00
5ef3606482
[Feature] Support VOC Dataset in YOLOv5 ( #134 )
...
* add yolov5 voc training
* fix mosaic bug
* fix mosaic bug and temp config
* fix mosaic bug
* update config
* support training on voc dataset
* format code
* format code
* Optimize Code. Change `RandomTransform` to `OneOf`
* Change `OneOf` to `mmcv.RandomChoice`
* fix yolov5coco dataset
* fix yolov5coco dataset
* fix bug, format code
* format config
* format code
* add yolov5 voc training
* rebase
* fix mosaic bug
* update config
* support training on voc dataset
* format code
* format code
* Optimize Code. Change `RandomTransform` to `OneOf`
* Change `OneOf` to `mmcv.RandomChoice`
* fix yolov5coco dataset
* fix yolov5coco dataset
* fix bug, format code
* format code
* add yolov5 voc training
* fix mosaic bug and temp config
* fix mosaic bug
* update config
* support training on voc dataset
* format code
* format code
* Optimize Code. Change `RandomTransform` to `OneOf`
* Change `OneOf` to `mmcv.RandomChoice`
* fix yolov5coco dataset
* fix yolov5coco dataset
* fix bug, format code
* format code
* add yolov5 voc training
* rebase
* fix mosaic bug
* update config
* support training on voc dataset
* format code
* format code
* Optimize Code. Change `RandomTransform` to `OneOf`
* Change `OneOf` to `mmcv.RandomChoice`
* fix yolov5coco dataset
* fix yolov5coco dataset
* fix bug, format code
* format code
* format code
* fix lint
* add unittest
* add auto loss_weight
* add doc; add model log url
* add doc; add model log url
* add doc; add model log url
2022-11-03 19:03:06 +08:00
eec1e1f5ed
support LoadAnnotations excluding instances ( #164 )
2022-11-03 19:03:06 +08:00
ba4a496562
Upgrade pre commit hooks ( #129 )
...
* Fix the version conflict in CircleCI
* Update pre-commit config and add zh-cn version.
* Add codespell ignore list.
* Add codespell ignore list.
* Add check copyright config.
2022-10-13 17:39:52 +08:00
e482e3a17c
Add changelog for v0.1.1 ( #111 )
...
* add changelog
* update
* update
* update
* update
* update
2022-09-29 19:09:56 +08:00
0d97ffec06
Fix LetterResize bug with imrescale api ( #105 )
...
* [Feature] Make docker image smaller (#67 )
* [Feature] Make docker image smaller
* Improve doc
* Use Compose directly in BaseMixImageTransform (#71 )
* Use 'Compose' directly
Use 'Compose' directly
* avoids some unnecessary calculations
avoids some unnecessary calculations.
* remove whitespace
remove whitespace
* fix resize bug
* fix resize bug
* fix resize bug
* fix
* fix resize bug
* add test letterresize without batchshape
* fix resize bug
Co-authored-by: HinGwenWoong <peterhuang0323@qq.com>
Co-authored-by: jason_w <wongzheng@126.com>
2022-09-29 19:09:56 +08:00
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
d705f1c57b
[Feature]Support plugin layers for backbone ( #75 )
...
* add plugin layer
* add docstring. add config
* clean the config
* add how_to-doc
* add ut
* update
* update
* update lint
* fix yoloxbug
* update
* update
* update doc
* update
* update
* update
* update
* update
2022-09-29 19:09:56 +08:00
c26a12ed32
Add docstring
2022-09-21 15:09:47 +08:00
66dcc4d3d2
fix import order
2022-09-20 10:57:33 +08:00
cc6ac864c9
update featmap_vis_demo and format code
2022-09-18 17:04:14 +08:00
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
bc4df98973
[Fix] replace pre to per ( #24 )
...
* [Fix] change pre to per
* [Fix] batch_size_pre_gpu to batch_size_per_gpu
2022-09-18 15:40:29 +08:00
3d0be5c868
Add UT and other code
2022-09-18 13:17:03 +08:00
a7a4f16d69
[Fix] Add image for dataset transform ut ( #16 )
2022-09-18 12:23:18 +08:00
56822f4185
[Feature]Add YOLOv5CocoDataset; BatchShapePolicy ( #10 )
...
* [Feature] Add YOLOv5CocoDataset; BatchShapePolicy
* Add unittest of YOLOv5CocoDataset
2022-09-18 11:18:31 +08:00
8945c76f81
[UT] Add unittest for dataset image mixing transforms ( #7 )
2022-09-18 11:04:04 +08:00
yechenzhi
JosonChan
Qingren
Haian Huang(深度眸)
Yue Sun
Nioolek
Yanyi Liu
Youfu
HinGwenWoong
tang576225574
RangiLyu
tripleMu
Range King
RunningLeon
hanrui1sensetime
Haian Huang(深度眸)
wanghonglie
kitecats
JiayuXu
huanghaian
古月闻星