python 信息可视化 利用pyvista库可视化点云

ShapeNet分割数据可视化对比

import os

import glob

import random

import pyvista as pv

result_paths = glob.glob(r'./examples/shapenet/results/predict_err_ply/*/*')

print(len(result_paths))

case_id = random.randint(0, len(result_paths) // 3)

point_size = 3

opacity = 0.8

gts = [p for p in result_paths if 'gt' in p]

print(len(gts))

gt_path = os.path.join(gts[case_id])

print('Test case:', gt_path)

pred_path = gt_path.replace('gt', 'pred')

diff_path = gt_path.replace('gt', 'diff')

gt_mesh = pv.read(gt_path)

pred_mesh = pv.read(pred_path)

diff_mesh = pv.read(diff_path)

pl = pv.Plotter(shape=(1, 3))

pl.set_background([0.9, 0.9, 0.9])

pl.subplot(0, 0)

pl.add_title('Gt', font_size=20, font='times')

pl.show_axes()

pl.add_mesh(gt_mesh, point_size=point_size, opacity=opacity)

pl.subplot(0, 1)

pl.add_title('Pred', font_size=20, font='times')

pl.show_axes()

pl.add_mesh(pred_mesh, point_size=point_size, opacity=opacity)

pl.subplot(0, 2)

pl.add_title('Diff', font_size=20, font='times')

pl.show_axes()

pl.add_mesh(diff_mesh, point_size=point_size, opacity=opacity)

pl.show()

参考阅读