代碼主頁: https://github.com/bgshih/aster
1.建立工程目錄
aster-code:
aster-master:
aster:
2.建立鏡像
編寫dockerfile
執行命令:
docker build -t aster-test:v2 .
docker run -i -t aster-test:v2 /bin/bash
3.進入目錄:
cd/..../aster-code/aster/master
執行命令:
docker run -v /..../aster-code/aster-master:/aster-code/aster-master/ -w/aster-code/aster-master aster-test:v2 protoc ./aster/protos/*.proto --python_out=.
於是會在aster-master/aster/protos生成一系列python文件:
在/aster-code/aster-master/aster/c_ops目錄下執行命令:
docker run -v /..../aster-code/aster-master/aster/c_ops:/aster-code/aster-master/aster/c_ops -w/aster-code/aster-master/aster/c_ops aster-test:v2 sh build.sh
執行demo文件:
docker run -v /...../aster-code/aster-master:/aster-code/aster-master/ -w/aster-code/aster-master/ aster-test:v2 python demo.py
則表示demo腳本運行成功!
Remark:本來運行該腳本會報錯:
解決方案是一定要在環境中裝tensorflow==1.4(一定要是這個版本!)
pip install tensorflow==1.4
最後測試的圖片是: