PaddlePaddle开源平台的应用

最近接触了百度的开源深度学习平台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

 

posted @ 2017-09-14 10:49  自由的小巴车  阅读(663)  评论(0编辑  收藏  举报