loss_maker.py 文件源码

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