feat sketch 提取接口

fix

app/service/image2sketch/opt.py

@@ -0,0 +1,45 @@
+class Config:
+    def __init__(self):
+        # 基本参数
+        self.dataroot = "service/image2sketch/datasets/ref_unpair"
+        self.name = 'semi_unpair'
+        self.gpu_ids = [0]
+        self.checkpoints_dir = 'service/image2sketch/checkpoints/'
+        # 模型参数
+        self.model = 'unpaired'
+        self.input_nc = 3
+        self.output_nc = 3
+        self.ngf = 64
+        self.ndf = 64
+        self.netD = 'basic'
+        self.netG = 'ref_unpair_cbam_cat'
+        self.netG2 = 'ref_unpair_recon'
+        self.n_layers_D = 3
+        self.norm = 'instance'
+        self.init_type = 'normal'
+        self.init_gain = 0.02
+        self.no_dropout = False  # 对应 `--no_dropout`
+        # 数据集参数
+        self.dataset_mode = 'single'
+        self.direction = 'AtoB'
+        self.serial_batches = True  # 对应 `--serial_batches`
+        self.num_threads = 4
+        self.batch_size = 4
+        self.load_size = 512
+        self.crop_size = 512
+        self.max_dataset_size = float("inf")
+        self.preprocess = 'resize_and_crop'
+        self.no_flip = False  # 对应 `--no_flip`
+        self.display_winsize = 256
+        # 额外参数
+        self.epoch = '100'
+        self.load_iter = 0
+        self.verbose = False  # 对应 `--verbose`
+        self.suffix = ''
+        self.isTrain = False
+        self.results_dir = 'service/image2sketch/results'
+        self.aspect_ratio = 1.0
+        self.phase = 'test'
+        self.eval = False
+        self.num_test = 1000
+        self.morm = 'batch'