def __init__(self, args): self.preInit() self.args = args self.sess = tf.Session() self.load_network() args.image_content = kld.prepare_image_dict(args.image_content) args.image_test = kld.prepare_image_dict(args.image_test) self.ycL = kld.get_dir_files(args.content_dir) args.num_iters = len(self.ycL) // args.batch_size self.build(args) self.test(args)
def __init__(self, args): self.preInit() self.args = args self.sess = tf.Session() self.load_network() args.image_style = kld.prepare_image_dict(args.image_style) args.image_content = kld.prepare_image_dict(args.image_content) self.ysL = self.load_image(args.style_dir, args.image_style) self.ycL = kld.get_dir_files(args.content_dir) args.num_iters = len(self.ycL) // args.batch_size self.style_layers = kld.ordered_sorted_dict(args.style_layer_ids) self.content_layers = kld.ordered_sorted_dict(args.content_layer_ids) self.build(args) self.train(args)
def __init__(self, args): self.preInit() self.args = args self.sess = tf.Session() self.load_network() args.image_test = kld.prepare_image_dict(args.image_test) self.build(args) self.test(args)