def __init__(self, inputs, targets, is_training, args, state="TRAIN"):
super(HDDRNet, self).__init__()
self.channels = args.channels
self.gamma_S = args.gamma_S
self.gamma_A = args.gamma_A
self.verbose = args.verbose
self.net_variables = None
self.num_GRL_HRB = args.num_GRL_HRB
self.num_SRe_HRB = args.num_SRe_HRB
self.GRLfeats = []
self.SRefeats = []
self.vgg = VGG19(None, None, None)
self.PreRecons, self.Recons = self.build_model(inputs,
is_training,
reuse=False,
verbose=self.verbose)
if state == "TRAIN":
self.use_perceptual_loss = args.perceptual_loss
self.loss = self.compute_loss(targets, self.PreRecons, self.Recons,
self.use_perceptual_loss)
elif state == "TEST":
self.use_perceptual_loss = None
self.loss = 0.0
else:
assert False, "State not supportable (only support 'TRAIN' and 'TEST')."
def __init__(self, inputs, targets, is_training, args):
super(HDDRNet, self).__init__()
self.channels = args.channels
self.gamma_S = args.gamma_S
self.gamma_A = args.gamma_A
self.use_perceptual_loss = args.perceptual_loss
self.verbose = args.verbose
self.net_variables = None
self.vgg = VGG19(None, None, None)
self.PreRecons, self.Recons = self.build_model(inputs,
is_training,
reuse=False,
verbose=self.verbose)
self.loss = self.compute_loss(targets, self.PreRecons, self.Recons,
self.use_perceptual_loss)