开放原子开发者工作坊

最近接触了百度的开源深度学习平台PaddlePaddle，想把使用的过程记录下来。

作用：按照这篇文章，能够实现对图像的训练和预测。我们准备了四种颜色的海洋球数据，然后给不同颜色的海洋球分类为0123四种。

---

 

一、安装paddlepaddle

1.系统：ubuntu14.04 64位

　　安装paddlepaddle：pip install paddlepaddle

　　注意：安装后可直接用train.py进行训练，然后用infer.py进行预测；如果安装失败，请更新pip，如果仍然不行，请使用docker。

2.安装docker

　　官网的安装教程：https://docs.docker.com/engine/installation/linux/docker-ce/ubuntu/#install-using-the-convenience-script

　　不想查看官网教程，那直接执行下面命令就可以。

　　sudo apt-get remove docker docker-engine

 

　　sudo apt-get update

　　sudo apt-get install Linux-image-extra-$(uname -r) linux-image-extra-virtual　　

　　sudo apt-get install apt-transport-https ca-certificates curl software-properties-common

 

　　curl -f sSL https://download.docker.com/linux/ubuntu/gpg | sudo apt-key add -

　　sudo add-apt-repository "deb [arch=amd64] https://download.docker.com/linux/ubuntu $(lsb_release -cs) stable"

 

　　sudo apt-get update

　　sudo apt-get install docker.io

3.配置docker加速器

　　国内从docker hub上push会遭到拒绝（防火墙的原因），FQ可以解决这个问题，还有就是正规渠道，配置docker加速器。

　　通过DaoCloud配置docker加速器：

　　curl -sSL https://get.daocloud.io/daotools/set_mirror.sh | sh -s http://d1a927e1.m.daocloud.io

　　注意：如果上面指令不可用，可以自己注册DaoCloud，登录后可以找到类似于上方的配置docker加速器的指令。

4.拉取PaddlePaddle镜像

　　sudo docker pull paddlepaddle/paddle:latest

---

 

二、训练

1.下载文件

　　git clone https://github.com/zzx2GH/FourClassificationModel.git

　　注意:下载后的文件请放在home目录下

2.将基于paddlepaddle的项目挂载到Paddle Docker镜像

　　sudo docker run -it -v /home/FourClassificationModel:/home paddlepaddle/paddle:latest /bin/bash

3.训练

　　cd /home

　　python train.py

　　训练时会在每10个pass后，在本地保存一个.tar的包，程序设定了200个pass，会在经历了200个pass后自动结束训练。

---

 

三、预测

　　python infer.py

　　注意：我们把球分成了四类，分别对应0，1，2，3；我们预测时读取test下的文件，例如：如果读取./test/class1/1目录下的图像，那预测结果应该是1

　　如果你想预测其他类别的球，请在infer.py文件中第65行文件路径处更改数字为对应类别的目录，例如要预测./test/class1/1目录下的图像，代码 file = './test/class3/3/'+i+'.png'

---

  

关于深度学习的详细内容可以参考PaddlePaddle的官方教程：http://www.paddlepaddle.org