Blind_Image_Restoration / data /data_config.py
SunderAli17's picture
Create data/data_config.py
7707ac5 verified
raw
history blame
329 Bytes
from dataclasses import dataclass, field
from typing import Optional, List
@dataclass
class SingleDataConfig:
dataset_folder: str
imagefolder: bool = True
dataset_weight: float = 1.0 # Not used yet
@dataclass
class DataConfig:
datasets: List[SingleDataConfig]
val_dataset: Optional[SingleDataConfig] = None