图像语义分割模型DeepLab训练Cityscapes数据集过程记录

参考：https://github.com/tensorflow/models/blob/master/research/deeplab/g3doc/cityscapes.md

https://github.com/tensorflow/models/blob/master/research/deeplab/local_test.sh

1、下载数据

cd models/research/deeplab/datasets
mkdir cityscapes

下载原始5000张图片leftImg8bit_trainvaltest.zip(11GB)并解压在leftImg8bit文件夹下；
下载惊标5000张图片gtFine_trainvaltest.zip(241MB)并解压在gtFine文件下下；
特别说明：发现下载一段时间就会出现服务器中断，国内有CSDN博客分享了百度云下载链接。

https://blog.csdn.net/zym19941119/article/details/81198315
链接: https://pan.baidu.com/s/108_NgFheDIpnQRrwz5uhmw 提取码: dhr8

2、转换数据

cd models/research/deeplab/datasets
sh convert_cityscapes.sh

执行后会在cityscapes目录下出现tfrecord文件夹。

3、下载checkpoint

https://github.com/tensorflow/models/blob/master/research/deeplab/g3doc/model_zoo.md
选择xception65_cityscapes_trainfine，下载并解压到cityscapes文件夹下

4、准备工作
加载py包路径，记住目录要用绝对路径：

export PYTHONPATH=$PYTHONPATH:/data/comcode/models/research/:/data/comcode/models/research/slim

创建输出文件夹

cd models/research/deeplab/datasets/cityscapes
mkdir exp
cd exp
mkdir train_on_train_set
cd train_on_train_set
mkdir train  --保存训练输出结果
mkdir val     --保存评价输出结果
mkdir vis      --保存可视化输出结果
mkdir export  --保存模型输出

当前目录结构如下图：

5、训练模型

cd models/research
python deeplab/train.py \--logtostderr \--training_number_of_steps=30000 \--fine_tune_batch_norm=False \--train_split="train" \--model_variant="xception_65" \--atrous_rates=6 \--atrous_rates=12 \--atrous_rates=18 \--output_stride=16 \--decoder_output_stride=4 \--train_crop_size="769,769" \--train_batch_size=1 \--dataset="cityscapes" \--tf_initial_checkpoint=deeplab/datasets/cityscapes/deeplabv3_cityscapes_train/model.ckpt \--train_logdir=deeplab/datasets/cityscapes/exp/train_on_trainval_set/train \--dataset_dir=deeplab/datasets/cityscapes/tfrecord

5、评价模型

python deeplab/eval.py \--logtostderr \--eval_split="val" \--model_variant="xception_65" \--atrous_rates=6 \--atrous_rates=12 \--atrous_rates=18 \--output_stride=16 \--decoder_output_stride=4 \--eval_crop_size="1025,2049" \--dataset="cityscapes" \--checkpoint_dir=deeplab/datasets/cityscapes/exp/train_on_trainval_set/train \--eval_logdir=deeplab/datasets/cityscapes/exp/train_on_trainval_set/val \--dataset_dir=deeplab/datasets/cityscapes/tfrecord \--max_number_of_evaluations=1

6、可视化输出

python deeplab/vis.py \--logtostderr \--vis_split="val" \--model_variant="xception_65" \--atrous_rates=6 \--atrous_rates=12 \--atrous_rates=18 \--output_stride=16 \--decoder_output_stride=4 \--vis_crop_size="1025,2049" \--dataset="cityscapes" \--colormap_type="cityscapes" \--checkpoint_dir=deeplab/datasets/cityscapes/exp/train_on_trainval_set/train \--vis_logdir=deeplab/datasets/cityscapes/exp/train_on_trainval_set/vis \--dataset_dir=deeplab/datasets/cityscapes/tfrecord

tensorboard看运行结果：

tensorboard --port=10072 --logdir=/data/comcode/models/research/deeplab/datasets/cityscapes/exp/train_on_trainval_set/val

图像语义分割模型DeepLab训练Cityscapes数据集过程记录相关推荐

图像语义分割python_图像语义分割 —利用Deeplab v3+训练VOC2012数据集
原标题:图像语义分割 -利用Deeplab v3+训练VOC2012数据集前言: 配置:windows10 + Tensorflow1.6.0 + Python3.6.4(笔记本无GPU) 源码: ...
视频教程-DeepLabv3+图像语义分割实战：训练自己的数据集-计算机视觉
DeepLabv3+图像语义分割实战:训练自己的数据集大学教授,美国归国博士.博士生导师:人工智能公司专家顾问:长期从事人工智能.物联网.大数据研究:已发表学术论文100多篇,授权发明专利10多项 ...
视频教程-U-Net图像语义分割实战：训练自己的数据集-计算机视觉
U-Net图像语义分割实战:训练自己的数据集大学教授,美国归国博士.博士生导师:人工智能公司专家顾问:长期从事人工智能.物联网.大数据研究:已发表学术论文100多篇,授权发明专利10多项白勇 ¥8 ...
深度学习-Tensorflow2.2-图像处理{10}-UNET图像语义分割模型-24
UNET图像语义分割模型简介代码 import tensorflow as tf import matplotlib.pyplot as plt %matplotlib inline import ...
带你1小时掌握Google图像语义分割模型，更有《深度学习》实体书免费送
计算机视觉作为人工智能的主流技术领域之一,历经图像分类-->目标定位-->目标检测,最终发展到图像语义分割技术. 如下图所示,从最初的识别图片信息进行单一分类,到单图片中多目标识别分析,而 ...
用PaddlePaddle实现图像语义分割模型ICNet
什么是图像语义分割? 图像语意分割顾名思义是将图像像素按照表达的语义含义的不同进行分组/分割,图像语义是指对图像内容的理解,例如,能够描绘出什么物体在哪里做了什么事情等,分割是指对图片中的每个像素点进 ...
当前主流的深度学习图像语义分割模型解析
转载自[量子位]公众号 QbitAI 原文地址:http://www.sohu.com/a/155907339_610300 图像语义分割就是机器自动从图像中分割出对象区域,并识别其中的内容. 量子位 ...
一文总结图像语义分割模型
作者:咫尺小厘米链接:https://zhuanlan.zhihu.com/p/133268053 本文转载自知乎,作者已授权,未经许可请勿二次转载语义分割是图像分割中的基本任务,是指对于图像将每 ...
语义分割之VOC2012、Cityscapes数据集介绍
PASCAL-VOC2012 PASCAL-VOC2012数据集介绍官网: 参考数据集下载地址:http://host.robots.ox.ac.uk/pascal/VOC/voc2012/VOCt ...

图像语义分割模型DeepLab训练Cityscapes数据集过程记录

图像语义分割模型DeepLab训练Cityscapes数据集过程记录相关推荐

最新文章

热门文章