Fully Convolutional Networks for Semantic Segmentation

1. 下载源码,caffe,vgg16,ZF,voc2012数据集

2.caffe编译,打开opencv,改matlab地址,INCLUDE_DIRS,LIBRARY_DIRS,打开WITH_PYTHON_LAYER.

3.下载voc-fcn32s,voc-fcn16s以及voc-fcn8s的caffemodel(根据提供好的caffemodel-url),指测试。略微修改infer.py,就可以测试我们自己的图片

4 .import numpy as np

from PIL import Image 

import matplotlib.pyplot as plt

 import caffe

im=Image.open('data/pascal/VOCdevkit/VOC2012/JPEGImages/2007_000129.jpg')

in_=np.array(im,dtype=np.float32)

in_=in_[:,:,::-1]

in_-=np.array((104.00698793,116.66876762,122.67891434))

in_=in_.transpose((2,0,1))

net=caffe.Net('voc-fcn8s/deploy.prototxt','voc-fcn8s/fcn8s-heavy-pascal.caffemodel',caffe.TEST)

net.blobs['data'].reshape(1,*in_.shape)

net.blobs['data'].data[...]=in_

net.forward()out=net.blobs['score'].data[0].argmax(axis=0)

接下来,只需要修改script中的图片路径和model的路径,就可以测试自己的图片了

python ./fcn/infer.py

果想train自己的model,只需要修改一些文件路径设置即可。

最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

推荐阅读更多精彩内容

  • 理论 论文思路 代码实现 其余参考资料 http://www.cnblogs.com/xuanxufeng/p/6...
    SIAAAAAA阅读 460评论 0 0
  • 月如盘,矗于低空,画面圆满,却亦显诡异,静逸如死…… 仙境往往应有仙人入景,可是今夜首先出场的角色...
    海峻阅读 385评论 0 3
  • 我刚刚三十。 我以为,今天会不一样,因为生日,因为三十。 果然,昨晚彻夜失眠,今天一天昏沉,上级要的材...
    雪绢阅读 262评论 0 0
  • 今天中午放学妈妈接我去奶奶家,可是做饭的人换成了爷爷,因为奶奶今天有事去姑姑家了。妈妈把我放到爷爷家就回单位上班了...
    刘子熙小名天天阅读 282评论 0 3