Explorar o código

refactor: 简化误报特征库文件路径加载逻辑

将误报特征库文件的路径加载逻辑从多层嵌套的`os.path.dirname`调用简化为直接从当前文件路径加载,以提高代码可读性和维护性
Hannnk hai 3 semanas
pai
achega
9e19a06fb5
Modificáronse 1 ficheiros con 1 adicións e 1 borrados
  1. 1 1
      src/core/inference.py

+ 1 - 1
src/core/inference.py

@@ -53,7 +53,7 @@ class ONNXDetector:
         # 加载误报特征库
         self.false_positive_features = {}
         try:
-            with open(os.path.join(os.path.dirname(os.path.dirname(os.path.dirname(__file__))), 'data/false_positive/false_positive_features.pkl'), 'rb') as f:
+            with open(os.path.join(os.path.dirname(__file__), 'false_positive_features.pkl'), 'rb') as f:
                 self.false_positive_features = pickle.load(f)
         except FileNotFoundError:
             print("未找到误报特征库文件,跳过加载")