mirror of
https://github.com/open-mmlab/mmyolo.git
synced 2025-06-03 15:00:20 +08:00
573ff033e4
* 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
21 lines
640 B
Python
21 lines
640 B
Python
_base_ = './yolov7_w-p6_syncbn_fast_8x16b-300e_coco.py'
|
|
|
|
model = dict(
|
|
backbone=dict(arch='E2E'),
|
|
neck=dict(
|
|
use_maxpool_in_downsample=True,
|
|
use_in_channels_in_downsample=True,
|
|
block_cfg=dict(
|
|
type='EELANBlock',
|
|
num_elan_block=2,
|
|
middle_ratio=0.4,
|
|
block_ratio=0.2,
|
|
num_blocks=6,
|
|
num_convs_in_block=1),
|
|
in_channels=[320, 640, 960, 1280],
|
|
out_channels=[160, 320, 480, 640]),
|
|
bbox_head=dict(
|
|
head_module=dict(
|
|
in_channels=[160, 320, 480, 640],
|
|
main_out_channels=[320, 640, 960, 1280])))
|