futurefantasy commited on
Commit
c52a7a7
·
verified ·
1 Parent(s): a9ae4f8

Print only prediction when output_jsonl is omitted

Browse files
Files changed (1) hide show
  1. quick_start/run_example.py +7 -6
quick_start/run_example.py CHANGED
@@ -345,12 +345,13 @@ def main() -> int:
345
  if output_path is not None:
346
  output_path.write_text(json.dumps(output_row, ensure_ascii=False) + "\n", encoding="utf-8")
347
 
348
- print(f"video_path={video_path}")
349
- print(f"output_jsonl={output_path if output_path is not None else '(not saved)'}")
350
- print(f"input_sample_hz={float(args.sample_hz):.4f}")
351
- print(f"input_frame_count={len(sampled_indices)}")
352
- print(f"pred_keypoint_count={len(pred_point_progress)}")
353
- print()
 
354
  print(response)
355
  return 0
356
 
 
345
  if output_path is not None:
346
  output_path.write_text(json.dumps(output_row, ensure_ascii=False) + "\n", encoding="utf-8")
347
 
348
+ if output_path is not None:
349
+ print(f"video_path={video_path}")
350
+ print(f"output_jsonl={output_path}")
351
+ print(f"input_sample_hz={float(args.sample_hz):.4f}")
352
+ print(f"input_frame_count={len(sampled_indices)}")
353
+ print(f"pred_keypoint_count={len(pred_point_progress)}")
354
+ print()
355
  print(response)
356
  return 0
357