feat sketch 提取接口
fix
This commit is contained in:
@@ -1,10 +1,13 @@
|
||||
from app.core.config import DEBUG
|
||||
|
||||
|
||||
class Config:
|
||||
def __init__(self):
|
||||
# 基本参数
|
||||
self.dataroot = "service/image2sketch/datasets/ref_unpair"
|
||||
self.dataroot = "app/service/image2sketch/datasets/ref_unpair"
|
||||
self.name = 'semi_unpair'
|
||||
self.gpu_ids = [0]
|
||||
self.checkpoints_dir = 'service/image2sketch/checkpoints/'
|
||||
self.checkpoints_dir = 'app/service/image2sketch/checkpoints/'
|
||||
# 模型参数
|
||||
self.model = 'unpaired'
|
||||
self.input_nc = 3
|
||||
@@ -43,3 +46,7 @@ class Config:
|
||||
self.eval = False
|
||||
self.num_test = 1000
|
||||
self.morm = 'batch'
|
||||
if DEBUG:
|
||||
self.style_image = "service/image2sketch/datasets/ref_unpair/testC/20180422151845_stEe4.jpeg"
|
||||
else:
|
||||
self.style_image = "app/service/image2sketch/datasets/ref_unpair/testC/20180422151845_stEe4.jpeg"
|
||||
|
||||
Reference in New Issue
Block a user