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

fix: update triangle crop dimensions

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

4
train.py
View File

@@ -84,8 +84,8 @@ def train_model():
BATCH_SIZE = 32 BATCH_SIZE = 32
LEARNING_RATE = 0.001 LEARNING_RATE = 0.001
# For the custom crop transform. User can adjust these. # For the custom crop transform. User can adjust these.
TRIANGLE_CROP_WIDTH = 400 TRIANGLE_CROP_WIDTH = 556
TRIANGLE_CROP_HEIGHT = 400 TRIANGLE_CROP_HEIGHT = 1184
RESIZE_DIM = 64 # Resize cropped image to this dimension (square) RESIZE_DIM = 64 # Resize cropped image to this dimension (square)
# --- Data Preparation --- # --- Data Preparation ---