前面編譯成功caffe的python接口,就可以 import caffe 調用了。
先通過caffe自帶的draw_net.py 玩玩看。
選擇examples 下 lenet_train_test.prototxt 、bvlc_caffenet_full_conv.prototxt。
一般來說,開始會出現錯誤,因爲缺少 dot.exe 完成 visualization 。
需要安裝
- graphviz 並配置系統環境變量,再pip install pygraphviz。
-
- 重啓生效後Run:
-
- python draw_net.py bvlc_caffenet_full_conv.prototxt b.png
- python draw_net.py bvlc_caffenet_full_conv.prototxt b.png
結果分別爲:
同理:
classify.py 、detect.py的使用方法: