XW-AIOT/security-ai-edge
4
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat(inference): 提高置信度阈值到0.5 - 减少误检

Browse Source 
- conf_threshold: 0.4 → 0.5
- 提高检测精度，减少误报
- 同时更新环境变量默认值

Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
This commit is contained in:
16337
2026-02-14 11:13:45 +08:00
parent 6f9cb9d960
commit 0bb9f7ebd2
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
config/settings.py
View File

@@ -117,7 +117,7 @@ class InferenceConfig:
input_width: int = 480 input_width: int = 480
input_height: int = 480 input_height: int = 480
batch_size: int = 1 batch_size: int = 1
conf_threshold: float = 0.4 conf_threshold: float = 0.5
nms_threshold: float = 0.45 nms_threshold: float = 0.45
device_id: int = 0 device_id: int = 0
fp16_mode: bool = True fp16_mode: bool = True
@@ -273,7 +273,7 @@ class Settings:
input_width=int(os.getenv("INPUT_WIDTH", "480")), input_width=int(os.getenv("INPUT_WIDTH", "480")),
input_height=int(os.getenv("INPUT_HEIGHT", "480")), input_height=int(os.getenv("INPUT_HEIGHT", "480")),
batch_size=int(os.getenv("BATCH_SIZE", "4")), batch_size=int(os.getenv("BATCH_SIZE", "4")),
conf_threshold=float(os.getenv("CONF_THRESHOLD", "0.4")), conf_threshold=float(os.getenv("CONF_THRESHOLD", "0.5")),
nms_threshold=float(os.getenv("NMS_THRESHOLD", "0.45")), nms_threshold=float(os.getenv("NMS_THRESHOLD", "0.45")),
) )