在这篇文章的基础下配置caffe运行FCN:点击打开链接
这篇讲的挺细,但是voc-fac32s训练存在几个小问题,如下:
问题1:
Traceback (most recent call last):
File "solve.py", line 19, in <module>
caffe.set_device(int(sys.argv[1]))
IndexError: list index out of range
把对应的两行注释掉,或者利用nvidia-smi看看存在那个gpu就选那个gpu
问题2:
solve.py文件中还应该做如下改动:
vgg_weights = '../ilsvrc-nets/vgg16-fcn.caffemodel'
vgg_proto = '../ilsvrc-nets/VGG_ILSVRC_16_layers_deploy.prototxt'
#weights = '../ilsvrc-nets/vgg16-fcn.caffemodel'
对应的文件下载好放在对应的路径,参考文献中vgg16-fcn.caffemodel是错误的,正确的模型如下点击打开链接(同时这也是造成下一个问题的原因)
问题3:
Canno