changes · 0e8901749e - animegan - Towhee

changes

main

Filip 3 years ago

parent

3ed918cf34

commit

0e8901749e

1 changed files with 2 additions and 1 deletions

Unified View

							
								1

animegan.py

								View File
							
				@ -30,5 +30,6 @@ class Animegan(Operator):
			
				    @arg(1, to_image_color('RGB'))
				
				    @arg(1, to_image_color('RGB'))
			
				    def __call__(self, image):
				
				    def __call__(self, image):
			
				        img = self.tfms(image).unsqueeze(0)
				
				        img = self.tfms(image).unsqueeze(0)
			
				        print(img.shape)
			
				        styled_image = self.model(img)
				
				        styled_image = self.model(img)
			
				        return Image(styled_image, 'RGB')
				
				        return Image(styled_image, 'RGB')