Yolo-FastestV2训练笔记

Yolo-FastestV2训练笔记 刘兵 2022-11-30 15:57:05 1960

官方代码download下来,数据集准备好,这些步骤做过AI数据集训练的人都没问题,在这里不用多说。
在训练时,跳出来loss.py的错误,按照网上说的改部分版本的软件,无法解决。我通过摸索,已经成功训练处理了数据集。原来代码的问题如下:

Append

        a = t[:, 6].long()  # anchor indices
        indices.append((b, a, gj.clamp_(0, gain[3] - 1), gi.clamp_(0, gain[2] - 1)))  # image, anchor, grid indices
        tbox.append(torch.cat((gxy - gij, gwh), 1))  # box
        anch.append(anchors_cfg[a])  # anchors
        tcls.append(c)  # class

出错之后就无法运行,如果坛子里面朋友遇到这种问题,可以按我的办法尝试。
原因是clamp_函数需要接受整形数,而gain是浮点型的,直接改成如下即可:

Append

        a = t[:, 6].long()  # anchor indices
        indices.append((b, a, gj.clamp_(0, (gain[3] - 1).long()), gi.clamp_(0, (gain[2] - 1).long())))  # image, anchor, grid indices
        tbox.append(torch.cat((gxy - gij, gwh), 1))  # box
        anch.append(anchors_cfg[a])  # anchors
        tcls.append(c)  # class

能够直接通过训练。可能环境不同、显卡不同、各个模块的版本不同,每一个人遇到的问题都不一样,各位如果遇到这个问题解决不了,可以按我的办法尝试一下。

声明:本文内容由易百纳平台入驻作者撰写,文章观点仅代表作者本人,不代表易百纳立场。如有内容侵权或者其他问题,请联系本站进行删除。
刘兵
红包 1 收藏 评论 打赏
评论
1个
内容存在敏感词
手气红包
  • 刘兵 2022-12-06 08:43:08
    回复
    很可惜,精确率、召回率、AP值都不高,哪位知道怎么样能训练的高一点? Precision:0.435165 Recall:0.461818 AP:0.345820 F1:0.447463
相关专栏
置顶时间设置
结束时间
删除原因
  • 广告/SPAM
  • 恶意灌水
  • 违规内容
  • 文不对题
  • 重复发帖
打赏作者
易百纳技术社区
刘兵
您的支持将鼓励我继续创作!
打赏金额:
¥1易百纳技术社区
¥5易百纳技术社区
¥10易百纳技术社区
¥50易百纳技术社区
¥100易百纳技术社区
支付方式:
微信支付
支付宝支付
易百纳技术社区微信支付
易百纳技术社区
打赏成功!

感谢您的打赏,如若您也想被打赏,可前往 发表专栏 哦~

举报反馈

举报类型

  • 内容涉黄/赌/毒
  • 内容侵权/抄袭
  • 政治相关
  • 涉嫌广告
  • 侮辱谩骂
  • 其他

详细说明

审核成功

发布时间设置
发布时间:
是否关联周任务-专栏模块

审核失败

失败原因
备注
拼手气红包 红包规则
祝福语
恭喜发财,大吉大利!
红包金额
红包最小金额不能低于5元
红包数量
红包数量范围10~50个
余额支付
当前余额:
可前往问答、专栏板块获取收益 去获取
取 消 确 定

小包子的红包

恭喜发财,大吉大利

已领取20/40,共1.6元 红包规则

    易百纳技术社区