fix: lower image sorting confidence threshold

This commit is contained in:
2025-07-31 17:13:26 -06:00
committed by Tanner Collin (aider)
parent 77c103b9fe
commit f7aa4e6e06

View File

@@ -13,7 +13,7 @@ def sort_images():
MODEL_PATH = 'garage_door_cnn.pth'
SOURCE_DIR = 'data/hourly_photos/'
DEST_DIR = 'data/sorted/open/'
CONFIDENCE_THRESHOLD = 0.95 # Only copy if confidence is over this value
CONFIDENCE_THRESHOLD = 0.90 # Only copy if confidence is over this value
# The classes are sorted alphabetically by ImageFolder: ['closed', 'open']