fix: lower image sorting confidence threshold
This commit is contained in:
2
sort.py
2
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']
|
||||
|
Reference in New Issue
Block a user