Exemplo n.º 1
0
def main(_):
    hparams = t2t_decoder.create_hparams()
    problem = hparams.problem
    frame_shape = [
        problem.frame_height, problem.frame_width, problem.num_channels
    ]
    video_metrics.compute_and_save_video_metrics(
        FLAGS.output_dir, FLAGS.problem, hparams.video_num_target_frames,
        frame_shape)
Exemplo n.º 2
0
def main(_):
    hparams = t2t_decoder.create_hparams()
    problem = hparams.problem
    frame_shape = [
        problem.frame_height, problem.frame_width, problem.num_channels
    ]
    decode_hp = t2t_decoder.create_decode_hparams()

    output_dirs = [
        os.path.join(FLAGS.output_dir, "decode_%05d" % decode_id)
        for decode_id in range(decode_hp.num_decodes)
    ]

    video_metrics.compute_and_save_video_metrics(
        output_dirs, FLAGS.problem, hparams.video_num_target_frames,
        frame_shape)
Exemplo n.º 3
0
def main(_):
  hparams = t2t_decoder.create_hparams()
  problem = hparams.problem
  frame_shape = [problem.frame_height,
                 problem.frame_width,
                 problem.num_channels]
  decode_hp = t2t_decoder.create_decode_hparams()

  output_dirs = [
      os.path.join(FLAGS.output_dir, "decode_%05d" % decode_id)
      for decode_id in range(decode_hp.num_decodes)
  ]

  video_metrics.compute_and_save_video_metrics(
      output_dirs,
      FLAGS.problem,
      hparams.video_num_target_frames,
      frame_shape)