init
This commit is contained in:
48
config/config.yaml
Normal file
48
config/config.yaml
Normal 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
|
Reference in New Issue
Block a user