tanner/doormind
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Adjust constants

Browse Source
This commit is contained in:
Tanner Collin
2025-07-31 18:44:05 -06:00
parent aa8f87a6d3
commit 1b5fee77c2
3 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
sort.py
View File

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