48 lines
1.1 KiB
YAML
48 lines
1.1 KiB
YAML
# Data Augmentation Configuration
|
|
# Main configuration file for image data augmentation
|
|
|
|
# Paths configuration
|
|
paths:
|
|
input_dir: "data/IDcards/processed"
|
|
output_dir: "out"
|
|
log_file: "logs/data_augmentation.log"
|
|
|
|
# Data augmentation parameters - ONLY ROTATION
|
|
augmentation:
|
|
# Geometric transformations - ONLY ROTATION
|
|
rotation:
|
|
enabled: true
|
|
angles: [30, 60, 120, 150, 180, 210, 240, 300, 330] # Specific rotation angles
|
|
probability: 1.0 # Always apply rotation
|
|
|
|
# Processing configuration
|
|
processing:
|
|
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"
|
|
quality: 95
|
|
|
|
# Supported image formats
|
|
supported_formats:
|
|
- ".jpg"
|
|
- ".jpeg"
|
|
- ".png"
|
|
- ".bmp"
|
|
- ".tiff"
|
|
|
|
# Logging configuration
|
|
logging:
|
|
level: "INFO" # DEBUG, INFO, WARNING, ERROR
|
|
format: "%(asctime)s - %(name)s - %(levelname)s - %(message)s"
|
|
handlers:
|
|
- type: "file"
|
|
filename: "logs/data_augmentation.log"
|
|
- type: "console"
|
|
|
|
# Performance settings
|
|
performance:
|
|
num_workers: 4
|
|
prefetch_factor: 2
|
|
pin_memory: true
|
|
use_gpu: false |