feat sketch 提取接口

fix

app/service/image2sketch/opt.py

@@ -7,7 +7,6 @@ class Config:
         self.dataroot = "app/service/image2sketch/datasets/ref_unpair"
         self.name = 'semi_unpair'
         self.gpu_ids = [0]
-        self.checkpoints_dir = 'app/service/image2sketch/checkpoints/'
         # 模型参数
         self.model = 'unpaired'
         self.input_nc = 3
@@ -48,5 +47,7 @@ class Config:
         self.morm = 'batch'
         if DEBUG:
             self.style_image = "service/image2sketch/datasets/ref_unpair/testC/20180422151845_stEe4.jpeg"
+            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"

app/service/image2sketch/server.py

@@ -55,9 +55,6 @@ class Image2SketchServer:
         self.data['A'] = transform(A)
         self.data['A'] = self.data['A'].unsqueeze(0).to(device)
 
-    def __del__(self):
-        torch.cuda.empty_cache()
-
     def get_result(self):
         self.model.set_input(self.data)
         self.model.test()  # run inference