全程参考:https://www.jianshu.com/p/1a07990705ee
环境-实验室电脑-D:\models-master\research\deeplab,tensorflow-GPU,tensoflow1.13.1,pycharm,python3.6.6,win10,
教程是针对老版本的tensorflow model,官网现下是最新的,所以跑的时候遇到了很多问题。最终还是有一些未解决,直接换了老版本的models,最终顺利跑通。
新版本(针对tensorflow2):官网下载https://github.com/tensorflow/models
老版本(针对教程是第二个):https://blog.csdn.net/qq_27882063/article/details/85127756
在跑最新版时遇到几个问题,记录一下。下次有时间跑一下最新版本。
error1:No module named ‘deeplab’
(这似乎是一个常见问题,跑老版时可能也受益于这个问题的解决)
参考博客:
https://blog.csdn.net/hdu2013/article/details/81177563
- 方法一:
设置系统环境变量PYTHONPATH ,取值(D:\models-master\research;D:\models-master\research\slim;F:\Programs\Python3.6.6\Scripts\)
//最有效方法 - 方法二:
在D:\models-master\research\文件夹下运行cmd命令,运行
SET PYTHONPATH=%cd%;%cd%\slim
//有效,但每次都需要设置
error2:module ‘tensorflow.compat’ has no attribute ‘v2’
- 方法一:
tensorflow版本问题升级到2.0
//未成功,安装失败 - 方法二:
slim版本更新,用老的slim模块替换
//用了这个,出现了问题③,换方法三
- 方法三:
找到报错行
File “D:\models-master\research\slim\nets\mobilenet\mobilenet.py”,
line 399, in def global_pool(input_tensor, pool_op=tf.compat.v2.nn.avg_pool2d): AttributeError: module ‘tensorflow.compat’ has no attribute ‘v2’
将【tf.compat.v2.nn.avg_pool2d】改成【tensorflow.contrib.layers.avg_pool2d()】
//用了这个可行,但又出现问题④。。。
- 方法四:
尝试将整个文件夹放到tensorflow目录下
//偶然看到的一个方法,未尝试
error3:cannot import name ‘mobilenet_v3’
//上面的问题的衍生,未解决,直接跳过
error4:
File “D:\models-master\research\deeplab\datasets\data_generator.py”,
line 290, in _preprocess_image
crop_width=self.crop_size[1],IndexError: list index out of range
版本问题,教程分开传值,新版本合并传值。将X-Train.bat中的这部分修改为【-–train_crop_size=513,513】
//X-Train.bat训练成功了
Error5:参考教程修改eval.py时未找到“slim.evaluation.evaluation_loop”
只找到对应代码块【evaluate_repeatedly】,并尝试修改,未成功。此时才意识到tensormodle版本问题。。。新手伤不起。
//鉴于后续可能还会有很多问题,决定把模型换成老版本
最终都顺利运行了。
待解决:训练自己的数据集。
用tensorboard查看miou步骤:
①在cmd命令行敲入:tensorboard --logdir=log地址
这里是:
tensorboard --logdir=D:\111learning\deeplabv3\models-master\research\deeplab
②打开浏览器输入端口号访问,这里是:
http://localhost:6006
贴一些代码防止系统崩溃。。:
X-remove_gt_colormap.bat 在D:\models-master\research\deeplab\datasets
echo "Removing the color map in ground truth annotations..."
python remove_gt_colormap.py ^
--original_gt_folder="./pascal_voc_seg/VOCdevkit/VOC2012/SegmentationClass" ^
--output_dir="./pascal_voc_seg/VOCdevkit/VOC2012/SegmentationClassRaw"
PAUSE
X-build_voc2012_data.bat在D:\models-master\research\deeplab\datasets
echo "Converting PASCAL VOC 2012 dataset..."
python build_voc2012_data.py ^
--image_folder=".\pascal_voc_seg\VOCdevkit\VOC2012\JPEGImages" ^
--semantic_segmentation_folder="./pascal_voc_seg/VOCdevkit/VOC2012/SegmentationClassRaw" ^
--list_folder="./pascal_voc_seg/VOCdevkit/VOC2012/ImageSets/Segmentation" ^
--image_format="jpg" ^
--output_dir="./pascal_voc_seg/tfrecord"
PAUSE
X-Train.bat 在D:\models-master\research\deeplab
python train.py ^
--logtostderr ^
--training_number_of_steps=30000 ^
--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=513 ^
--train_crop_size=513 ^
--train_batch_size=1 ^
--fine_tune_batch_norm=False ^
--dataset="pascal_voc_seg" ^
--tf_initial_checkpoint=.\model.ckpt ^
--train_logdir=../deeplab ^
--dataset_dir=./datasets/pascal_voc_seg/tfrecord
X-Eval.bat 在D:\models-master\research\deeplab
python eval.py ^
--logtostderr ^
--eval_split="val" ^
--model_variant="xception_65" ^
--atrous_rates=6 ^
--atrous