【MindSpore】用coco2017训练Model_zoo上的 yolov4,迭代了两千多batch_size之后报错,大佬们帮忙看看。

问题描述:

运行环境:Atlas 800 9000      CANN20.2       

batch_size: 4,8,16,32(四种分别尝试过,都在一个epoch训练了80%左右报这个错了)

数据集:coco2017 

模型:Model_zoo   YOLOv4_Cspdarknet53  (配置都是用的默认的)

解决方案:

从错误截图中初步分析的结论是,数据集可能不完整或是有错误,导致一定step之后再也无法迭代出数据进行训练了。

我们建议按照如下方式排查故障:

1、确保coco2017数据集来源正确,mindspore model_zoo中yolo v4 README提供了coco数据集的官方下载链接:COCO - Common Objects in Context

2、上述错误一般源于src/yolo_dataset.py读取coco数据集时发生了错误,建议采用如下方式进行排查:

在src/yolo_dataset.py的代码底部增加如下调试代码:

from src.config import ConfigYOLOV3DarkNet53

if __name__ == '__main__':

    config = ConfigYOLOV3DarkNet53()

    config.label_smooth = 0

    config.label_smooth_factor = 0.1

    image_dir = "/your_path/coco2017/train2017"

    anno_path = "/your_path/coco2017/annotations/instances_train2017.json"

    yolo_dataset = COCOYoloDataset(image_dir, anno_path)

    nums = 0

    for data in yolo_dataset:

        nums += 1

        print("nums is ", nums)

以上代码主要用于调试COCOYoloDataset,正常情况下,yolo_dataset是可以正确完整输出所有数据的,请参照以上调试代码,确保读取coco数据集完整无误。

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值