Stupid-Tyro的博客

初始化模型 root=’/home/*/caffe-master’ deploy=root+’examples/mnist/deply.prototxt’ //deploy位置 caffe_model=root+’examples/mnist/lenet_iter_5000.caffemodel’ //模型位置 net=caffe.Net(deploy,caffe_model,caffe.TEST) //加载model和net 参数访问 net.params保存各层的参数值(w和b) net.blo..

安装相关依赖 pip3 install pydot sudo apt-get install graphviz 在caffe-master/python目录下 python3 draw_net.py /home/xxx/caffe-master/examples/mnist/lenet_train_test.prototxt lenet.png