Browse Source
Return BGR
Signed-off-by: shiyu22 <shiyu.chen@zilliz.com>
main
2 changed files with
2 additions and
2 deletions
-
README.md
-
animegan.py
|
|
@ -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. |
|
|
The new image. |
|
|
|
|
|
|
|
|
|
|
|
@ -39,4 +39,4 @@ class Animegan(Operator): |
|
|
styled_image = numpy.array(styled_image) |
|
|
styled_image = numpy.array(styled_image) |
|
|
styled_image = styled_image[:, :, ::-1].copy() |
|
|
styled_image = styled_image[:, :, ::-1].copy() |
|
|
|
|
|
|
|
|
return Image(styled_image, 'RGB') |
|
|
|
|
|
|
|
|
return Image(styled_image, 'BGR') |
|
|
|
