IntelRealSense d415深度相機拍照制作數據集python腳本

最近想做一個陰影數據集，手上d415相機像素好一點，就用來調用拍照啦。
上源碼：

import pyrealsense2 as rs import numpy as np import cv2# Configure depth and color streams pipeline = rs.pipeline() config = rs.config() config.enable_stream(rs.stream.depth, 1280, 720, rs.format.z16, 30) config.enable_stream(rs.stream.color, 1280, 720, rs.format.bgr8, 30)# Start streaming pipeline.start(config)# 每種樣本50個圖像，一共八種，50*8=400個初始圖像，之后再做數據增廣 filePath = "/home/qyt/DataSet/Shadow/" id = "1"i = 0 j=0 #只是采集圖像，不用深度信息 while i<=50:frames = pipeline.wait_for_frames()color_frame = frames.get_color_frame()# Convert images to numpy arrays 把圖像轉換為numpy datacolor_image = np.asanyarray(color_frame.get_data())cv2.namedWindow('MyD415')cv2.imshow('MyD415', color_image)key = cv2.waitKey(0)if i < 10:imageName = "shadow" + id + "_0" + str(i) + ".jpg"else:imageName = "shadow" + id + "_" + str(i) + ".jpg"imagePath = filePath + imageNameif key & 0xFF == ord('q') or key == 27:cv2.destroyAllWindows()pipeline.stop()breakelif key == ord('s'): # 按下s鍵時保存并退出cv2.imwrite(imagePath, color_image)i += 1#cv2.destroyAllWindows()j=j+1#方便看保存到第幾張圖了print("iter:",j,"i:", i)

總結

以上是生活随笔為你收集整理的IntelRealSense d415深度相机拍照制作数据集python脚本的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。