美文网首页
Fully Convolutional Networks for

Fully Convolutional Networks for

作者: 3152af65e389 | 来源:发表于2016-11-28 22:20 被阅读0次

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,只需要修改一些文件路径设置即可。

相关文章

网友评论

      本文标题:Fully Convolutional Networks for

      本文链接:https://www.haomeiwen.com/subject/tqxapttx.html