Skip to content

Commit cbc23ed

Browse files
committed
Fix argument alignment in benchmark_videos call
This commit corrects the parameter names and order and removes unsupported arguments from `benchmark_videos` call: - device - precision - draw_keypoint_names - get_sys_info
1 parent 526a162 commit cbc23ed

1 file changed

Lines changed: 4 additions & 8 deletions

File tree

dlclive/benchmark.py

Lines changed: 4 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -847,23 +847,19 @@ def main():
847847

848848
# Call the benchmark_videos function with the parsed arguments
849849
benchmark_videos(
850-
video_path=args.video_path,
851850
model_path=args.model_path,
852851
model_type=args.model_type,
853-
device=args.device,
854-
precision=args.precision,
852+
video_path=args.video_path,
853+
output=args.save_dir,
855854
display=args.display,
856855
pcutoff=args.pcutoff,
857856
display_radius=args.display_radius,
858-
resize=tuple(args.resize) if args.resize else None,
857+
resize=args.resize,
859858
cropping=args.cropping,
860859
dynamic=tuple(args.dynamic),
861860
save_poses=args.save_poses,
862-
save_dir=args.save_dir,
863-
draw_keypoint_names=args.draw_keypoint_names,
864-
cmap=args.cmap,
865-
get_sys_info=args.get_sys_info,
866861
save_video=args.save_video,
862+
cmap=args.cmap,
867863
)
868864

869865

0 commit comments

Comments
 (0)