Ubuntu20搭建pytorch深度学习框架——运行Dlinknet提取道路(二)——代码运行

已于 2022-12-08 12:44:48 修改
阅读量387 收藏 2
点赞数 1
分类专栏： Ubuntu平台Dlinknet深度学习道路水体全过程 Linux 深度学习 文章标签： windows 深度学习 Dinknet
于 2022-06-09 15:33:07 首次发布
版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。
本文链接：https://blog.csdn.net/Laney_Midory/article/details/125204441
版权

成功运行后把我的12468个原图+标签
分辨率512*512大小的图片
在这里插入图片描述
放进去开始训练
在这里调整了
ATCHSIZE_PER_CARD = 16
total_epoch = 500

现在是2022.6.9 15：27：40
看看能够多久运行成功
现在是2022.6.9 20：37：50 现在是运行了epoch：78
在这里插入图片描述
慢慢等待一下结果

结果
在这里插入图片描述
运行到了epoch=98结束
在这里插入图片描述

再把训练的原图作为需要预测的
看看训练结果如何
直接跑会报错
在这里插入图片描述
这是由于我的预测图片的大小是6060
也需要经过裁剪成512*512
这样就可以成功运行啦

预测完成后觉得效果不错
进行原样本的预测后的miou计算
在这里插入图片描述
突然发现miou的计算也很慢。。。

这是预测结果
在这里插入图片描述
道路还是差点

同时进行一下河流样本的训练！
使用该网络进行河流的训练
由于之前河流样本体量大
但是我在裁剪的时候没注意到它不是2的倍数
所以现在需要重新做
在这里插入图片描述
现在正确了
应该是之前我在删除全黑样本的时候出现了一些问题

Dinknet只能使用后缀为png的标签，后缀为tif的原图进行训练

一共23006张图片
训练开始时间为 2022.6.10
18：16：45

2022.6.12
11：38：30已经训练完毕
在这里插入图片描述
然后对训练的1500个样本进行预测
预测结果如下
请添加图片描述

Laney_Midory
关注
  • 1
    点赞
  • 踩
  • 2
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 2
    评论
专栏目录
搭建Pytorch深度学习框架所需资源包，无需再下载任何依赖包
01-16
整理了一套完整的深度学习所需依赖包，包含：pytorch+cuda11.0+cudnn11.0+anaconda3.7+pycharm2020+opencv4.5.1，亲测了Yolo-V5模型，无需额外依赖包。
PyTorch深度学习代码及数据PyTorch深度学习代码及数据
最新发布
03-01
PyTorch深度学习代码及数据PyTorch深度学习代码及数据PyTorch深度学习代码及数据PyTorch深度学习代码及数据PyTorch深度学习代码及数据PyTorch深度学习代码及数据PyTorch深度学习代码及数据PyTorch深度学习代码及数据...
pytorch中使用numpy生成随机数每个epoch都一样
icylling的专栏
04-20 1609
pytorch中使用numpy生成随机数每个epoch都一样问题原因解决途径 问题 最近在使用pytorch做实验的过程中发现，在dataset里面使用numpy生成随机数，每个epoch生成的随机数是一样的。 代码 import numpy as np import torch from torch.utils.data import Dataset from torch.utils.data ...
windows10使用cuda11搭建pytorch深度学习框架——运行Dlinknet提取道路(三)——模型精度评估代码完善
Laney_Midory的博客
01-12 1678
重新调试好代码，使用Dinknet34模型对数据集进行训练 数据集大小为1480张图片 运行时间为2022年1月12日16：00 记录下该模型训练时间 但如何评估模型的精度也是一个问题，因此作如下总结 模型精度评估 评价指标： 　　准确率 （Accuracy），混淆矩阵 （Confusion Matrix），精确率（Precision），召回率（Recall），平均正确率（AP），mean Average Precision(mAP)，交除并（IoU），ROC + AUC，非极大值抑制（NMS）。 IoU
Pytorch | dataloader 多线程下numpy每个线程随机种子都一样解决方案。
jmu201521121021的博客
12-04 1903
问题描述 pytorch的Dataloader用于加载数据。在num_works >1时， 每个线程中numpy.random产生的随机数 一样，也就是随机种子相同。random 和 torch两个模块的随机数不会出现这种情况。 DataLoader(dataset, batch_size=1, shuffle=False, sampler=None, batch_...
PyTorch深度学习框架，实战解析，43页PPT资源
07-10
PyTorch 是一个开源的机器学习框架，广泛应用于深度学习领域。它由Facebook的人工智能研究团队开发，并于2017年发布。PyTorch 提供了丰富的工具和函数，使得开发者可以方便地构建和训练神经网络模型。 以下是 ...
基于Pytorch深度学习框架进行整体环境搭建.zip
01-14
基于Pytorch深度学习框架进行整体环境搭建，包括数据集制作，模型训练，模型测试，模型优化；基于kinova机器人搭建实际抓取环境；采用级联网络Cascade R-CNN提取特征。
深度学习 - 搭建pytorch深度学习环境.pdf
03-01
Torch是一个用于深度学习的=数学计算库，而Pytorch则是一个基于...而搭建Pytorch虚拟环境，通俗地讲，就是搭建一个包含了Pytorch的相关包的Python解释器的环境，即是专门用于处理基于Python的深度学习的问题的环境。
Pytorch/paddle单机多卡RTX 3060×2的Ubuntu深度学习训练环境配置+代码模板+常见问题解决
m0_46339652的博客
02-24 2934
下载nccl: https://developer.nvidia.com/nccl/nccl-legacy-downloads 选择对应的操作系统和CUDA版本(这里选择离线安装): 下载好后进行离线安装: 接下来提示需要安装一个,必须先安装公钥才能进行接下来的操作. 安装公钥: 更新软件源: 清华源最好用: 然后更新: 安装nccl: 将nccl添加到环境变量中: nccl默认的安装目录是 ,修改~/.bashrc文件，添加如下内容到文件中: 添加好之后保存文件，使用让文件的配置生效，在通过查看环境变量
PyTorch的学习（4）--加载数据集
weixin_46042803的博客
11-01 289
dataloader
Ubuntu18.04系统搭建深度学习环境
Ayu147258的博客
09-15 3557
Ubuntu18.04系统安装深度学习环境，安装概要：显卡驱动+cuda11.7+cudnn+pip换源+Anaconda3+pycharm
Pytorch中求模型准确率的几种方法
热门推荐
zihao_c的博客
10-31 2万+
方法一：直接在epoch过程中求取准确率 简介：此段代码是LeNet5中截取的。 def train_model(model,train_loader): optimizer = torch.optim.Adam(model.parameters()) loss_func = nn.CrossEntropyLoss() EPOCHS = 5 for epoch in range(EPOCHS): correct = 0 for batch
【图像分割】卫星遥感影像道路分割：D-LinkNet算法解读
兴趣使然的创作者
05-20 3864
毕设的一小部分
每个epoch使用的是同样的数据，重复同样的过程，为什么模型的性能还会发生变化？
weixin_51025471的博客
03-28 1043
参考一下，学习做笔记
Ubuntu20搭建pytorch深度学习框架——运行Dlinknet提取道路后对结果图进行后处理(三)——代码+软件实现
Laney_Midory的博客
07-18 1040
得到黑白结果图后由于某些原因需要将结果图转为矢量图进行后处理这是由于大多数软件比如arcgis对矢量shp文件处理较为方便而对栅格图片的处理有很多不便因此我们首先需要将我们的黑白栅格图片转为矢量shp文件这里主要使用arcgis软件实现。.......................................
windows10使用cuda11搭建pytorch深度学习框架——运行Dlinknet提取道路(二)——代码运行问题解决
Laney_Midory的博客
01-11 1873
运行程序 去github上下载Dlinknet的代码 https://github.com/zlckanata/DeepGlobe-Road-Extraction-Challenge 直接使用anaconda的powershell运行程序 python train.py 出现错误 RuntimeError: An attempt has been made to start a new process before the… 参考这篇博文 我采用了 去掉num_workers参数解决 下一个问题接踵而至
Ubuntu配置深度学习环境(Anaconda3+cuda+cudnn+Jupyter_Notebook+code-server)
cheng_bian的博客
10-20 455
为准备用服务器进行深度学习者配置环境准备
如何进行pytorch深度学习框架搭建
04-20
很高兴回答你的问题。进行pytorch深度学习框架搭建，需要首先安装pytorch库并与Python环境配合。接着，你可以使用pytorch自带的模块和函数进行模型的定义、训练和预测。具体实现方法可以参考pytorch官方文档和相关的学习资料。希望能够帮助到你。

“相关推荐”对你有帮助么？

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 2
添加红包

请填写红包祝福语或标题

个

红包个数最小为10个

元

红包金额最低5元

当前余额3.43元 前往充值 >
需支付：10.00元
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

Laney_Midory

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付：¥1
获取中
扫码支付

您的余额不足，请更换扫码支付或充值

打赏作者

实付元
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明：

1.余额是钱包充值的虚拟货币，按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载，可以购买VIP、付费专栏及课程。

余额充值