Parcourir la source

chore: 更新.gitignore和调整路径及模型文件

更新.gitignore文件,将`Outputs/`改为`Output/`以保持一致性。调整`performance_analyzer.py`中的结果目录路径,并更新`inference.py`中的默认模型路径以使用最新模型文件。
Hannnk il y a 3 semaines
Parent
commit
c9dec2a4a3
3 fichiers modifiés avec 3 ajouts et 3 suppressions
  1. 1 1
      .gitignore
  2. 1 1
      src/analysis/performance_analyzer.py
  3. 1 1
      src/core/inference.py

+ 1 - 1
.gitignore

@@ -32,7 +32,7 @@ env/
 *.swo
 
 # Project specific
-Outputs/
+Output/
 Data/
 *.onnx
 *.pkl

+ 1 - 1
src/analysis/performance_analyzer.py

@@ -209,7 +209,7 @@ class PerformanceAnalyzer:
 
 def main():
     # 示例使用
-    result_dir = r"D:\PythonProject\Model\Output\output_20250328_173528_results"
+    result_dir = r"D:\PythonProject\Model\Output\output_20250328_161223_results"
     analyzer = PerformanceAnalyzer(result_dir)
     
     # 分析性能

+ 1 - 1
src/core/inference.py

@@ -20,7 +20,7 @@ class ONNXDetector:
         folder_path = filedialog.askdirectory(title='选择输入目录')
         return folder_path if folder_path else None
 
-    def __init__(self, model_path: str = 'data/models/UAV-250321.onnx', threshold: float = 0.5, 
+    def __init__(self, model_path: str = 'data/models/250411_Anti_UAV.onnx', threshold: float = 0.5, 
                  output_dir: str = "None", save_empty: bool = False, 
                  max_bbox_ratio: float = 0.5,  # 已有该参数
                  input_dir: str = "None"):