net = caffe.Net(MODEL_FILE,PRETRAINED)
net.set_mode_cpu()
data4D = np.zeros([max_value,1,96,96],np.float32)
data4DL = np.zeros([max_value,30,1,1], np.float32)
data4D[0:max_value,:] = X[:1000,:,:,:]
net.set_input_arrays(data4D.astype(np.float32),data4DL.astype(np.float32))
net.forward()
Traceback (most recent call last):
File "output.py", line 51, in <module>
net.set_input_arrays(data4D.astype(np.float32),data4DL.astype(np.float32))
File "/home/pbu/Desktop/caffe/python/caffe/pycaffe.py", line 345, in _Net_set_input_arrays
return self._set_input_arrays(data, labels)
RuntimeError: labels array has wrong number of channels