首页专栏详情
打赏
Yolo-FastestV2训练笔记
Yolo-FastestV2训练笔记
易百纳技术社区 刘兵 2022-11-30 15:57:05

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

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

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

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

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



声明:本文内容由易百纳平台入驻作者撰写,文章观点仅代表作者本人,不代表易百纳立场。如有内容侵权或者其他问题,请联系本站进行删除。

打赏
共1人已赏
评论
1个
内容存在敏感词
  • 易百纳技术社区
    很可惜,精确率、召回率、AP值都不高,哪位知道怎么样能训练的高一点? Precision:0.435165 Recall:0.461818 AP:0.345820 F1:0.447463
相关专栏
打赏作者
易百纳技术社区
刘兵
您的支持将鼓励我继续创作!
打赏金额:
¥1 易百纳技术社区
¥5 易百纳技术社区
¥10 易百纳技术社区
¥50 易百纳技术社区
¥100 易百纳技术社区
支付方式:
微信支付
支付宝支付
易百纳技术社区 微信支付
易百纳技术社区
打赏成功!

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

审核成功

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

审核失败

失败原因
备注
Loading...
易百纳技术社区
确定要删除此文章、专栏、评论吗?
确定
取消
易百纳技术社区
易百纳技术社区
在专栏模块发布专栏,可获得其他E友的打赏
易百纳技术社区
回答悬赏问答,被题主采纳后即可获得悬赏金
易百纳技术社区
在上传资料时,有价值的资料可设置为付费资源
易百纳技术社区
达到一定金额,收益即可提现~
收益也可用来充值ebc,下载资料、兑换礼品更容易
易百纳技术社区
活动规则
  • 1.周任务为周期性任务,每周周一00:00刷新,上周完成的任务不会累计到本周,本周需要从头开始任务,当前任务完成后才可以完成下一个任务
  • 2.发布的专栏与资料需要与平台的板块有相关性,禁止注水,专栏/资料任务以审核通过的篇数为准
  • 3.任务完成后,现金奖励直接打款到微信账户;EBC/收益将自动发放到个人账户,可前往“我的钱包”查看;其他奖励请联系客服兑换
  • 4.每周最后三个任务将会有以下奖品掉落:社区热卖开发板、小米音响、视频年度会员、京东卡、华为手机等等
易百纳技术社区
升级提醒
易百纳技术社区

恭喜您由入门

社区送出礼品一份

请填写您的收件地址,礼品将在3个工作日寄出

易百纳技术社区