This commit is contained in:
Nguyễn Phước Thành
2025-08-05 19:09:55 +07:00
commit 24060e4ce7
25 changed files with 2268 additions and 0 deletions

48
config/config.yaml Normal file
View File

@@ -0,0 +1,48 @@
# Data Augmentation Configuration
# Main configuration file for image data augmentation
# Paths configuration
paths:
input_dir: "data/Archive"
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: [224, 224] # [width, height]
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