kitecats
|
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 |
HinGwenWoong
|
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 |
Nioolek
|
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 |
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 |
|
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 |
hanrui1sensetime
|
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 |
|
Nioolek
|
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 |
Haian Huang(深度眸)
|
eec1e1f5ed
|
support LoadAnnotations excluding instances (#164)
|
2022-11-03 19:03:06 +08:00 |
JiayuXu
|
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 |
|
Haian Huang(深度眸)
|
e482e3a17c
|
Add changelog for v0.1.1 (#111)
* add changelog
* update
* update
* update
* update
* update
|
2022-09-29 19:09:56 +08:00 |
|
Nioolek
|
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 |
|
Haian Huang(深度眸)
|
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 |
|
wanghonglie
|
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 |
huanghaian
|
c26a12ed32
|
Add docstring
|
2022-09-21 15:09:47 +08:00 |
|
huanghaian
|
66dcc4d3d2
|
fix import order
|
2022-09-20 10:57:33 +08:00 |
hha
|
cc6ac864c9
|
update featmap_vis_demo and format code
|
2022-09-18 17:04:14 +08:00 |
|
HinGwenWoong
|
1915c04e5d
|
[Fix] Improve type hint (#22)
* Fix type hint
* Delete `-> None` when no return val
* format code
|
2022-09-18 17:02:00 +08:00 |
古月闻星
|
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 |
|
hha
|
3d0be5c868
|
Add UT and other code
|
2022-09-18 13:17:03 +08:00 |
|
HinGwenWoong
|
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 |
|
HinGwenWoong
|
8945c76f81
|
[UT] Add unittest for dataset image mixing transforms (#7)
|
2022-09-18 11:04:04 +08:00 |
|
HinGwenWoong
|
3f0b2f7c97
|
Add unittest for dataset transforms (#3)
|
2022-09-18 10:49:41 +08:00 |
|
wanghonglie
|
85e504fe67
|
[Feature] Add Backbone;Neck (#2)
* add backbone neck
* update
|
2022-09-18 10:47:44 +08:00 |
|
hha
|
cfec074cf6
|
init commit
|
2022-09-18 10:11:55 +08:00 |