|
CNTK工具使用總結(jié)
一、下載及安裝CNTK（Cognitive Toolkit）是微軟公司開發(fā)的深度學(xué)習(xí)工具箱，本文使用已編譯好的CNTK工具版本，如果想自己編譯可以的話，可以使用源碼版本。官網(wǎng)安裝指南：https://github.com/Microsoft/CNTK/wiki/Automated-binary-install 。 二、使用指南 以MNIST數(shù)據(jù)集為例，簡單介紹下CNTK的使用。 1. 打開Win10系統(tǒng)下的命令提示符窗口，輸入C:\local\cntk\Scripts\cntkpy34.bat，運(yùn)行批處理文件cntkpy34.bat文件，打開CNTK-Python環(huán)境。 2. 在CNTK-Python環(huán)境下，切換C:\repos\CNTK\Examples\Image\DataSets\MNIST文件夾為當(dāng)前目錄，輸入python install_mnist.py命令，運(yùn)行Python腳本install_mnist.py下載MNIST訓(xùn)練集和測試集，并轉(zhuǎn)換為.txt文檔。 3. 將當(dāng)前目錄切換到C:\repos\CNTK\Examples\Image\GettingStarted，輸入命令cntk configFile=01_OneHidden.cntk運(yùn)行配置文件。配置文件由微軟的BrainScript語言編寫，包含訓(xùn)練過程模塊（trainNetwork）和測試過程模塊（testNetwork），trainNetwork模塊里包含對網(wǎng)絡(luò)結(jié)構(gòu)的定義。運(yùn)行該配置文件即可用訓(xùn)練集訓(xùn)練網(wǎng)絡(luò)模型，再用測試集測試之。最終訓(xùn)練完的模型和中間的訓(xùn)練結(jié)果存儲在 .\Output\Models 文件夾下。當(dāng)模型訓(xùn)練完后，可以在C++或者C#語言中（其他語言也可，待探索）部署訓(xùn)練好的模型，在其他的數(shù)據(jù)集中得到預(yù)測結(jié)果。 4. 3中的訓(xùn)練過程和測試過程也可以用Python語言編寫的.py文件實(shí)現(xiàn)，將當(dāng)前目錄切換到C:\repos\CNTK\bindings\python\examples\MNIST，輸入命令python SimpleMNIST.py運(yùn)行.py文件，進(jìn)行網(wǎng)絡(luò)模型的訓(xùn)練與測試。這種方式下模型訓(xùn)練完后，利用何種語言調(diào)用該模型進(jìn)行部署還有待進(jìn)一步探索。 |
|
|
