1、Cannot copy param 0 weights from 'xxxx'
使用预训练的模型,当希望修改部分网络结构并继续使用相同结构的网络参数时,经常遇见这个问题。以为改的天衣无缝,但是这个问题始终存在,而且无视你的修改。比如,网络层 conv1明明已经改成了conv_1了,可是还是提示Cannot copy param 0 weights from 'conv1'。。。。后来才发现,训练的时候也是会用到test.prototxt的,因此必须同时将test里面的模型也改了,才能顺利进行。
记录一下,并惋惜一下浪费掉的两天时间。
2、 No module named caffe.proto 解决方法
参考下面的链接:
https://blog.youkuaiyun.com/curious999/article/details/81225624
就是一个.py文件中的绝对路径出错了。
3、使用GPU进行部署
caffe提供的SSD的demo默认是CPU模式,在deploy中设置为CUDNN也没用,这时候需要在运行前设置一句话,caffe.set_mode_gpu(),这时候你会发现模型推理速度会提升一个数量级!惊喜~~