Skip to content
GitLab
Projects
Groups
Snippets
/
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in
Toggle navigation
Menu
Open sidebar
gapou
caffe
Commits
211c3c29
Commit
211c3c29
authored
May 23, 2014
by
Evan Shelhamer
Browse files
follow-up on #443 to invert k channels (instead of 3)
parent
e5fcf030
Changes
1
Hide whitespace changes
Inline
Side-by-side
python/caffe/pycaffe.py
View file @
211c3c29
...
...
@@ -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
...
...
Write
Preview
Supports
Markdown
0%
Try again
or
attach a new file
.
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment