拘谨的远山和叶 2023-02-07 22:15 采纳率: 100%
浏览 207
已结题

Mask rcnn训练自己的数据集出现问题!

导师您好最近在研究Mask rcnn,在使用您的代码很顺利就利用coco数据集训练出了权重。但是在训练自己的数据集时出现了以下问题。
我自己的训练集是从coco数据集抽取的8类(person,car,tuck等),在修改num-class与coco91_indices.json文件后,训练这个数据集时出现了coco各项指标指标十分低的情况。情况如下(此处列举1个epoch的情况):
epoch:1 0.0956 0.1569 0.1016 0.0435 0.0895 0.1337 0.0637 0.1350 0.1482 0.1114 0.1480 0.1727 0.5606 0.008000
其余的epoch结果也是相近,我试着在my_dataset_coco中将coco91_indices.json文件路径替换为修改后的8个类别json文件,但是依然没有解决问题。
在我看来利用coco数据集抽取的几类进行训练,map应该是有一个提升的吧,所以想咨询导师一下这个问题。烦请导师回答,麻烦导师不吝赐教。
  • 写回答

8条回答 默认 最新

  • Funny_AI_LAB 2023-02-08 14:39
    关注

    如果出现这种情况,可能是因为你的数据集和模型结构不匹配。
    1.检查模型是否支持你的类别数量(8个),确保num_classes参数正确。
    2.检查你的训练数据是否正确标注,确保训练数据类别标签与num_classes对应。
    3.检查损失函数是否匹配你的数据集,例如,你是否需要使用多类损失函数。
    如果以上都没有问题,可以尝试以下方法调节训练:
    1.调节训练参数(例如学习率、批量大小),试图找到合适的参数。
    2.增加训练数据,防止模型过拟合。
    3.加载预训练模型,继续训练。
    还有,如果还没有解决,你可以发布你的代码以及详细的错误信息,以便更详细地回答。

    本回答被题主选为最佳回答 , 对您是否有帮助呢?
    评论
查看更多回答(7条)

报告相同问题?

问题事件

  • 系统已结题 2月17日
  • 已采纳回答 2月9日
  • 创建了问题 2月7日

悬赏问题

  • ¥20 用HslCommunication 连接欧姆龙 plc有时会连接失败。报异常为“未知错误”
  • ¥15 网络设备配置与管理这个该怎么弄
  • ¥20 机器学习能否像多层线性模型一样处理嵌套数据
  • ¥20 西门子S7-Graph,S7-300,梯形图
  • ¥50 用易语言http 访问不了网页
  • ¥50 safari浏览器fetch提交数据后数据丢失问题
  • ¥15 matlab不知道怎么改,求解答!!
  • ¥15 永磁直线电机的电流环pi调不出来
  • ¥15 用stata实现聚类的代码
  • ¥15 请问paddlehub能支持移动端开发吗?在Android studio上该如何部署?