最近接觸了百度的開源深度學習平臺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
?