deeplab_V3+训练自己的数据集(用macbook制作毕业设计:链蓖机台车侧板裂缝 全流程)

deeplab_V3+训练自己的数据集(用macbook制作毕业设计:链蓖机台车侧板裂缝 全流程)

主要参考博客:
https://blog.csdn.net/malvas/article/details/90776327
https://blog.csdn.net/hccq1/article/details/95990148

一、配置环境
1.1 配置信息
系统: macOS Catalina 10.15.4

python版本: Python3.6.4

tensorflow版本: tensorflow——cpu 1.13.rcl
(由于tensorflow1.2以上版本macos不支持GPU,这里我选择安装tensorflow-CPU 1.13.0rc1 但是这样效率极低,对于batch_size=12 crop_size=[321,321]的训练尺寸依旧需要67.573 sec/step。这里具体后面的章节会细谈。)

Deeplab版本:Deeplab_V3+ r1.13.0
1.2 环境搭建
Mac对于内存管理要求很严格,如果不创建一个虚拟环境可能会导致出现一个错误后所有的库都无法正常运行。所以先要搭建一个虚拟环境来分隔开:
1)先安装anaconda(官网:https://www.anaconda.com),接下来需要利用conda来创建一个虚拟环境。anaconda的Mac版安装非常简单,一直next就可以了。
2)在anaconda安装好后重新打开终端,可以看到在用户名前出现了(base)的的提示,这说明anaconda安装成功,基本的环境已经有了。
图 1 环境搭建
接下来我们需要创建一个新的环境,输入在终端输入命令:conda create -n deeplab python=3.6.4(这里deeplab是我的环境名称,可以自己随意选择名称;python版本最好不要超过3.7,因为Mac版的tensorflow超过3.7会出现无法安装的情况,这也是我刚一开始搭建环境遇到的问题);在运行命令后可能会出现安装不成功的问题,这是因为原来的下载源连接速率比较慢,我们可以选择更改conda源为清华源:在终端输入命令:

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
conda config --set show_channel_urls yes

这样再创建环境就成功了。
3)这里我们需要激活刚才所创建的环境,终端输入命令:source activate deeplab来激活环境。激活后刚才的(base)就变成了(deeplab),这样环境搭建就基本完成了。
4)安装所需要的基本程序和库:这里推荐使用pip安装,conda安装后可能会出现ModuleNotFoundError。
a.首先需要安装Numpy:在终端输入
pip install numpy -i https://pypi.tuna.tsinghua.edu.cn/simple
b.接着同样的方法安装pillow:在终端输入
pip install pillow -i https://pypi.tuna.tsinghua.edu.cn/simple
c.Jupyter notebook : 在终端输入pip install Jupyter notebook -i https://pypi.tuna.tsinghua.edu.cn/simple
d.Matplotlib:在终端输入pip install Matplotlib -i https://pypi.tuna.tsinghua.edu.cn/simple
e.需要注意的是Tensorflow安装我们需要安装1.13.0版本,1.13以上的版本可能会无法识别,进而会出现ModuleNotFoundError,所以这里需要安装tensorflow1.13.0:在终端输入:pip install tensorflow==1.13.0rc0 -i https://pypi.tuna.tsinghua.edu.cn/simple
到此环境搭建完成。
1.3源码准备
在deeplab官网https://github.com/tensorflow/models下载分支r1.13.0。由于model-master增加了对tensorflow2.0以上的支持,在Mac上运行会出现诸多错误,困难太多了所以我们放弃model-master版本。所以选择branch r1.13.0来下载。

1.4环境配置
按照官方安装说明安装必要库:Numpy、Pillow 1.0、tf Slim (which is included in the “tensorflow/models/research/” checkout)、Jupyter notebook、Matplotlib、Tensorflow,这些库我们已经在1.2的最后一个环节安装过了。

图 2 运行所需库
这里的tf Slim已经在research文件夹里了,所以每次运行前需要添加python路径,在终端里进入到research文件夹后运行命令:

export PYTHONPATH=$PYTHONPATH:`pwd`:`pwd`/slim

图 3 路径配置
依旧在research文件夹里运行测试命令(由于是cpu运行,这里我用了22.492s,而gpu则会快很多):

python deeplab/model_test.py

图 4 测试代码
在运行时可能会报错:

ASSertionError:First sequence</
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值