refactor YOLO crop model

2025-08-05 20:53:47 +07:00
parent a4e7573dca
commit 3fd270c8bd
24 changed files with 136 additions and 479 deletions
--- a/config/config.yaml
+++ b/config/config.yaml
@@ -3,7 +3,7 @@

 # Paths configuration
 paths:
-  input_dir: "data/Archive"
+  input_dir: "data/IDcards/processed"
  output_dir: "out"
  log_file: "logs/data_augmentation.log"

@@ -17,7 +17,7 @@ augmentation:

 # Processing configuration
 processing:
-  target_size: [224, 224]  # [width, height]
+  target_size: [640, 640]  # [width, height] - Increased for better coverage
  batch_size: 32
  num_augmentations: 3  # number of augmented versions per image
  save_format: "jpg"