图像识别
楼下二狗
这个作者很懒,什么都没留下…
展开
-
使用yolov5-6.0训练自定义数据集时对代码的修改
【代码】使用yolov5-6.0训练自定义数据集时对代码的修改。原创 2024-04-10 16:22:48 · 69 阅读 · 1 评论 -
yolov5训练自定义数据集模型优化
图表中没有显示任何线条,似乎表明在训练或验证集上没有发生任何分类错误。换句话说,这意味着模型在训练和验证过程中可能完全正确地分类了所有的图像。其中,图3参数训练出的效果最好,上图中的。在训练集中是否应该放入一些负样本。原创 2024-04-09 09:09:23 · 165 阅读 · 0 评论 -
使用yolov5s.pt直接训练时的报错
重新设置虚拟环境也无法解决。原创 2024-03-26 15:32:19 · 119 阅读 · 0 评论 -
使用yolov5-5.0训练自定义数据集时对代码的修改
解决:在utils文件夹下找到loss.py,找到gain = torch.ones(7, device=targets.device),改为:gain = torch.ones(7, device=targets.device).long()gj = gj.clamp_(0, gain[3] - 1).long() # 确保gj是长整型。gi = gi.clamp_(0, gain[2] - 1).long() # 确保gi是长整型。但此改动依然报错,还是应该执行上面的修改方法)原创 2024-03-29 09:29:02 · 300 阅读 · 1 评论 -
yolov5训练自定义数据集使用Albumentations库做数据增强,得到增强后图片与txt标注文件
yolov5训练自定义数据集使用Albumentations库做数据增强原创 2024-04-03 15:35:34 · 663 阅读 · 1 评论