|
|
@ -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) |