From: Noiredd Date: Wed, 12 Apr 2017 09:59:06 +0000 (+0200) Subject: fixes pycaffe forward() and backward() behavior for nets whose layer names do not... X-Git-Url: http://review.tizen.org/git/?a=commitdiff_plain;h=35a7b87ad87457291dfc79bf8a7e7cf7ef278cbb;p=platform%2Fupstream%2Fcaffe.git fixes pycaffe forward() and backward() behavior for nets whose layer names do not match respective tops --- diff --git a/python/caffe/pycaffe.py b/python/caffe/pycaffe.py index 6360659..4a7b5a2 100644 --- a/python/caffe/pycaffe.py +++ b/python/caffe/pycaffe.py @@ -113,7 +113,7 @@ def _Net_forward(self, blobs=None, start=None, end=None, **kwargs): if end is not None: end_ind = list(self._layer_names).index(end) - outputs = set([end] + blobs) + outputs = set(self.top_names[end] + blobs) else: end_ind = len(self.layers) - 1 outputs = set(self.outputs + blobs) @@ -161,7 +161,7 @@ def _Net_backward(self, diffs=None, start=None, end=None, **kwargs): if end is not None: end_ind = list(self._layer_names).index(end) - outputs = set([end] + diffs) + outputs = set(self.bottom_names[end] + diffs) else: end_ind = 0 outputs = set(self.inputs + diffs)