10 Commits

Author SHA1 Message Date
05ac4be541 feat: add dropout and weight decay to prevent overfitting
Co-authored-by: aider (gemini/gemini-2.5-pro-preview-05-06) <aider@aider.chat>
2025-07-31 17:18:40 -06:00
77c103b9fe feat: add ColorJitter augmentation to training data
Co-authored-by: aider (gemini/gemini-2.5-pro-preview-05-06) <aider@aider.chat>
2025-07-31 17:09:48 -06:00
d3a4bd7ce9 feat: add --check-crop flag to verify image cropping
Co-authored-by: aider (gemini/gemini-2.5-pro-preview-05-06) <aider@aider.chat>
2025-07-31 16:55:32 -06:00
c1d11ea3f7 refactor: centralize transform constants in model.py
Co-authored-by: aider (gemini/gemini-2.5-pro-preview-05-06) <aider@aider.chat>
2025-07-31 16:52:23 -06:00
078c893770 fix: handle class imbalance with WeightedRandomSampler
Co-authored-by: aider (gemini/gemini-2.5-pro-preview-05-06) <aider@aider.chat>
2025-07-31 16:50:19 -06:00
5f18d8bce2 refactor: extract model to module and add image sorting script
Co-authored-by: aider (gemini/gemini-2.5-pro-preview-05-06) <aider@aider.chat>
2025-07-31 16:22:46 -06:00
587f38b9ce refactor: Update data directory structure to 'data/labelled' 2025-07-31 16:22:29 -06:00
62743006c8 fix: update triangle crop dimensions 2025-07-31 16:06:27 -06:00
4d6eb70728 feat: add garage door CNN training script
Co-authored-by: aider (gemini/gemini-2.5-pro-preview-05-06) <aider@aider.chat>
2025-07-31 15:58:21 -06:00
64a2b03507 feat: add training script 2025-07-31 15:58:13 -06:00