feat sketch 提取接口

fix
2024-08-15 10:24:19 +08:00
parent 8b876b103f
commit 7ff3a72d8c
3 changed files with 5 additions and 4 deletions
--- a/app/service/image2sketch/server.py
+++ b/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