logo
Browse Source

Return BGR

Signed-off-by: shiyu22 <shiyu.chen@zilliz.com>
main
shiyu22 3 years ago
parent
commit
ff0a369482
  1. 2
      README.md
  2. 2
      animegan.py

2
README.md

@ -94,7 +94,7 @@ Takes in a numpy rgb image in channels first. It transforms input into animated
**Returns**: *towhee.types.Image (a sub-class of dumpy.ndarray)*
**Returns**: *towhee.types.Image (a sub-class of numpy.ndarray)*
​ The new image.

2
animegan.py

@ -39,4 +39,4 @@ class Animegan(Operator):
styled_image = numpy.array(styled_image)
styled_image = styled_image[:, :, ::-1].copy()
return Image(styled_image, 'RGB')
return Image(styled_image, 'BGR')

Loading…
Cancel
Save