Browse Source

feat(tooling): add --frames support to test_shenron.py

main
RUSHIL AMBARISH KADU 1 month ago
parent
commit
daffc69bca
  1. 5
      scripts/test_shenron.py

5
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) np.save(met_base / "angle_axis.npy", models[r_type].processor.angleAxis)
lidar_files = sorted(list(lidar_dir.glob("*.npy"))) 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"): for lidar_file in tqdm.tqdm(lidar_files, desc=" Simulating Radars", unit="frame"):
data = np.load(lidar_file) data = np.load(lidar_file)
# Pad to [x, y, z, intensity, cos_inc_angle, obj, tag] if needed # 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__": if __name__ == "__main__":
parser = argparse.ArgumentParser(description="Shenron Physics Iteration Testbench") 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("--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() args = parser.parse_args()
run_testbench(args.iter) run_testbench(args.iter)
Loading…
Cancel
Save