使用opencv的c++代码
时间: 2024-11-03 15:00:36 浏览: 32
要使用 OpenCV 实现图像分割并计算两个重叠水果的质心和方向,可以按照以下步骤编写 C++ 代码:
1. **读取图像**:加载含有两个重叠水果的图像。
2. **颜色通道选择**:选择合适的颜色通道(例如 HSV 中的 V 通道)。
3. **阈值化分割**:将图像转换为二值图像。
4. **形态学操作**:去除噪声和填充孔洞。
5. **分水岭算法**:使用分水岭算法进行分割。
6. **计算质心和方向**:对每个分割区域计算质心和方向。
以下是具体的代码示例:
```cpp
#include <opencv2/opencv.hpp>
#include <iostream>
int main() {
// 1. 读取图像
cv::Mat img = cv::imread("path_to_your_image.jpg");
if (img.empty()) {
std::cerr << "Image not found or unable to open" << std::endl;
return -1;
}
// 2. 颜色通道选择
cv::Mat hsv;
cv::cvtColor(img, hsv, cv::COLOR_BGR2HSV);
cv::Mat v_channel;
cv::split(hsv, {cv::Mat(), cv::Mat(), v_channel});
// 3. 阈值化分割
cv::Mat binary;
cv::threshold(v_channel, binary, 0, 255, cv::THRESH_BINARY_INV + cv::THRESH_OTSU);
// 4. 形态学操作
cv::Mat kernel = cv::getStructuringElement(cv::MORPH_ELLIPSE, cv::Size(3, 3));
cv::morphologyEx(binary, binary, cv::MORPH_OPEN, kernel, cv::Point(-1, -1), 2);
cv::dilate(binary, binary, kernel, cv::Point(-1, -1), 1);
// 5. 分水岭算法
cv::Mat sure_bg;
cv::dilate(binary, sure_bg, kernel, cv::Point(-1, -1), 3);
cv::distanceTransform(binary, binary, cv::DIST_L2, 3);
cv::normalize(binary, binary, 0, 1.0, cv::NORM_MINMAX);
cv::threshold(binary, binary, 0.7 * binary.at<float>(binary.rows / 2, binary.cols / 2), 255, cv::THRESH_BINARY);
cv::Mat sure_fg;
binary.convertTo(sure_fg, CV_8U);
cv::Mat unknown;
cv::subtract(sure_bg, sure_fg, unknown);
cv::Mat markers = cv::Mat::zeros(sure_fg.size(), CV_32S);
cv::connectedComponents(sure_fg, markers);
markers += 1;
markers(unknown == 255) = 0;
cv::watershed(img, markers);
// 6. 计算质心和方向
std::vector<std::vector<cv::Point>> contours;
std::vector<cv::Vec4i> hierarchy;
cv::findContours((markers == -1).clone(), contours, hierarchy, cv::RETR_CCOMP, cv::CHAIN_APPROX_SIMPLE);
for (size_t i = 0; i < contours.size(); ++i) {
cv:: Moments moments = cv::moments(contours[i]);
double area = moments.m00;
if (area > 100) {
cv::Point2f centroid(moments.m10 / area, moments.m01 / area);
cv::circle(img, centroid, 4, cv::Scalar(0, 255, 0), -1);
cv::RotatedRect rotated_rect = cv::fitEllipse(contours[i]);
cv::ellipse(img, rotated_rect, cv::Scalar(0, 0, 255), 2);
std::cout << "Centroid: (" << centroid.x << ", " << centroid.y << ")" << std::endl;
std::cout << "Angle: " << rotated_rect.angle << " degrees" << std::endl;
}
}
// 显示结果
cv::imshow("Result", img);
cv::waitKey(0);
return 0;
}
```
### 解释
1. **读取图像**:使用 `cv::imread` 读取图像文件。
2. **颜色通道选择**:将 BGR 图像转换为 HSV 格式,并提取 V 通道。
3. **阈值化分割**:使用 Otsu 方法进行自适应阈值处理,得到二值图像。
4. **形态学操作**:通过开运算和膨胀操作去除噪声和填充孔洞。
5. **分水岭算法**:使用分水岭算法进行图像分割。
6. **计算质心和方向**:使用 `cv::moments` 和 `cv::fitEllipse` 计算每个分割区域的质心和旋转矩形的方向。
希望这段代码对你有所帮助!如果有任何问题,请随时提问。
阅读全文