yahoo / open_nsfw

Not Suitable for Work (NSFW) classification using deep neural network Caffe models.

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

'Incompatible number of blobs for layer bn_1' - Help please!

midaha opened this issue · comments

When running nsfw_classify.py as per the example code I get the following error:

Check failed: target_blobs.size() == source_layer.blobs_size() (5 vs. 3) Incompatible number of blobs for layer bn_1
*** Check failure stack trace: ***

Does the deploy.prototxt and weights not line up properly? I've downloaded the files several times (to check it didn't corrupt) - and not modified anything.

Any tips?

which caffe version did you use? i think you can you the BVLC newest caffe to try

NVIDIA/DIGITS#1648

this worked for me