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
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
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
f54e5603fd
[Enchance] Extending `DetDataPreprocessor` flexibility ( #511 )
...
* Extending DetDataPreprocessor flexibility
* update
* add cast_data
2023-02-06 17:47:22 +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
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
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
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
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
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
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
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
3f0b2f7c97
Add unittest for dataset transforms ( #3 )
2022-09-18 10:49:41 +08:00
JosonChan
Haian Huang(深度眸)
Nioolek
tang576225574
HinGwenWoong
Haian Huang(深度眸)
JiayuXu
huanghaian
古月闻星