fix filename in inference example

wangzhe0912 · Sep 10, 2016 · f15882c · f15882c
1 parent f9cf99a
commit f15882c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/infer.py b/infer.py
@@ -11,7 +11,7 @@
 in_ = in_.transpose((2,0,1))
 
 # load net
-net = caffe.Net('voc-fcn8s/deploy.prototxt', 'voc-fcn8s/fcn8s-heavy-40k.caffemodel', caffe.TEST)
+net = caffe.Net('voc-fcn8s/deploy.prototxt', 'voc-fcn8s/fcn8s-heavy-pascal.caffemodel', caffe.TEST)
 # shape for input (data blob is N x C x H x W), set data
 net.blobs['data'].reshape(1, *in_.shape)
 net.blobs['data'].data[...] = in_