diff --git a/scripts/test_shenron.py b/scripts/test_shenron.py index a7490ef..b6abf3b 100644 --- a/scripts/test_shenron.py +++ b/scripts/test_shenron.py @@ -229,6 +229,10 @@ def run_testbench(iter_name): np.save(met_base / "angle_axis.npy", models[r_type].processor.angleAxis) lidar_files = sorted(list(lidar_dir.glob("*.npy"))) + if args.frames and args.frames > 0: + print(f" [INFO] Limiting to first {args.frames} frames as requested.") + lidar_files = lidar_files[:args.frames] + for lidar_file in tqdm.tqdm(lidar_files, desc=" Simulating Radars", unit="frame"): data = np.load(lidar_file) # Pad to [x, y, z, intensity, cos_inc_angle, obj, tag] if needed @@ -478,6 +482,7 @@ def run_testbench(iter_name): if __name__ == "__main__": parser = argparse.ArgumentParser(description="Shenron Physics Iteration Testbench") parser.add_argument("--iter", required=True, help="Name of the current debug iteration (e.g., 01_baseline)") + parser.add_argument("--frames", type=int, default=0, help="Number of frames to process (0 for all)") args = parser.parse_args() run_testbench(args.iter)