junyanz / interactive-deep-colorization

Deep learning software for colorizing black and white images with a few clicks.
https://richzhang.github.io/ideepcolor/
MIT License
2.69k stars 447 forks source link

Incompatible number of blobs for layer conv1_2norm #56

Open LeonSun0101 opened 5 years ago

LeonSun0101 commented 5 years ago

when I load the model using the following code colorModel.prep_net(gpu_id,'./models/reference_model/deploy_nodist.prototxt','./models/reference_model/model.caffemodel') I got error as follows Check failed: target_blobs.size() == source_layer.blobs_size() (5 vs. 3) Incompatible number of blobs for layer conv1_2norm How can I solve the problem? Thx~

junyanz commented 5 years ago

Not sure how to fix it. It might be related to this issue.