def __init__(self,
data_path: Path,
sketch_path: Path,
line_method: LineArt,
extension=".jpg",
train_size=224,
valid_size=256,
color_space="rgb",
line_space="rgb",
src_perturbation=0.5,
tgt_perturbation=0.2):
self.data_path = data_path
self.pathlist = list(self.data_path.glob(f"**/*{extension}"))
self.train_list, self.val_list = self._train_val_split(self.pathlist)
self.train_len = len(self.train_list)
self.train_size = train_size
self.valid_size = valid_size
self.line_process = LineProcessor(sketch_path, line_method)
self.color_space = color_space
self.line_space = line_space
self.sketch_path = sketch_path
self.src_per = src_perturbation
self.tgt_per = tgt_perturbation
self.thre = 50
self.src_const = np.array([[-0.5, -0.5], [0.5, -0.5], [-0.5, 0.5],
[0.5, 0.5], [0.2, -0.2], [-0.2, 0.2],
[0.2, 0.2], [-0.2, -0.2]])
def __init__(self,
data_path: Path,
sketch_path: Path,
flat_path: Path,
line_method: Literal,
extension=".png",
train_size=224,
valid_size=256,
color_space="rgb",
line_space="rgb"):
self.data_path = data_path
self.flat_path = flat_path
self.pathlist = list(self.data_path.glob(f"**/*{extension}"))
self.train_list, self.val_list = self._train_val_split(self.pathlist)
self.train_len = len(self.train_list)
self.sketch_path = sketch_path
self.train_size = train_size
self.valid_size = valid_size
self.line_process = LineProcessor(sketch_path, line_method)
self.color_space = color_space
self.line_space = line_space
def __init__(self,
data_path: Path,
sketch_path=None,
line_method=["xdog"],
extension=".jpg",
train_size=256,
valid_size=256,
color_space="rgb",
line_space="rgb"):
self.data_path = data_path
self.pathlist = list(data_path.glob(f"**/*{extension}"))
self.train_list, self.val_list = self._train_val_split(self.pathlist)
self.train_len = len(self.train_list)
self.train_size = train_size
self.valid_size = valid_size
self.line_process = LineProcessor(sketch_path, line_method)
self.color_space = color_space
self.line_space = line_space