def _forward_reinput(self, color, image_rgb, image_real, test):
outputs = []
for i_reinput in range(len(self.args.loss_blend_ratio_reinput)):
model_reinput = self.model_reinput_list[i_reinput]
image_input = model_reinput.xp.copy(color.data)
image_input_residual = model_reinput.xp.copy(image_input)
# convert gray color range '''output to input'''
image_input[:, 0, :, :] = comicolorization.utility.color.normalize(
image_input[:, 0, :, :],
in_min=self.range_output_luminance[0], in_max=self.range_output_luminance[1],
out_min=self.range_input_luminance[0], out_max=self.range_input_luminance[1],
)
color, other, disc_real, disc_gen = \
self._forward_model(model_reinput, image_input, image_rgb, image_real, test=test)
if self.args.use_residual_reinput:
color += chainer.Variable(image_input_residual)
outputs.append([color, other, disc_real, disc_gen])
return outputs
评论列表
文章目录