FCN模型训练中遇到的困难

前前后后大概忙了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