Spaces:
Running
Running
| import torch | |
| from pathlib import Path | |
| RELU = "ReLU" | |
| LEAKY_RELU = "LeakyReLU" | |
| SIMPLE_GATE = "simple_gate" | |
| LOSS = "loss" | |
| LOSS_MSE = "MSE" | |
| METRIC_PSNR = "PSNR" | |
| METRIC_SSIM = "SSIM" | |
| METRIC_LPIPS = "LPIPS" | |
| SELECTED_METRICS = "selected_metrics" | |
| DATALOADER = "data_loader" | |
| BATCH_SIZE = "batch_size" | |
| SIZE = "size" | |
| TRAIN, VALIDATION, TEST = "train", "validation", "test" | |
| LENGTH = "length" | |
| ID = "id" | |
| NAME = "name" | |
| PRETTY_NAME = "pretty_name" | |
| NB_EPOCHS = "nb_epochs" | |
| ARCHITECTURE = "architecture" | |
| MODEL = "model" | |
| NAME = "name" | |
| N_PARAMS = "n_params" | |
| OPTIMIZER = "optimizer" | |
| LR = "lr" | |
| PARAMS = "parameters" | |
| DEVICE = "cuda" if torch.cuda.is_available() else "cpu" | |
| SCHEDULER_CONFIGURATION = "scheduler_configuration" | |
| SCHEDULER = "scheduler" | |
| REDUCELRONPLATEAU = "ReduceLROnPlateau" | |
| ARCHITECTURE = "architecture" | |
| CONFIG_DEAD_LEAVES = "config_dead_leaves" | |
| CONFIG_DEGRADATION = "config_degradation" | |
| REDUCTION_SUM = "reduction_sum" | |
| REDUCTION_AVERAGE = "reduction_average" | |
| REDUCTION_SKIP = "reduction_skip" | |
| TRACES_TARGET = "target" | |
| TRACES_DEGRADED = "degraded" | |
| TRACES_RESTORED = "restored" | |
| TRACES_METRICS = "metrics" | |
| TRACES_ALL = "all" | |
| DEGRADATION_BLUR_NONE = "none" | |
| DEGRADATION_BLUR_MAT = "mat" | |
| DEGRADATION_BLUR_GAUSS = "gauss" | |
| SAMPLER_SATURATED = "saturated" | |
| SAMPLER_UNIFORM = "uniform" | |
| SAMPLER_NATURAL = "natural" | |
| SAMPLER_DIV2K = "div2k" | |
| DATASET_FOLDER = "__dataset" | |
| DATASET_PATH = Path(__file__).parent.parent.parent/DATASET_FOLDER | |
| DATASET_DL_RANDOMRGB_1024 = "deadleaves_randomrgb_1024" | |
| DATASET_DL_DIV2K_1024 = "deadleaves_div2k_1024" | |
| DATASET_DL_DIV2K_512 = "deadleaves_div2k_512" | |
| DATASET_DL_EXTRAPRIMITIVES_DIV2K_512 = "deadleaves_primitives_div2k_512" | |
| DATASET_SYNTH_LIST = [DATASET_DL_DIV2K_512, DATASET_DL_DIV2K_1024, | |
| DATASET_DL_RANDOMRGB_1024, DATASET_DL_EXTRAPRIMITIVES_DIV2K_512] | |
| DATASET_BLUR_KERNEL_PATH = DATASET_PATH / "kernels" / "custom_blur_centered_selection.mat" | |
| AUGMENTATION_FLIP = "flip" | |
| AUGMENTATION_ROTATE = "rotate" | |
| DATASET_DIV2K = "div2k" | |