技术专栏
Yolo-FastestV2训练笔记
官方代码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
相关专栏
-
浏览量:13747次2020-11-16 02:10:04
-
浏览量:651次2023-12-20 16:31:10
-
浏览量:1523次2023-04-14 14:37:29
-
浏览量:1322次2023-12-19 17:25:07
-
浏览量:2006次2023-05-25 16:32:18
-
浏览量:618次2023-12-18 18:07:01
-
浏览量:1881次2024-02-02 18:15:06
-
浏览量:605次2023-07-05 10:17:52
-
浏览量:1130次2023-12-18 17:48:09
-
浏览量:1177次2023-12-05 17:30:53
-
浏览量:1213次2023-12-14 17:15:07
-
浏览量:2510次2024-02-28 15:36:09
-
浏览量:1834次2024-02-20 13:54:36
-
浏览量:1653次2024-02-04 17:13:47
-
浏览量:1165次2024-02-28 16:15:25
-
浏览量:1085次2024-07-12 09:12:43
-
浏览量:1292次2023-05-13 21:35:31
-
2024-02-02 14:41:10
-
浏览量:6181次2021-06-07 11:48:50
切换马甲
上一页
下一页
打赏用户
共 1 位
我要创作
分享技术经验,可获取创作收益
分类专栏
-
3篇
热门专栏
- YOLOV5 代码复现以及搭载服务器运行
- ubuntu系统tensorflow框架搭建与测试 —— 深度学习(四)
- RKNN-Toolkit转换Tensorflow模型至Rockchip NPU推理并进行性能评估
- 迁移学习的 PyTorch 实现
- 深度学习之卷积神经网络CNN及tensorflow代码实现示例
- Tensorflow vs PyTorch —— 卷积神经网络
- 高性能 PyTorch 训练 (1):性能瓶颈的调查与分析
- 语义分割系列24-PointRend(pytorch实现)
- ubuntu18.04配置pytorch框架并进行fcn网络并训练 —— 深度学习(一)
- 让PyTorch训练速度更快,你需要掌握这17种方法
置顶时间设置
结束时间
删除原因
-
广告/SPAM
-
恶意灌水
-
违规内容
-
文不对题
-
重复发帖
打赏作者
刘兵
您的支持将鼓励我继续创作!
打赏金额:
¥1
¥5
¥10
¥50
¥100
支付方式:
微信支付
打赏成功!
感谢您的打赏,如若您也想被打赏,可前往 发表专栏 哦~
举报反馈
举报类型
- 内容涉黄/赌/毒
- 内容侵权/抄袭
- 政治相关
- 涉嫌广告
- 侮辱谩骂
- 其他
详细说明
审核成功
发布时间设置
发布时间:
请选择发布时间设置
是否关联周任务-专栏模块
审核失败
失败原因
请选择失败原因
备注
请输入备注