Code/preprocess

NPY visualize

Shy_un 2023. 11. 3. 10:46

1. Img

import numpy as np
import matplotlib.pyplot as plt

# 이미지 데이터를 NumPy 배열로 읽어옵니다.
image_data = np.load('D:/K-Radar/dir_processed_img/img_driving_corridor/1/front_00002.npy')  # 파일 경로를 적절하게 지정해야 합니다.

# 이미지를 시각화합니다.
plt.imshow(image_data, cmap='gray')  # cmap을 원하는 색상 맵으로 변경할 수 있습니다.
#plt.title('Image')
plt.axis('off')  # 축을 비활성화하여 이미지에 불필요한 눈금을 표시하지 않습니다.
plt.show()

 

2. PCD

import numpy as np
import open3d as o3d

# NumPy 배열로 .npy 파일을 읽어옵니다.
npy_data = np.load('D:/K-Radar/dir_processed_lpc/lpc_driving_corridor/1/lpc_00001.npy')

# NumPy 배열을 Open3D PointCloud로 변환합니다.
point_cloud = o3d.geometry.PointCloud()
point_cloud.points = o3d.utility.Vector3dVector(npy_data[:,:-1]) #x,y,z,r에서 x,y,z만 사용

# 3D 시각화 창을 엽니다.
o3d.visualization.draw_geometries([point_cloud])