Python下Caffe的使用

前面編譯成功caffe的python接口,就可以 import  caffe 調用了。

先通過caffe自帶的draw_net.py 玩玩看。


選擇examples 下 lenet_train_test.prototxt  、bvlc_caffenet_full_conv.prototxt。

一般來說,開始會出現錯誤,因爲缺少 dot.exe 完成 visualization 。


需要安裝

  1. graphviz 並配置系統環境變量,再pip install pygraphviz
    1. 重啓生效後Run:
      1. python draw_net.py bvlc_caffenet_full_conv.prototxt b.png

結果分別爲:




同理:

classify.py 、detect.py的使用方法:





發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章