可以使用OpenCV库读取双目图像,并使用SGBM算法或BM算法计算视差图。然后,可以使用OpenCV的reprojectImageTo3D函数将视差图映射到三维空间中,生成点云。
以下是代码示例:
import cv2
import numpy as np
import torch
# 读入双目图像
imgL = cv2.imread('left.jpg', 0)
imgR = cv2.imread('right.jpg', 0)
# 计算视差图
stereo = cv2.createStereoBM(numDisparities=16, blockSize=15)
disp