Commit d828f9b7 authored by valentini's avatar valentini
Browse files

Carica un nuovo file

parent 540be1bc
scaling: 2 # Model Scaling Factor (used also for loading the high resolution images)
dataset:
# Note that the DataLoader expect to have specular files (only the base path provided in low_sr_dir and high_sr_dir differ)
low_sr_dir: "path/to/low_sr" # Directory containing low resolution YUV images
high_sr_dir: "path/to/high_sr" # Directory containing high resolution YUV images
width: 256 # Image width
height: 256 # Image height
pix_fmt: "yuv420p" # Pixel format, e.g., yuv420p, yuv444p, yuv422p, yuv444p10le, etc...
range: 255.0 # Value to normalize pixel values (e.g., 255.0 for 8-bit, 1.0 for already normalized)
luminance_only: false # If true, only use the Y (luminance) channel
clone_luminance_as_rgb: false # If true and luminance_only is true, repeat Y channel 3 times to simulate RGB
upsample_uv: false # If true, upsample U and V channels to match Y size if needed
uv_interpolation: "nearest" # Interpolation for upsampling U/V: nearest, linear, cubic, area, lanczos
batch_size: 4 # Batch size for DataLoader (default: 4)
patch_size: 128 # Size of pachets to extract from the Input and output image (randomly sampled) ( the outout patch is scaled )
scheduler:
use_plateau_scheduler: false # If true, use ReduceLROnPlateau; otherwise, use StepLR
lr_step_size: 10 # StepLR: number of epochs between LR reductions
lr_gamma: 0.5 # StepLR/Plateau: factor to reduce LR by
Supports Markdown
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment