diff --git a/config/settings.py b/config/settings.py index 61673f6..28fec69 100644 --- a/config/settings.py +++ b/config/settings.py @@ -117,7 +117,7 @@ class InferenceConfig: input_width: int = 480 input_height: int = 480 batch_size: int = 1 - conf_threshold: float = 0.5 + conf_threshold: float = 0.4 nms_threshold: float = 0.45 device_id: int = 0 fp16_mode: bool = True @@ -273,7 +273,7 @@ class Settings: input_width=int(os.getenv("INPUT_WIDTH", "480")), input_height=int(os.getenv("INPUT_HEIGHT", "480")), batch_size=int(os.getenv("BATCH_SIZE", "4")), - conf_threshold=float(os.getenv("CONF_THRESHOLD", "0.5")), + conf_threshold=float(os.getenv("CONF_THRESHOLD", "0.4")), nms_threshold=float(os.getenv("NMS_THRESHOLD", "0.45")), )