Commit Graph

13 Commits

Author SHA1 Message Date
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
89b407c564 feat: only copy images with confidence over 95%
Co-authored-by: aider (gemini/gemini-2.5-pro-preview-05-06) <aider@aider.chat>
2025-07-31 16:57:59 -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
c8ba5e51c7 feat: add script to process JSON and sort images 2025-07-31 15:51:17 -06:00
624ff24b52 feat: add move functionality 2025-07-31 15:51:05 -06:00
a9a34eacec Initial commit 2025-07-31 14:50:29 -06:00