diff options
| author | Evan Shelhamer <shelhamer@imaginarynumber.net> | 2014-05-23 09:13:44 -0700 |
|---|---|---|
| committer | Evan Shelhamer <shelhamer@imaginarynumber.net> | 2014-05-23 09:21:17 -0700 |
| commit | 211c3c29bb3f3acb6ef5e8c3f341e029d8946143 (patch) | |
| tree | e002b3912147935b6edad289427f000796b04262 | |
| parent | e5fcf030fb2683e7b286b22cde44e3bce2d98229 (diff) | |
| download | caffe-211c3c29bb3f3acb6ef5e8c3f341e029d8946143.tar.gz caffe-211c3c29bb3f3acb6ef5e8c3f341e029d8946143.tar.bz2 caffe-211c3c29bb3f3acb6ef5e8c3f341e029d8946143.zip | |
follow-up on #443 to invert k channels (instead of 3)
| -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 48273007..8bed7046 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: - channel_order_inverse = tuple([channel_order.index(i) for i in (0,1,2)]) + channel_order_inverse = [channel_order.index(i) + for i in range(decaf_in.shape[2])] decaf_in = decaf_in[:, :, channel_order_inverse] if input_scale: decaf_in /= input_scale |