MMDetection实验记录踩坑记录
  gpLnA9Ilm5Ka 2023年11月02日 124 0


AP值始终为0

在实验MMDetection的DAB-DETR模型进行实验时,AP值始终上不去。
可以看到,在第22个epoch时的AP值仅为0.002

MMDetection实验记录踩坑记录_参数设置

因为在此之前已经运行过YOLOX,Faster-RCNN等模型,所以数据集的设置肯定是没有问题的,而博主也只是修改了DAB-DETR的类别参数而已,一想到这,才猛然想起,DETR类模型的类别数还需要加上一类,即背景类,先前在修改参数时,看到原始参数为80,还以为不用加1了呢。
讲num_class参数数值加一,OK。再次运行:

就是可惜博主白白运行了一个晚上。

断点训练问题

MMDetection针对模型在训练过程中意外中断,避免重头训练,设计了断点训练参数 resume,只需要将该参数设置为最后一次训练的权重即可。

parser.add_argument(
        '--resume',
        nargs='?',
        default="/home/ubuntu/programs/mmdetection/output/deformable-detr/epoch_2.pth",
        type=str,
        const='auto',
        help='If specify checkpoint path, resume from it, while if not '
        'specify, try to auto resume from the latest checkpoint '
        'in the work directory.')


【版权声明】本文内容来自摩杜云社区用户原创、第三方投稿、转载,内容版权归原作者所有。本网站的目的在于传递更多信息,不拥有版权,亦不承担相应法律责任。如果您发现本社区中有涉嫌抄袭的内容,欢迎发送邮件进行举报,并提供相关证据,一经查实,本社区将立刻删除涉嫌侵权内容,举报邮箱: cloudbbs@moduyun.com

  1. 分享:
最后一次编辑于 2023年11月08日 0

暂无评论

推荐阅读
gpLnA9Ilm5Ka