open3d+opencv实现矩形框裁剪点云操作(C++)

👑主页:吾名招财

👓简介:工科学硕,研究方向机器视觉,爱好较广泛...

​💫签名:面朝大海,春暖花开!

open3d+opencv实现矩形框裁剪点云操作(C++)

1,引言

针对彩色图和深度图以及相机内参可以合成点云,而对某一区域的点云数据截取,可以通过二维ROI区域截取深度图及彩色图出来。不过不能整张图像裁剪,其图像大小不能变,把除了截取区域外的给置为0就行了。

2,相关测试数据资源如下

本人上传了一个用于三维重建测试的公开数据集,内含彩色图、深度图、相机内参、相机位姿等相关数据,可用于相关测试
https://download.csdn.net/download/qq_44870829/90236553

3,彩色图、深度图和相机内参


4,C++代码

复制代码
#include <string>
#include <iostream>
#include "Open3D/Open3D.h"
#include <opencv2/opencv.hpp>

using namespace std;


open3d::geometry::Image o3d_cv(const cv::Mat& A) {
	open3d::geometry::Image B;
	int bytes_per_channel = (A.depth() / 2 + 1);//refer to the fuction depth
	B.Prepare(A.cols, A.rows, A.channels(), bytes_per_channel);
	std::memcpy(B.data_.data(), A.data, A.total() * A.channels() * bytes_per_channel);
	return B;
}

int main(int argc, char* argv[]) {

	//--------------------------------------1,相关参数----------------------------------
	//要截取的ROI区域大小
	int x = 285;
	int y = 363;
	int w = 150;
	int h = 120;

    //相机内参设置
	int width = 640;   // 输入图像的宽度
	int height = 480;  // 输入图像的高度
	double fx = 585; // x轴焦距 
	double fy = 585; // y轴焦距
	double cx = 320; // 相机原点的x坐标
	double cy = 240; // 相机原点的y坐标

	// 方式一
	auto intrinsic = open3d::camera::PinholeCameraIntrinsic(width, height, fx, fy, cx, cy); // 使用自定义内参
	 方式二
	open3d::camera::PinholeCameraIntrinsic intrinsic = open3d::camera::PinholeCameraIntrinsic();
	intrinsic.SetIntrinsics(width, height, fx, fy, cx, cy);
	//open3d::camera::PinholeCameraIntrinsic intrinsic = open3d::camera::PinholeCameraIntrinsic(
	//	open3d::camera::PinholeCameraIntrinsicParameters::PrimeSenseDefault); // 使用默认内参

    // -RGBD图像参数设置
	double depth_scale = 1000.0; // 深度值的缩放倍数
	double depth_trunc = 3.0;    // 深度值的截断系数
	bool convert_rgb_to_intensity = false; // 是否将彩色图转为强度图

	//-------------------------------------2,读取图像并裁剪-------------------------------------
	//用opencv读取二维彩色图及深度图像,将某一ROI区域裁剪出来,然后再进行点云生成
	cv::Mat colorimg = cv::imread("frame-000276.color.jpg");
	cv::Mat depthimg = cv::imread("frame-000276.depth.png", -1);  //深度图要读取原图

	//制作ROI区域掩膜
	cv::Mat templateImg_color = cv::Mat::zeros(colorimg.size(), colorimg.type());
	cv::Mat mask = cv::Mat::zeros(colorimg.size(), colorimg.type());
	cv::rectangle(mask, cv::Point(x, y), cv::Point(x + w, y + h), cv::Scalar(255, 255, 255), -1);//矩形的两个顶点,两个顶点都包括在矩形内部
	colorimg.copyTo(templateImg_color, mask);

	cv::Mat templateImg_depth = cv::Mat::zeros(depthimg.size(), depthimg.type());
	cv::Mat mask2 = cv::Mat::zeros(depthimg.size(), CV_8UC1);
	cv::rectangle(mask2, cv::Point(x, y), cv::Point(x + w, y + h), cv::Scalar(255, 255, 255), -1);//矩形的两个顶点,两个顶点都包括在矩形内部
	//cv::bitwise_and();
	depthimg.copyTo(templateImg_depth, mask2);

	//-------------------------------------3,显示未裁剪前的原始点云-------------------------------------
	open3d::geometry::Image color_o = o3d_cv(colorimg);
	open3d::geometry::Image depth_o = o3d_cv(depthimg);
	// 生成RGBD图像
	std::shared_ptr<open3d::geometry::RGBDImage> rgbd_image_o = open3d::geometry::RGBDImage::CreateFromColorAndDepth(
		color_o,                    // 输入的彩色图像
		depth_o,                    // 输入的深度图像
		depth_scale,              // 深度值的缩放倍数
		depth_trunc,              // 深度值大于该值将被截断为0
		convert_rgb_to_intensity);// 设置是否将彩色图像转为强度图
   // RGBD转点云
	auto pcd_o = open3d::geometry::PointCloud::CreateFromRGBDImage(*rgbd_image_o, intrinsic);
	open3d::visualization::DrawGeometries({ pcd_o });

	//-------------------------------------4,显示并保存裁剪后的点云-------------------------------------
	open3d::geometry::Image color = o3d_cv(templateImg_color);
	open3d::geometry::Image depth = o3d_cv(templateImg_depth);

	直接使用open3d读取图像
	//open3d::geometry::Image color, depth;
	//open3d::io::ReadImage("0.png", color); // 读取彩色图像
	//open3d::io::ReadImage("0_depth.png", depth); // 读取深度图像

	//将裁剪后的深度图及彩色图转换成点云并显示保存

	//输出图像基本信息
	open3d::utility::LogInfo("Reading RGBD image : ");
	open3d::utility::LogInfo("     Color : {:d} x {:d} x {:d} ({:d} bits per channel)",
		color.width_, color.height_, color.num_of_channels_,
		color.bytes_per_channel_ * 8);
	open3d::utility::LogInfo("Depth : {:d} x {:d} x {:d} ({:d} bits per channel)",
		depth.width_, depth.height_, depth.num_of_channels_,
		depth.bytes_per_channel_ * 8);

	// 生成RGBD图像
	std::shared_ptr<open3d::geometry::RGBDImage> rgbd_image = open3d::geometry::RGBDImage::CreateFromColorAndDepth(
		color,                    // 输入的彩色图像
		depth,                    // 输入的深度图像
		depth_scale,              // 深度值的缩放倍数
		depth_trunc,              // 深度值大于该值将被截断为0
		convert_rgb_to_intensity);// 设置是否将彩色图像转为强度图
    // RGBD转点云
	auto pcd = open3d::geometry::PointCloud::CreateFromRGBDImage(*rgbd_image, intrinsic);
	open3d::visualization::DrawGeometries({ pcd });

	//5.保存点云文件 保存
	open3d::io::WritePointCloudToPCD("moban.pcd", *pcd, false);
	//open3d::io::WritePointCloudToPCD("search.pcd", *pcd, false);

	return 0;
}

5,最终效果

未裁剪前的点云

裁剪后的点云

相关推荐
阿linlin42 分钟前
OpenCV--图像预处理学习01
opencv·学习·计算机视觉
车载小杜1 小时前
基于指针的线程池
开发语言·c++
云 无 心 以 出 岫3 小时前
贪心算法QwQ
数据结构·c++·算法·贪心算法
换一颗红豆3 小时前
【C++ 多态】—— 礼器九鼎,釉下乾坤,多态中的 “风水寻龙诀“
c++
随便昵称4 小时前
蓝桥杯专项复习——前缀和和差分
c++·算法·前缀和·蓝桥杯
commonbelive4 小时前
团体程序设计天梯赛——L1-100 四项全能
c++
genispan4 小时前
QT/C++ 多线程并发下载实践
开发语言·c++·qt
jndingxin4 小时前
OpenCV 图形API(10)用于执行标量除以矩阵的逐元素操作函数divRC()
人工智能·opencv
小卡皮巴拉5 小时前
【力扣刷题实战】矩阵区域和
开发语言·c++·算法·leetcode·前缀和·矩阵
Pacify_The_North5 小时前
【C++进阶三】vector深度剖析(迭代器失效和深浅拷贝)
开发语言·c++·windows·visualstudio