From ff0a369482f40ecf219a9acd2b6686bc360466de Mon Sep 17 00:00:00 2001 From: shiyu22 Date: Mon, 23 May 2022 17:41:19 +0800 Subject: [PATCH] Return BGR Signed-off-by: shiyu22 --- README.md | 2 +- animegan.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/README.md b/README.md index 1bd3d6e..246770a 100644 --- a/README.md +++ b/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. diff --git a/animegan.py b/animegan.py index 0982a80..161a339 100644 --- a/animegan.py +++ b/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')