算法思路:
1图片灰度化
2ostu阈值分割
3形态学处理,去除细小噪声
4使用canny轮廓提取,根据面积大小作为约束条件
未做完。
main.cpp
```代码块
#include"Cal.h"
using namespace std;
using namespace cv;
int main(int argc, char* argv[])
{
std::string folder_path = "D:\\Aluosang\\test.jpg"; //更改文件夹测试,或改为实时读取图片 img
std::vector
cv::glob(folder_path, file_names);
cv::Mat img;
for (int i = 0; i < file_names.size(); i++) {
std::cout << file_names[i] << std::endl; //黑窗口输出文件名字,如果实时更新,for循环删改。
img = cv::imread(file_names[i]);
if (!img.data) {
continue;
}
if (img.empty()) // 判断读入图片是否为空
{
cout << "image is empty" << endl;
return -1;
}
/*cv::Mat src_crop = img(cv::Rect(200, 300, 700, 500)); // 裁剪后的图 src_crop*/
Cal ed(img);
ed.cannyProcess(200, 220);
ed.getContours();
cv::imshow("img", img); //输出显示
cv::waitKey(2000); //调时间快慢
waitKey(0);
return 0;
}
}
```
Cal.cpp
```
#include "Cal.h"
Cal::Cal(cv::Mat image)
{
m_img = image;
}
bool Cal::cannyProcess(unsigned int downThreshold, unsigned int upThreshold)
{
bool ret = true;
if (m_img.empty())
{
ret = false;
}
cv::Canny(m_img, m_canny, downThreshold, upThreshold);
cv::imshow("Canny", m_canny);
return ret;
}
bool Cal::getContours()
{
bool ret = true;
if (m_canny.empty())
{
ret = false;
}
cv::Mat k = cv::getStructuringElement(cv::MORPH_RECT, cv::Size(3, 3), cv::Point(-1, -1));
cv::dilate(m_canny, m_canny, k);
imshow("dilate", m_canny);
// 轮廓发现与绘制
vector
vector
findContours(m_canny, contours, cv::RETR_EXTERNAL, CHAIN_APPROX_SIMPLE, Point());
for (size_t i = 0; i < contours.size(); ++i)
{
// 最大外接轮廓
cv::Rect rect = cv::boundingRect(contours[i]);
cv::rectangle(m_img, rect, cv::Scalar(0, 255, 0), 2, LINE_8);
// 最小外接轮廓
RotatedRect rrt = minAreaRect(contours[i]);
Point2f pts[4];
rrt.points(pts);
// 绘制旋转矩形与中心位置
for (int i = 0; i < 4; i++) {
line(m_img, pts[i % 4], pts[(i + 1) % 4], Scalar(0, 0, 255), 2, 8, 0);
}
Point2f cpt = rrt.center;
circle(m_img, cpt, 2, Scalar(255, 0, 0), 2, 8, 0);
}
imshow("contours", m_img);
return ret;
}
Cal::~Cal()
{
}
```
Cal.h
```
#pragma once
#include
#include
using namespace std;
using namespace cv;
class Cal
{
cv::Mat m_img;
cv::Mat m_canny;
public:
Cal(cv::Mat iamge);
bool cannyProcess(unsigned int downThreshold, unsigned int upThreshold);
bool getContours();
~Cal();
};
```