caffe: Echec de la vérification: target_blobs.size() == source_layer.blobs_size() (2 vs. 1) Nombre de blobs incompatibles pour la couche conv1

Question

Je modifie le réseau FCN et crée un nouveau réseau, dans lequel j'utilise deux ImageData Couche comme paramètre d'entrée et espère que le réseau produit une image en sortie. voici le train_val.prototxt et deploy.prototxt

l'image originale et l'étiquette sont à la fois échelle de gris et les tailles sont 224 * 224. Je me suis entraîné un caffemodel et utiliser infer.py d'utiliser la caffemodel faire une segmentation, mais répondre à l'erreur:

F0505 06:15:08.072602 30713 net.cpp:767] Check failed: target_blobs.size() == source_layer.blobs_size() (2 vs. 1) Incompatible number of blobs for layer conv1 

est le fichier infer.py ici:

import numpy as np 
from PIL import Image 
caffe_root = '/home/zhaimo/' 
import sys 
sys.path.insert(0, caffe_root + 'caffe-master/python') 

import caffe 
im = Image.open('/home/zhaimo/fcn-master/data/vessel/test/13.png') 
in_ = np.array(im, dtype=np.float32) 
#in_ = in_[:,:,::-1] 
#in_ -= np.array((104.00698793,116.66876762,122.67891434)) 
#in_ = in_.transpose((2,0,1)) 


net = caffe.Net('/home/zhaimo/fcn-master/mo/deploy.prototxt', '/home/zhaimo/fcn-master/mo/snapshot/train/_iter_200000.caffemodel', caffe.TEST) 
net.blobs['data'].reshape(1, *in_.shape) 
net.blobs['data'].data[...] = in_ 
net.forward() 
out = net.blobs['score'].data[0].argmax(axis=0) 

plt.axis('off') 
plt.savefig('/home/zhaimo/fcn-master/mo/result/13.png') 

comment résoudre ce problème?

Answer 1

Le problème est avec votre terme de biais dans conv1. Dans votre train.prototxt, il est défini sur false. Mais dans votre deploy.prototxt ce n'est pas le cas et par défaut c'est vrai. C'est pourquoi le loader de poids recherche deux blobs.