前面编译成功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的使用方法: