import tensorflow as tf

  print(args)

  return args

if __name__ == "__main__":

  args = parse_args()
  if args.gpu==1:
    os.environ['CUDA_VISIBLE_DEVICES']="0"
  else:
    os.environ['CUDA_VISIBLE_DEVICES']=""
  config = tf.ConfigProto()
  config.gpu_options.per_process_gpu_memory_fraction = 1.0
  config.gpu_options.allow_growth = True
  config.log_device_placement=True
  sess = tf.Session(config=config)

  if args.command == "compress":
    rootdir, filename = os.path.split(args.input)
    if not args.output:
      args.output = filename.split('.')[0]
      print(args.output)
