feat sketch 提取接口

fix
2024-08-14 17:13:47 +08:00
parent 2505c55bfd
commit 90d8b47884
2 changed files with 10 additions and 3 deletions
--- a/app/service/image2sketch/server.py
+++ b/app/service/image2sketch/server.py
@@ -47,7 +47,7 @@ class Image2SketchServer:
        self.model.setup(self.opt)
        transform_list = [transforms.ToTensor(), transforms.Normalize([0.5], [0.5])]
        transform = transforms.Compose(transform_list)
-        style_img = Image.open(r"E:\workspace\trinity_client_aida\app\service\image2sketch\datasets\ref_unpair\testC\20180422151845_stEe4.jpeg").convert('L')
+        style_img = Image.open(self.opt.style_image).convert('L')
        style_img = transform(style_img)
        self.data['B'] = style_img
        self.data['B'] = self.data['B'].unsqueeze(0).to(device)