你的代码思路基本上是正确的,但在实现中可能会存在一些小问题。以下是一个改进和修正后的完整代码示例,确保程序可以正常运行:
#include "../pre_process/ImageQuality.h" // 确保这个路径是正确的并且包含BRISQUE函数的实现
#include <opencv2/opencv.hpp>
#include <iostream>
int main()
{
cv::Mat img1;
float score;
img1 = cv::imread("Data/img_in/5.bmp");
if (img1.empty()) {
std::cerr << "Error: Could not open image." << std::endl;
return -1;
}
int blockSize = 320; // 块大小320x320
cv::Mat resultImage = img1.clone(); // 创建结果图像
cv::Mat greenMask(img1.size(), CV_8UC3, cv::Scalar(0, 0, 0)); // 初始化绿色遮罩
cv::Mat blueMask(img1.size(), CV_8UC3, cv::Scalar(0, 0, 0)); // 初始化蓝色遮罩
for (int row = 0; row < img1.rows / blockSize; ++row) {
for (int col = 0; col < img1.cols / blockSize; ++col) {
// 定义当前小块的矩形区域
cv::Rect roi(col * blockSize, row * blockSize, blockSize, blockSize);
cv::Mat smallBlock = img1(roi);
// 计算 BRISQUE 得分
score = BRISQUE(smallBlock); // 注意,这里需要确保BRISQUE返回的是一个float类型
if (score < 30.0f) { // 假设30是阈值,可以根据实际情况调整
rectangle(greenMask, roi, cv::Scalar(0, 255, 0), cv::FILLED); // 在绿色遮罩上填充块
}
else {
rectangle(blueMask, roi, cv::Scalar(255, 0, 0), cv::FILLED); // 在蓝色遮罩上填充块
}
}
}
// 将原图与遮罩合成,设置透明度,例如:透明度为50%
double alpha = 0.5;
addWeighted(resultImage, alpha, greenMask, alpha, 0.0, resultImage);
addWeighted(resultImage.clone(), alpha * (score >= 30 ? 1 : 0),
blueMask * (score >= 30 ? alpha : cv::Scalar(0)),
alpha * (score >= 30 ? alpha : cv::Scalar(1)),
resultImage);
imwrite("Data/img_out/marked_result.bmp", resultImage);
imshow("Result Image", resultImage);
cv::waitKey(0);
return 0;
}
注意事项:
- 请确保
BRISQUE
函数的实现已经正确加载,并能对输入的小块图像返回模糊得分。 - 如果要处理每个小块的得分,你需要注意如何存储或使用得分进行颜色标记。上述示例假设最后一段合成代码只考虑最后一个小块的得分。如果你希望综合所有小块的信息,需要调整该逻辑。
- 确保文件路径和权限都是正确的,以避免因无法读取图片或写入结果而导致的问题。
希望这可以帮助你完成项目!如有其他问题,请随时询问。
内容由零声教学AI助手提供,问题来源于学员提问