diff options
Diffstat (limited to 'python')
| -rw-r--r-- | python/caffe/pycaffe.py | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/python/caffe/pycaffe.py b/python/caffe/pycaffe.py index 537e737b..48273007 100644 --- a/python/caffe/pycaffe.py +++ b/python/caffe/pycaffe.py @@ -284,7 +284,8 @@ def _Net_deprocess(self, input_name, input_): decaf_in += mean decaf_in = decaf_in.transpose((1,2,0)) if channel_order: - decaf_in = decaf_in[:, :, channel_order[::-1]] + channel_order_inverse = tuple([channel_order.index(i) for i in (0,1,2)]) + decaf_in = decaf_in[:, :, channel_order_inverse] if input_scale: decaf_in /= input_scale return decaf_in |