(
results_path = "results",
sample_folder ="20251017T071730_400000_1024_text_False",
dataset = 'nuscenes', # nuscenes kitti360
metrics = ['mmd', 'jsd', 'fsvd', 'fpvd'], # specify metrics to evaluate, ['mmd', 'jsd', 'frid', 'fsvd', 'fpvd']
mmd_kitti_path = "",
use_reference = False
)
| 289 | return pointss |
| 290 | |
| 291 | def eval_results( |
| 292 | results_path = "results", |
| 293 | sample_folder ="20251017T071730_400000_1024_text_False", |
| 294 | dataset = 'nuscenes', # nuscenes kitti360 |
| 295 | metrics = ['mmd', 'jsd', 'fsvd', 'fpvd'], # specify metrics to evaluate, ['mmd', 'jsd', 'frid', 'fsvd', 'fpvd'] |
| 296 | mmd_kitti_path = "", |
| 297 | use_reference = False |
| 298 | ): |
| 299 | |
| 300 | reference = [] |
| 301 | mmd_kitti = []# GT的点云 |
| 302 | if use_reference: |
| 303 | if(dataset == "nuscenes"): |
| 304 | reference = get_nuScenes_points() # GT的点云 |
| 305 | elif(dataset == "kitti360"): |
| 306 | reference = get_kitti360_points() |
| 307 | |
| 308 | if(metrics.__contains__("mmd")): |
| 309 | if(mmd_kitti_path != "" and os.path.exists(mmd_kitti_path)): |
| 310 | index = common.read_pkl(mmd_kitti_path) |
| 311 | else: |
| 312 | index = random.sample(range(len(reference)), 2000) |
| 313 | common.save_pkl(save_path=mmd_kitti_path, infos=index) |
| 314 | |
| 315 | mmd_kitti = [reference[i] for i in index]) |
| 316 | |
| 317 | mmd_kitti = [reference[i] for i in index] |
| 318 | samples = get_generated_points(sample_folder) # 生成的点云 |
| 319 | |
| 320 | results = evaluate( |
| 321 | reference=reference, |
| 322 | mmd_kitti=mmd_kitti, |
| 323 | samples=samples, |
| 324 | metrics=metrics, |
| 325 | dataset=dataset, |
| 326 | results_path=results_path |
| 327 | ) |
| 328 | |
| 329 | return results |
| 330 | |
| 331 | def single_result( |
| 332 | results_path = "./results", |
no test coverage detected