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

8
app/service/image2sketch/opt.py
View File

@@ -46,8 +46,12 @@ class Config:
self.num_test = 1000
self.morm = 'batch'
if DEBUG:
self.style_image = "service/image2sketch/datasets/ref_unpair/testC/20180422151845_stEe4.jpeg"
self.style_image1 = "service/image2sketch/datasets/ref_unpair/testC/style_1.jpg"
self.style_image2 = "service/image2sketch/datasets/ref_unpair/testC/style_2.jpeg"
self.style_image3 = "service/image2sketch/datasets/ref_unpair/testC/style_3.png"
self.checkpoints_dir = 'service/image2sketch/checkpoints/'
else:
self.checkpoints_dir = 'app/service/image2sketch/checkpoints/'
self.style_image = "app/service/image2sketch/datasets/ref_unpair/testC/20180422151845_stEe4.jpeg"
self.style_image1 = "app/service/image2sketch/datasets/ref_unpair/testC/style_1.jpg"
self.style_image2 = "app/service/image2sketch/datasets/ref_unpair/testC/style_2.jpeg"
self.style_image3 = "app/service/image2sketch/datasets/ref_unpair/testC/style_3.png"