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')