Check failed: registry.count(type) == 1 (0 vs. 1) Unknown layer type: Normalize (known types: AnnotatedData, BatchNorm, Bias, Concat, Convolution, DetectionEvaluate, DetectionOutput, Eltwise, Flatten, Input, LRN, MemoryData, MultiBoxLoss, Permute, Pooling, Power, PriorBox, Python, ReLU, Reshape, Scale, Sigmoid, SmoothL1Loss, Softmax, SoftmaxWithLoss, Split, TanH)
Windows 下用 build_win.cmd 直接编译CPU版caffe
从上面博客的第3步 —— 3.修改Caffe配置文件 开始操作,根据自己的需求进行相关配置修改就可以了
配置GPU版本 下面的步骤以及硬件要求详见我的另一篇博客:Windows 下用 build_win.cmd 直接编译GPU版caffe
需要提前安装 CUDA+cudnn ,然后从上面博客的第5步 —— 5.修改Caffe配置文件 开始操作,根据自己的需求进行相关配置修改就可以
经过上面第2步成功编译以后,Pycaffe 已经编译完成了,剩下的就是将相关的 caffe_ssd_windows/python 路径加入到环境变量。 具体操作见我的另一篇博客:Windows下 Pycaffe 的配置与使用
在此处下载自己需要的预训练模型 https://github.com/weiliu89/caffe/tree/ssd , 需要模型的 deploy.prototxt 文件和 对应的 caffemodel,下载完模型后在 caffe_ssd_windows/examples/ssd/ 中有一个名为 ssd-windows.py 的文件,里面的内容基本和caffe_ssd_windows/examples/下的 ssd_detect.ipynb 一样,这里为了测试方便转成了 .py 文件。labelmap_voc.prototxt在我们可以在 caffe_ssd_windows/data/VOC0712 找到,就可以直接在Pycharm 中 运行。
最后就可以成功运行啦!