diff options
Diffstat (limited to 'python/caffe/classifier.py')
-rw-r--r-- | python/caffe/classifier.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/python/caffe/classifier.py b/python/caffe/classifier.py index 983760a7..64d804be 100644 --- a/python/caffe/classifier.py +++ b/python/caffe/classifier.py @@ -23,7 +23,7 @@ class Classifier(caffe.Net): def __init__(self, model_file, pretrained_file, image_dims=None, mean=None, input_scale=None, raw_scale=None, channel_swap=None): - caffe.Net.__init__(self, model_file, pretrained_file, caffe.TEST) + caffe.Net.__init__(self, model_file, caffe.TEST, weights=pretrained_file) # configure pre-processing in_ = self.inputs[0] |