mmyolo/configs/yolov6/yolov6_l_syncbn_fast_8xb32-300e_coco.py
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

24 lines
753 B
Python

_base_ = './yolov6_m_syncbn_fast_8xb32-300e_coco.py'
deepen_factor = 1
widen_factor = 1
model = dict(
backbone=dict(
deepen_factor=deepen_factor,
widen_factor=widen_factor,
hidden_ratio=1. / 2,
block_cfg=dict(
type='ConvWrapper',
norm_cfg=dict(type='BN', momentum=0.03, eps=0.001)),
act_cfg=dict(type='SiLU', inplace=True)),
neck=dict(
deepen_factor=deepen_factor,
widen_factor=widen_factor,
hidden_ratio=1. / 2,
block_cfg=dict(
type='ConvWrapper',
norm_cfg=dict(type='BN', momentum=0.03, eps=0.001)),
block_act_cfg=dict(type='SiLU', inplace=True)),
bbox_head=dict(head_module=dict(widen_factor=widen_factor)))