当前位置：首页 > news >正文

OpenCV学习（二）-二维、三维识别

news 2025/7/22 5:35:02

OpenCV 是一个功能强大的计算机视觉库，可以用于识别和处理二维图像和三维图像。以下是关于二维图像和三维图像识别的基础知识和示例代码。

1. 二维图像识别

二维图像识别通常包括图像分类、对象检测、特征提取等任务。以下是一些常见的操作：

1.1 图像分类

使用预训练模型对图像进行分类，例如使用深度学习模型（如 ResNet、MobileNet 等）。

import cv2# 加载预训练的深度学习模型
net = cv2.dnn.readNetFromCaffe("deploy.prototxt", "resnet.caffemodel")# 读取图像并预处理
image = cv2.imread("image.jpg")
blob = cv2.dnn.blobFromImage(image, scalefactor=1.0, size=(224, 224), mean=(104, 117, 123))# 输入模型并进行推理
net.setInput(blob)
out = net.forward()# 输出结果
print("预测结果：", out)

1.2 对象检测

使用目标检测模型（如 YOLO、SSD）来检测图像中的对象。

import cv2# 加载模型
net = cv2.dnn.readNetFromDarknet("yolov3.cfg", "yolov3.weights")# 读取图像
image = cv2.imread("image.jpg")
height, width = image.shape[:2]# 预处理图像
blob = cv2.dnn.blobFromImage(image, 1/255.0, (416, 416), swapRB=True, crop=False)
net.setInput(blob)# 推理
layer_names = net.getLayerNames()
output_layers = [layer_names[i - 1] for i in net.getUnconnectedOutLayers()]
outputs = net.forward(output_layers)# 解析结果并绘制检测框
for output in outputs:for detection in output:scores = detection[5:]class_id = np.argmax(scores)confidence = scores[class_id]if confidence > 0.5:center_x = int(detection[0] * width)center_y = int(detection[1] * height)w = int(detection[2] * width)h = int(detection[3] * height)x = int(center_x - w / 2)y = int(center_y - h / 2)cv2.rectangle(image, (x, y), (x + w, y + h), (0, 255, 0), 2)# 显示结果
cv2.imshow("Detection", image)
cv2.waitKey(0)

1.3 特征提取

使用特征提取算法（如 SIFT、ORB）提取图像中的关键点。

import cv2# 读取图像
image = cv2.imread("image.jpg")
gray = cv2.cvtColor(image, cv2.COLOR_BGR2GRAY)# 创建 SIFT 检测器
sift = cv2.SIFT_create()
keypoints, descriptors = sift.detectAndCompute(gray, None)# 绘制关键点
image_with_keypoints = cv2.drawKeypoints(image, keypoints, None)
cv2.imshow("Keypoints", image_with_keypoints)
cv2.waitKey(0)

2. 三维图像识别

OpenCV 本身对三维图像的支持有限，通常需要结合其他库（如 PCL、Open3D）进行三维图像处理。以下是一些基本概念和示例：

2.1 点云处理

点云是三维图像的一种常见表示形式，通常包含三维坐标、颜色等信息。

import open3d as o3d# 读取点云文件
pcd = o3d.io.read_point_cloud("point_cloud.ply")# 可视化点云
o3d.visualization.draw_geometries([pcd])

2.2 三维重建

三维重建是通过多视角图像生成三维模型，通常需要使用深度相机（如 Kinect）或立体相机。

import cv2
import numpy as np# 模拟深度图像
depth_image = np.random.rand(480, 640).astype(np.float32)# 转换为点云
fx, fy = 525.0, 525.0  # 相机内参
cx, cy = 319.5, 239.5points = []
for y in range(depth_image.shape[0]):for x in range(depth_image.shape[1]):z = depth_image[y, x]if z == 0:continuepoint = [(x - cx) * z / fx, (y - cy) * z / fy, z]points.append(point)# 使用 Open3D 可视化
pcd = o3d.geometry.PointCloud()
pcd.points = o3d.utility.Vector3dVector(points)
o3d.visualization.draw_geometries([pcd])

2.3 三维特征提取

三维特征提取通常包括法线估计、关键点检测等。

import open3d as o3d# 读取点云
pcd = o3d.io.read_point_cloud("point_cloud.ply")# 估计法线
pcd.estimate_normals(search_param=o3d.geometry.KDTreeSearchParamHybrid(radius=0.1, max_nn=30))# 可视化法线
o3d.visualization.draw_geometries([pcd], point_show_normal=True)