哪些深度相机有python接口_python 从深度相机realsense生成pcl点云
簡單說下步驟:
一、通過realsense取得深度信息和彩色信息
二、獲取坐標和色彩信息
三、通過pcl可視化點云
一、通過realsense取得深度信息和彩色信息
ubuntu下intel realsense的軟件可以打開realsen的界面,里面可以得到彩色圖像和深度圖像,我們通過realsense的python接口獲取彩色信息和深度信息。
1.基礎的獲取彩色和深度信息,realsense中的視頻流轉換為python的numpy格式,通過opencv輸出
import pyrealsense2 as rs
import numpy as np
import cv2
import pcl
if __name__ == "__main__":
# Configure depth and color streams
pipeline = rs.pipeline()
config = rs.config()
config.enable_stream(rs.stream.depth, 640, 480, rs.format.z16, 30)
config.enable_stream(rs.stream.color, 640, 480, rs.format.bgr8, 30)
# Start streaming
pipeline.start(config)
#深度圖像向彩色對齊
align_to_color=rs.align(rs.stream.color)
try:
while True:
# Wait for a coherent pair of frames: depth and color
frames = pipeline.wait_for_frames()
frames = align_to_color.process(frames)
depth_frame = frames.get_depth_frame()
color_frame = frames.get_color_frame()
if not depth_frame or not color_frame:
continue
# Convert images to numpy arrays
depth_image = np.asanyarray(depth_frame.get_data())
color_image = np.asanyarray(color_frame.get_data())
# Apply colormap on depth image (image must be converted to 8-bit per pixel first)
depth_colormap = cv2.applyColorMap(cv2.convertScaleAbs(depth_image, alpha=0.03), cv2.COLORMAP_JET)
# Stack both images horizontally
images = np.hstack((color_image, depth_colormap))
# Show images
cv2.namedWindow('RealSense', cv2.WINDOW_AUTOSIZE)
cv2.imshow('RealSense', images)
key = cv2.waitKey(1)
# Press esc or 'q' to close the image window
if key & 0xFF == ord('q') or key == 27:
cv2.destroyAllWindows()
break
finally:
# Stop streaming
pipeline.stop()
2.獲取內參和保存圖片
分別用opencv和scipy.misc保存圖片,略微會有一些差異,同時我們也獲取了相機參數。
import pyrealsense2 as rs
import numpy as np
import cv2
import scipy.misc
import pcl
def get_image():
# Configure depth and color streams
pipeline = rs.pipeline()
config = rs.config()
config.enable_stream(rs.stream.depth, 640, 480, rs.format.z16, 30)
config.enable_stream(rs.stream.color, 640, 480, rs.format.bgr8, 30)
# Start streaming
pipeline.start(config)
#獲取圖像,realsense剛啟動的時候圖像會有一些失真,我們保存第100幀圖片。
for i in range(100):
data = pipeline.wait_for_frames()
depth = data.get_depth_frame()
color = data.get_color_frame()
#獲取內參
dprofile = depth.get_profile()
cprofile = color.get_profile()
cvsprofile = rs.video_stream_profile(cprofile)
dvsprofile = rs.video_stream_profile(dprofile)
color_intrin=cvsprofile.get_intrinsics()
print(color_intrin)
depth_intrin=dvsprofile.get_intrinsics()
print(color_intrin)
extrin = dprofile.get_extrinsics_to(cprofile)
print(extrin)
depth_image = np.asanyarray(depth.get_data())
color_image = np.asanyarray(color.get_data())
# Apply colormap on depth image (image must be converted to 8-bit per pixel first)
depth_colormap = cv2.applyColorMap(cv2.convertScaleAbs(depth_image, alpha=0.03), cv2.COLORMAP_JET)
cv2.imwrite('color.png', color_image)
cv2.imwrite('depth.png', depth_image)
cv2.imwrite('depth_colorMAP.png', depth_colormap)
scipy.misc.imsave('outfile1.png', depth_image)
scipy.misc.imsave('outfile2.png', color_image)
二、獲取坐標和色彩信息
1. 通過realsense攝像頭,獲取了頂點坐標和色彩信息。具體并不是很了解,pc.mac_to() 和 points=pc.calculate()是把色彩和深度結合了??再通過points獲取頂點坐標。我們將頂點坐標和彩色相機得到的像素存入txt文件,。
def my_depth_to_cloud():
pc = rs.pointcloud()
points = rs.points()
pipeline = rs.pipeline()
config = rs.config()
config.enable_stream(rs.stream.depth, 640, 480, rs.format.z16, 30)
config.enable_stream(rs.stream.color, 640, 480, rs.format.bgr8, 30)
pipe_profile = pipeline.start(config)
for i in range(100):
data = pipeline.wait_for_frames()
depth = data.get_depth_frame()
color = data.get_color_frame()
frames = pipeline.wait_for_frames()
depth = frames.get_depth_frame()
color = frames.get_color_frame()
colorful = np.asanyarray(color.get_data())
colorful=colorful.reshape(-1,3)
pc.map_to(color)
points = pc.calculate(depth)
#獲取頂點坐標
vtx = np.asanyarray(points.get_vertices())
#獲取紋理坐標
#tex = np.asanyarray(points.get_texture_coordinates())
with open('could.txt','w') as f:
for i in range(len(vtx)):
f.write(str(np.float(vtx[i][0])*1000)+' '+str(np.float(vtx[i][1])*1000)+' '+str(np.float(vtx[i][2])*1000)+' '+str(np.float(colorful[i][0]))+' '+str(np.float(colorful[i][1]))+' '+str(np.float(colorful[i][2]))+'\n')
三、通過pcl可視化點云
https://github.com/strawlab/python-pcl/blob/master/examples/example.py
1.在pcl中,要顯示三維加色彩的點云坐標,每個點云包含了 x,y,z,rgb四個參數,特別的,rbg這個參數是由三維彩色坐標轉換過來的。剛才得到的could.txt中,x,y,z,r,g,b 轉換為x,y,z,rgb,只改顏色數據np.int(data[i][3])<<16|np.int(data[i][4])<<8|np.int(data[i][5])。保存為cloud_rbg.txt。
with open('could.txt','r') as f:
lines = f.readlines()
num=0
for line in lines:
l=line.strip().split()
data.append([np.float(l[0]),np.float(l[1]),np.float(l[2]),np.float(l[3]),np.float(l[4]),np.float(l[5])])
#data.append([np.float(l[0]), np.float(l[1]), np.float(l[2])])
num = num+1
with open('cloud_rgb.txt', 'w') as f:
for i in range(len(data)):
f.write(str(np.float(data[i][0])) + ' ' + str(np.float(data[i][1])) + ' ' + str(np.float(data[i][2]))+ ' ' + str(np.int(data[i][3])<<16|np.int(data[i][4])<<8|np.int(data[i][5]))+'\n')
2. 顯示
def visual():
cloud = pcl.PointCloud_PointXYZRGB()
points = np.zeros((307200,4),dtype=np.float32)
n=0
with open('cloud_rgb.txt','r') as f:
lines = f.readlines()
for line in lines:
l=line.strip().split()
#保存xyz時候擴大了1000倍,發現并沒有用
points[n][0] = np.float(l[0])/1000
points[n][1] = np.float(l[1])/1000
points[n][2] = np.float(l[2])/1000
points[n][3] = np.int(l[3])
n=n+1
print(points[0:100]) #看看數據是怎么樣的
cloud.from_array(points) #從array構建點云的方式
visual = pcl.pcl_visualization.CloudViewing()
visual.ShowColorCloud(cloud)
v = True
while v:
v = not (visual.WasStopped())
3.想生成pcd,再讀取pcd,但是下面的程序在可視化的時候報錯
def txt2pcd():
import time
filename='cloud_rgb.txt'
print("the input file name is:%r." % filename)
start = time.time()
print("open the file...")
file = open(filename, "r+")
count = 0
# 統計源文件的點數
for line in file:
count = count + 1
print("size is %d" % count)
file.close()
# output = open("out.pcd","w+")
f_prefix = filename.split('.')[0]
output_filename = '{prefix}.pcd'.format(prefix=f_prefix)
output = open(output_filename, "w+")
list = ['# .PCD v0.7 - Point Cloud Data file format\n', 'VERSION 0.7\n', 'FIELDS x y z rgb\n', 'SIZE 4 4 4 4\n',
'TYPE F F F U\n', 'COUNT 1 1 1 1\n']
output.writelines(list)
output.write('WIDTH ') # 注意后邊有空格
output.write(str(count))
output.write('\nHEIGHT ')
output.write(str(1)) # 強制類型轉換,文件的輸入只能是str格式
output.write('\nPOINTS ')
output.write(str(count))
output.write('\nDATA ascii\n')
file1 = open(filename, "r")
all = file1.read()
output.write(all)
output.close()
file1.close()
end = time.time()
print("run time is: ", end - start)
import pcl.pcl_visualization
cloud = pcl.load_XYZRGB('cloud_rgb.pcd')
visual = pcl.pcl_visualization.CloudViewing()
visual.ShowColorCloud(cloud, 'cloud')
flag = True
while flag:
flag != visual.WasStopped()
TypeError: expected bytes, str found
原圖,深度圖,云圖(云圖一片黑,鼠標滾輪翻一下) 如下:
opencv保存的顏色圖:
scipy保存的顏色圖
深度圖
深度圖可視化(這個是每有對齊的深度圖align):
云圖(深度和色彩沒有對齊的圖):
總結
以上是生活随笔為你收集整理的哪些深度相机有python接口_python 从深度相机realsense生成pcl点云的全部內容,希望文章能夠幫你解決所遇到的問題。
- 上一篇: pls-00302: 必须声明 组件
- 下一篇: 弗洛伊德算法_弗洛伊德算法