Ubuntu20搭建pytorch深度学习框架——使用增强样本运行Dlinknet提取道路水体(六)——训练iou和测试iou

这里需要声明一下
之前博文里的内容都是从我的训练样本中挑选出的部分数据来作为测试样本
那么这里计算出的iou其实是训练iou

即使训练iou达到了99.9%
很可能你的测试iou也还是70%多

但大多数适用场景需要的是测试iou
因此我在这里将我的数据划分成75%、25%
两个数据集是没有重叠的

75%的作为训练集重新训练
15%的作为干净的没有训练的测试集

将75%的训练集进行增强处理
差不多六万多张样本
水体的测试iou却只有78%
因此需要针对Dinknet网络进行一些参数修改和样本调试

  • 不加增强样本,训练样本数量大概在1万7千多张
  • 调整跳出训练的次数,完整训练400次(Dinknet网络设置的有break,小伙伴可以去看看相关语句)
  • 每5次epoch输出一个模型,将输出的模型保存下来
  • 添加计算iou的代码,计算训练iou和测试iou,并将两条曲线绘制在一张图中
  • 计算每张样本的类别iou,并将iou低的样本挑选出来,分析iou低的原因
  • 改变图片大小,修改尺寸为1024

以上为目前需要对源代码进行修改调试的任务
这里贴一下一些好的博文
loss曲线解读
参数调整
来帮助我们更好的理解深度学习相关参数

这次我是在windows系统上运行的
但是出现了问题
在pycharm里跑不出结果了
报错
. : 无法加载文件 D:\Documents\WindowsPowerShell\profile.ps1,因为在此系统上禁止运行脚本。有关详细信息,请参阅 https:/go.microsoft.c

这里尝试了很多方法
这个博客的方法适用于我这个情况

这个语句是有用的

Set-ExecutionPolicy -ExecutionPolicy RemoteSigned

现在运行是没有前面的红字部分
而且有了base但是还是运行了出不来结果
Ubuntu20搭建pytorch深度学习框架——使用增强样本运行Dlinknet提取道路水体(六)——训练iou和测试iou_第1张图片天啦,这是什么情况啊


这个解决方法就是把之前的环境删除

重新新建一个pytorch的环境
然后再安装需要的包
就可以跑啦

截止到2022.10.12日上述的任务已经全部完成
代码比较零碎
但这里有个计算每张相片的iou
不知道为什么之前csdn复制粘贴代码可以自动换行
保留格式 最近又不行了
调格式太麻烦
这里就不分享我的代码了
太长了

你可能感兴趣的:(深度学习,windows,深度学习,pytorch,python)