self.reader = ImageReader(["image", "conditioning"])
else: // in the inference process use image input only
self.reader = ImageReader(["conditioning"])
if self.reader:
self.reader.initialise_reader(data_param, task_param)
if self.net_param.normalise_foreground_only:
foreground_masking_layer = BinaryMaskingLayer(
type_str=self.net_param.foreground_type,
multimod_fusion=self.net_param.multimod_foreground_type,
After Change
self.readers = [ImageReader(["image", "conditioning"], phase="train"),
ImageReader(["conditioning"], phase="validation")]
else: // in the inference process use image input only
self.readers = [ImageReader(["image"], phase="test")]
for reader in self.readers:
reader.initialise_reader(data_param, task_param)
if self.net_param.normalise_foreground_only:
foreground_masking_layer = BinaryMaskingLayer(
type_str=self.net_param.foreground_type,
multimod_fusion=self.net_param.multimod_foreground_type,
