diff --git a/sort.py b/sort.py index e514965..c246068 100644 --- a/sort.py +++ b/sort.py @@ -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']