I tried to run PyConSe15-cat-vs-dog iPython notebook, but the following line crashes the notebook(The kernel appears to have died. It will restart automatically.):
net = caffe.Classifier(MODEL_FILE, PRETRAINED,
mean=np.load(caffe_root + 'python/caffe/imagenet/ilsvrc_2012_mean.npy').mean(1).mean(1),
channel_swap=(2,1,0),
raw_scale=255,
image_dims=(256, 256))
Am I doing something wrong here?
I have set up Caffe, CUDA and all the prerequisites mentioned in the notebook.