mirror of https://github.com/open-mmlab/mmyolo.git
22 lines
672 B
Python
22 lines
672 B
Python
_base_ = './yolov7_w-p6_syncbn_fast_8x16b-300e_coco.py'
|
|
|
|
model = dict(
|
|
backbone=dict(arch='D'),
|
|
neck=dict(
|
|
use_maxpool_in_downsample=True,
|
|
use_in_channels_in_downsample=True,
|
|
block_cfg=dict(
|
|
type='ELANBlock',
|
|
middle_ratio=0.4,
|
|
block_ratio=0.2,
|
|
num_blocks=6,
|
|
num_convs_in_block=1),
|
|
in_channels=[384, 768, 1152, 1536],
|
|
out_channels=[192, 384, 576, 768]),
|
|
bbox_head=dict(
|
|
head_module=dict(
|
|
in_channels=[192, 384, 576, 768],
|
|
main_out_channels=[384, 768, 1152, 1536],
|
|
aux_out_channels=[384, 768, 1152, 1536],
|
|
)))
|