Update imagenet/wrapper.py to use the new Net interface

diff --git a/python/caffe/imagenet/wrapper.py b/python/caffe/imagenet/wrapper.py
index c29f1ab..4a5b6ed 100644 (file)
--- a/python/caffe/imagenet/wrapper.py
+++ b/python/caffe/imagenet/wrapper.py
@@ -76,7 +76,7 @@ class ImageNetClassifier(object):
       num = 1
     else:
       num = 10
-    self.caffenet = caffe.CaffeNet(model_def_file, pretrained_model)
+    self.caffenet = caffe.Net(model_def_file, pretrained_model)
     self._output_blobs = [np.empty((num, num_output, 1, 1), dtype=np.float32)]
     self._center_only = center_only