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示例链接: