前前后后大概忙了3个月了 中间穿插了导师给的项目和论文的任务 总算把fcn给run起来了
之前也有参考一些博客作为指导 不过博客有的是有误导性的 导致我的loss居高不下 根本不收敛
还有一些奇奇怪怪的问题 我在下面逐一列举
个人有一篇从零开始运行FCN的博客 如果需要的话可以去看下:http://blog.csdn.net/wangkun1340378/article/details/70238290
问题1.使用infer.py时候遇到no display name and no $DISPLAY environment variable
出现这个问题 是由于在远程服务器上运行 并且 服务器上没有显卡 直接导致plt无法运行 所以报错
2个解决方法:
方法一:
在infer.py中加入
import cv2
print type(out)
print out, out.shape
cv2.imwrite("output.png", out)
用cv2来保存图片,并且注释掉plt
不过这样虽然能保存图片 但是图片效果不行 具体表现为颜色不对 结果可以看下图
正确的结果如下所示 大家可以对比一下
如果觉得方法一不行 就采用方法2
方法2:采用python notebook
用本机访问自己在远程服务器上的账号 在本机上执行代码
问题2:lisi out of range
这个是由于solve.py中
caffe.set_device(int(sys.argv[1]))
caffe.set_mode_gpu()
可以把这两行代码注释掉 或者 利用nvidia-smi看看存在那个gpu就选那个gpu
问题3:no module named caffe
方法 在py文件import caffe前加上
import sys
sys.path.append('/home/my/caffe-master/caffe-master/python')
具体路径根据自己实际情况而定
问题4:no module named XXX
解决方法 把fcn.berkeleyvision.org-master目录下所有的py文件统统复制到你的solve.p