记低级错误-mmdet-centernet

原创

已于 2022-08-26 16:10:14 修改 · 577 阅读

1 ·

CC 4.0 BY-SA版权

文章标签：

#算法 #python #人工智能

于 2022-08-26 14:48:53 首次发布

mmdet训练、测试centernet，经常使用框架，却对里面的参数细节研究不深入，发生严重低级错误。

训练和测试pipline代码

train_pipeline = [
    dict(type='LoadImageFromFile', to_float32=True, color_type='color'),
    dict(type='LoadAnnotations', with_bbox=True),
    dict(
        type='RandomCenterCropPad',
        crop_size=(512, 512),
        ratios=(0.6, 0.7, 0.8, 0.9, 1.0, 1.1, 1.2, 1.3),
        mean=[0, 0, 0],
        std=[1, 1, 1],
        to_rgb=True,
        test_pad_mode=None),
    dict(type='Resize', img_scale=(512, 512), keep_ratio=True),
    dict(type='RandomFlip', flip_ratio=0.5),
    dict(type='Normalize', **img_norm_cfg),
    #dict(type='Pad', size_divisor=32),
    dict(type='DefaultFormatBundle'),
    dict(type='Collect', keys=['img', 'gt_bboxes', 'gt_labels'])
]
test_pipeline = [
    dict(type='LoadImageFromFile', to_float32=True),
    dict(
        type='MultiScaleFlipAug',
        image_scale=(512,512),
        flip=False,
        transforms=[
            dict(type='Re