其他分享
首页 > 其他分享> > Pytorch移植Deeplabv3+训练CityScapes数据集详细步骤

Pytorch移植Deeplabv3+训练CityScapes数据集详细步骤

作者:互联网

源代码链接: https://github.com/fregu856/deeplabv3#paperspace

这个源代码相对简单,可以用来仔细看下,学习思想。

环境配置

源代码使用的是pytorch0.4,我使用的是pytorch1.13,会报某个函数版本警告问题,但是不影响训练过程。对于pytorch安装,可以直接到官网采用相应的命令进行一键式安装。

训练过程

1、准备好数据集Cityscapes

相关数据集介绍可以查看这篇博客。https://niecongchong.github.io/2019/08/10/CityScapes%E6%95%B0%E6%8D%AE%E9%9B%86%E7%AE%80%E4%BB%8B%E4%B8%8E%E6%95%B0%E6%8D%AE%E5%A4%84%E7%90%86%E5%92%8C%E7%B2%BE%E5%BA%A6%E6%8C%87%E6%A0%87/

CityScapes数据集官网下载并不容易,我准备好了相应的百度网盘链接供学习使用:

链接:https://pan.baidu.com/s/1VHcDOXCanN1fKXwFdPdHMw 提取码:6pg6

2、将源代码下载好后进行解压

  1. 解压后,首先编译deeplabv3 / utils / preprocess_data.py文件,编译之前需要修改一点代码:

修改preprocess_data.py中第97、98行的路径,cityscapes_data_path是你下载的cityscapes数据的路径。cityscapes_meta_path是你想要存放labelimg的图像位置,这里我直接将cityscapes_meta_path设置成和cityscapes_data_path相同路径。

之后运行 preprocess_data.py。等待一段时间后可以发现项目路径下多了一个label_imgs文件夹,里面存放了许多的mask图片。

2.  运行train.py ,运行之前也要做出一点修改。

在第40行左右调整以下路径

在59行左右调整以下路径:

之后便可以训练了,如果有修改问题的,可以联系我,我发我修改好的给你,对比一下即可。

训练能输出以下信息就代表成功了:

 

标签:路径,CityScapes,py,E6%,修改,Pytorch,Deeplabv3,data,cityscapes
来源: https://blog.csdn.net/qq_41964545/article/details/115252939