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