Spaces:

fffiloni
/

InstantIR

Running on Zero

InstantIR / data /data_config.py

Upload folder using huggingface_hub

0305a63 verified almost 2 years ago

330 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

	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