OpenCV(影像NaN處理)

做梦当财神發表於2024-09-02

目錄
  • 1. 目的
  • 2. 示例



1. 目的

src 矩陣中的某些區域(由 mask 指定的非零區域)設定為 NaN

SetToNan 函式實現了這一功能。



2. 示例

#include<iostream>
#include<opencv2/opencv.hpp>

using namespace std;
using namespace cv;

void SetToNan(cv::Mat& src, const cv::Mat& mask);

int main(void)
{
	Mat A = Mat::ones(500, 500, CV_32FC1);
	Mat mask = Mat::zeros(500, 500, CV_8UC1);
	circle(mask, Point2i(250, 250), 100, 255, -1);
	Mat A2 = A.clone();
	SetToNan(A2, mask);

	system("pause");
	return 0;
}

void SetToNan(cv::Mat& src, const cv::Mat& mask)
{
	CV_Assert(src.type() == CV_32FC1);  // 32FC是float型
	cv::Mat _nan(src.size(), src.type(), nan(""));
	_nan.setTo(0, mask);  // 將掩碼 mask 中非零(即值為255)的區域對應的 _nan 矩陣中的元素設定為0。
	// nan加任何值都為nan
	src = src + _nan;
}

A2 矩陣中,非掩碼區域將保持為1(原始值),而在掩碼指定的圓形區域內,所有值將變為 NaN

  • 使用 cv::Mat::setTo() 函式和掩碼機制,將部分矩陣元素替換為特定的數值(0或NaN),然後透過矩陣相加的方式將原矩陣的部分元素設為NaN

來自:https://zhaitianbao.blog.csdn.net/article/details/116145717



相關文章