利用deeplab v3+开源代码训练PASCAL VOC 2012数据集

本文档详细介绍了如何配置deeplab v3+环境,准备PASCAL VOC 2012数据集,将其转换为TFRecord格式,并进行模型训练、评估及结果可视化。通过运行train.py和eval.py,在训练过程中遇到内存限制问题,通过调整参数解决。最终,使用vis.py展示模型分割结果。
摘要由CSDN通过智能技术生成

deeplab v3+环境配置好之后则可以进行模型的训练
1、准备PASCAL VOC 2012数据集:可以在ubantu命令行上运行

#From the tensorflow/models/research/deeplab/datasets directory.
  sh download_and_convert_voc2012.sh

下载并把数据集转换成 TFRecord格式

我已有VOC 2012数据集压缩包,所以省了下载步骤。因此手动在\research\deeplab\datasets目录下创建一个名为pascal_voc_seg的文件夹,把压缩包放进该文件夹里,再运行 sh download_and_convert_voc2012.sh(代码无需改动),运行后会多了下图红色方框里的文件(在虚拟机ubantu 16.04运行的程序,一下截图为虚拟机共享文件夹里的截图):
在这里插入图片描述

在这里插入图片描述

运行效果图
至此,可以运行 train.py开始训练模型

2、train.py的运行
github示例链接:

评论 6
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值