From c24ade7696f0d9458a95ce6e62ba219fdb9804ee Mon Sep 17 00:00:00 2001 From: zhouchengrong Date: Fri, 16 Aug 2024 15:31:59 +0800 Subject: [PATCH] =?UTF-8?q?feat=20=20=20sketch=20=E6=8F=90=E5=8F=96?= =?UTF-8?q?=E6=8E=A5=E5=8F=A3=20fix?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- app/service/image2sketch/server.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/app/service/image2sketch/server.py b/app/service/image2sketch/server.py index ebd363e..82f1843 100644 --- a/app/service/image2sketch/server.py +++ b/app/service/image2sketch/server.py @@ -52,8 +52,7 @@ class Image2SketchServer: self.data['B'] = style_img self.data['B'] = self.data['B'].unsqueeze(0).to(device) A, self.width, self.height = self.get_image(self.image_url) - self.data['A'] = transform(A) - self.data['A'] = self.data['A'].unsqueeze(0).to(device) + def get_result(self): self.model.set_input(self.data)