add new_added_pts compute

This commit is contained in:
2024-10-01 14:51:45 +08:00
parent 16d1f8ab67
commit 7b28cf9e91
3 changed files with 19 additions and 0 deletions

5
app.py
View File

@@ -177,6 +177,9 @@ def get_frame_data():
frame_data['new_point_cloud'] = sampled_point_cloud.tolist()
frame_data['combined_point_cloud'] = combined_point_cloud.tolist()
new_added_pts = ReconstructionUtil.get_new_added_points(combined_point_cloud, sampled_point_cloud)
frame_data["new_added_pts"] = new_added_pts.tolist()
combined_point_cloud = np.concatenate([combined_point_cloud, sampled_point_cloud], axis=0)
combined_point_cloud = PtsUtil.voxel_downsample_point_cloud(combined_point_cloud, 0.01)
@@ -218,6 +221,8 @@ def parse_to_frame_data(inference_result):
coverage_rate = inference_result["coverage_rate_seq"][idx]
frame_data['new_point_cloud'] = target_pts.tolist()
frame_data['combined_point_cloud'] = combined_point_cloud.tolist()
new_added_pts = ReconstructionUtil.get_new_added_points(combined_point_cloud, target_pts)
frame_data["new_added_pts"] = new_added_pts.tolist()
combined_point_cloud = np.concatenate([combined_point_cloud, target_pts], axis=0)
frame_data["coverage_rate"] = float(coverage_rate)
delta_CR = frame_data["coverage_rate"] - last_CR