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)