def main(): # reset tensorflow graph tf.reset_default_graph() options = ModelOptions().parse() if options.filename == '*': for filename in os.listdir(options.input_dir): bw2color(options, inputname = options.filename, inputpath = options.input_dir, outputpath = options.output_dir) else: bw2color(options, inputname = options.filename, inputpath = options.input_dir, outputpath = options.output_dir) # cleanup cv2.destroyAllWindows() return 0
model.build() sess.run(tf.global_variables_initializer()) # load model only after global variables initialization model.load() if options.mode == 0: args = vars(options) print('\n------------ Options -------------') with open(os.path.join(options.checkpoints_path, 'options.dat'), 'w') as f: for k, v in sorted(args.items()): print('%s: %s' % (str(k), str(v))) f.write('%s: %s\n' % (str(k), str(v))) print('-------------- End ----------------\n') model.train() elif options.mode == 1: model.evaluate() while True: model.sample() else: model.turing_test() if __name__ == "__main__": main(ModelOptions().parse())
from options import ModelOptions from main import main options = ModelOptions().parse() options.mode = 2 main(options)