From 2a83054687f590fd0e3c5fd7293538786c98ce57 Mon Sep 17 00:00:00 2001
From: liyubo <liyubo@ebupt.com>
Date: Thu, 11 Dec 2025 16:20:32 +0800
Subject: [PATCH] =?UTF-8?q?=E6=97=A5=E5=BF=97=E6=89=93=E5=8D=B0=E9=85=8D?=
 =?UTF-8?q?=E7=BD=AE=EF=BC=8C=E8=AF=86=E5=88=AB=E5=AE=8C=E6=88=90=E5=90=8E?=
 =?UTF-8?q?=E6=B8=85=E7=90=86=E4=B8=B4=E6=97=B6=E7=9B=AE=E5=BD=95?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 predict/predict_yolo11seg.py |  1 +
 util/smb_tool.py             | 22 ++++++++++++++++++++++
 yolo_api.py                  |  4 ++++
 3 files changed, 27 insertions(+)

diff --git a/predict/predict_yolo11seg.py b/predict/predict_yolo11seg.py
index b35f5cc..4c70ae9 100644
--- a/predict/predict_yolo11seg.py
+++ b/predict/predict_yolo11seg.py
@@ -710,6 +710,7 @@ class YOLOSegmentationInference:
             move_file_shutil(output_dir+".zip", output_dir)
             remote_dir = f"{tmpConfig['dir']}_识别/{task_id}/{current_time}"
             scanner.upload_directory(output_dir, config['share'], remote_dir=remote_dir)
+            del_file_shutil(output_dir)
 
             return results
 
diff --git a/util/smb_tool.py b/util/smb_tool.py
index 4fd78b2..49f6a52 100644
--- a/util/smb_tool.py
+++ b/util/smb_tool.py
@@ -891,6 +891,28 @@ def move_file_shutil(source_path, destination_path):
         print(f"移动失败: {e}")
         return False
 
+def del_file_shutil(path):
+    """
+    使用shutil删除文件或文件夹
+    
+    Args:
+        path: 文件/文件夹路径
+    """
+    try:
+        # 检查源文件是否存在
+        if not os.path.exists(path):
+            print(f"错误: 文件不存在 - {path}")
+            return False
+        
+        # 执行删除操作
+        shutil.rmtree(path)
+        print(f"成功删除: {path}")
+        return True
+        
+    except Exception as e:
+        print(f"删除失败: {e}")
+        return False
+
 def main():
     # 配置信息
     config = {
diff --git a/yolo_api.py b/yolo_api.py
index dc3a7f7..5bb51dd 100644
--- a/yolo_api.py
+++ b/yolo_api.py
@@ -173,6 +173,7 @@ import sys
 import json
 import time
 import subprocess
+import logging
 from typing import Dict, Optional, List, Any
 from pathlib import Path
 
@@ -239,6 +240,9 @@ app.config.update({
     "DEBUG": Config.DEBUG,
     "WORKERS": Config.WORKERS
 })
+# 设置 smbprotocol 的日志级别
+logging.getLogger("smbprotocol").setLevel(logging.WARNING)
+logging.getLogger("smbprotocol.open").setLevel(logging.WARNING)
 
 # 全局状态
 active_tasks: Dict[str, TaskInfo] = {}