deeplabV3代码运行 Windows10 tensorflow1 pycharm

博主在Windows10上使用Tensorflow1.x和PyCharm运行deeplabV3时遇到了多个错误,包括找不到'deeplab'模块、'tensorflow.compat'无'v2'属性、无法导入'mobilenet_v3'等。通过查阅资料和尝试不同解决方法,如修改环境变量、替换旧版slim模块、调整代码以适配新版本,最终成功运行老版本的models。然而,训练自己数据集的部分仍存在问题。文章提供了错误解决方案和tensorboard查看miou的步骤,并分享了相关资源链接。
摘要由CSDN通过智能技术生成

全程参考: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
    //未成功,安装失败
  • 方法二:

参考:https://bbs.aianaconda.com/thread-790-1-1.html

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目录下

参考:https://blog.csdn.net/yeler082/article/details/82933178

//偶然看到的一个方法,未尝试

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】

参考:https://blog.csdn.net/xjtdw/article/details/92848032

//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
  • 1
    点赞
  • 8
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值