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

refactor: Update data directory structure to 'data/labelled'

Browse Source
This commit is contained in:
Tanner Collin
2025-07-31 16:22:29 -06:00
committed by Tanner Collin (aider)
parent 62743006c8
commit 587f38b9ce
2 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

0
sort.py Normal file
View File

4
train.py
View File

@@ -78,7 +78,7 @@ class GarageDoorCNN(nn.Module):
def train_model(): def train_model():
# --- Hyperparameters and Configuration --- # --- Hyperparameters and Configuration ---
DATA_DIR = 'data' DATA_DIR = 'data/labelled'
MODEL_SAVE_PATH = 'garage_door_cnn.pth' MODEL_SAVE_PATH = 'garage_door_cnn.pth'
NUM_EPOCHS = 10 NUM_EPOCHS = 10
BATCH_SIZE = 32 BATCH_SIZE = 32
@@ -158,7 +158,7 @@ def train_model():
if __name__ == '__main__': if __name__ == '__main__':
# Check if data directory exists # Check if data directory exists
if not os.path.isdir('data/open') or not os.path.isdir('data/closed'): if not os.path.isdir('data/labelled/open') or not os.path.isdir('data/labelled/closed'):
print("Error: Data directories 'data/open' and 'data/closed' not found.") print("Error: Data directories 'data/open' and 'data/closed' not found.")
print("Please create them and place your image snapshots inside.") print("Please create them and place your image snapshots inside.")
else: else: