aidlab/AiDA_Python
6
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat image2sketch 新增风格上传 自定义风格

Browse Source 
fix
This commit is contained in:
zhouchengrong
2024-09-20 17:03:48 +08:00
parent 1385fde9ce
commit 2e07bc2de9
9 changed files with 28 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
app/service/image2sketch/infer.py
View File

@@ -54,7 +54,7 @@ def load_img(filepath):
if __name__ == '__main__':
img_A = "/workspace/Semi_ref2sketch_code/datasets/ref_unpair/testA/real_Dress_732caedc416a0cbfedd0e6528040eac7.jpg_Img.jpg"
img_B = "/workspace/Semi_ref2sketch_code/datasets/ref_unpair/testC/styleA.png"
img_B = "/workspace/Semi_ref2sketch_code/datasets/ref_unpair/testC/style_3.png"
from opt import Config
opt = Config() # get test options
@@ -73,7 +73,7 @@ if __name__ == '__main__':
model.eval()
data = {}
print(os.getcwd())
B = reference, _, _ = load_img(r"E:\workspace\trinity_client_aida\app\service\image2sketch\datasets\ref_unpair\testC\styleA.png")
B = reference, _, _ = load_img(r"/app/service/image2sketch/datasets/ref_unpair/testC/style_3.png")
style_img = transform(reference)
data['B'] = style_img
data['B'] = data['B'].unsqueeze(0).to(device)