mirror of https://github.com/open-mmlab/mmyolo.git
18 lines
544 B
Python
18 lines
544 B
Python
_base_ = './yolov5_s-v61_fast_1xb64-50e_voc.py'
|
|
|
|
deepen_factor = 0.33
|
|
widen_factor = 0.25
|
|
|
|
load_from = 'https://download.openmmlab.com/mmyolo/v0/yolov5/yolov5_n-v61_syncbn_fast_8xb16-300e_coco/yolov5_n-v61_syncbn_fast_8xb16-300e_coco_20220919_090739-b804c1ad.pth' # noqa
|
|
|
|
model = dict(
|
|
backbone=dict(
|
|
deepen_factor=deepen_factor,
|
|
widen_factor=widen_factor,
|
|
),
|
|
neck=dict(
|
|
deepen_factor=deepen_factor,
|
|
widen_factor=widen_factor,
|
|
),
|
|
bbox_head=dict(head_module=dict(widen_factor=widen_factor)))
|