mirror of https://github.com/open-mmlab/mmyolo.git
24 lines
793 B
Python
24 lines
793 B
Python
_base_ = './ppyoloe_s_fast_8xb32-300e_coco.py'
|
|
|
|
# The pretrained model is geted and converted from official PPYOLOE.
|
|
# https://github.com/PaddlePaddle/PaddleDetection/blob/release/2.5/configs/ppyoloe/README.md
|
|
checkpoint = 'https://download.openmmlab.com/mmyolo/v0/ppyoloe/ppyoloe_pretrain/cspresnet_x_imagenet1k_pretrained-81c33ccb.pth' # noqa
|
|
|
|
deepen_factor = 1.33
|
|
widen_factor = 1.25
|
|
|
|
train_batch_size_per_gpu = 16
|
|
|
|
model = dict(
|
|
backbone=dict(
|
|
deepen_factor=deepen_factor,
|
|
widen_factor=widen_factor,
|
|
init_cfg=dict(checkpoint=checkpoint)),
|
|
neck=dict(
|
|
deepen_factor=deepen_factor,
|
|
widen_factor=widen_factor,
|
|
),
|
|
bbox_head=dict(head_module=dict(widen_factor=widen_factor)))
|
|
|
|
train_dataloader = dict(batch_size=train_batch_size_per_gpu)
|