caffe mobilenet-ssd测试demo.py报错Check failed: target_blobs.size() == source_layer.blobs_size() (1 v...

原因:


original Caffe 

It looks like that Batchnorm layer and adjacent Scale layer are 

integrated in single Batchnorm layer in NVIDIA caffe. However original 

BVLC/caffe doesn’t integrate them. That means any networks and 

pretrained models using batchnorm and scale layers with original Caffe 

cannot use in NVIDIA caffe and DIGITS.

即:

包含:Batchnorm层和adjacent Scale层分开的; 

NVIDIA Caffe/DIGITS包含:Batchnorm层和adjacent Scale层合并在一个BN层中; 

所以在预测过程中不能使用NVIDIA Caffe,也就是GPU。


解决方法:修改deploy.prototxt


左边为修改前,右边为修改后
©著作权归作者所有,转载或内容合作请联系作者
【社区内容提示】社区部分内容疑似由AI辅助生成,浏览时请结合常识与多方信息审慎甄别。
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

相关阅读更多精彩内容

友情链接更多精彩内容