シード ポイントがわかっている場合、セグメンテーションは次の方法で簡単かつ非常に高速に実行できます (現在のマシンでは約 0.001 秒) floodFill
。
#include <opencv2/opencv.hpp>
#include <boost/chrono/include.hpp>
#include <iostream>
#include <ctime>
std::vector<std::vector<cv::Point>> segment_lane(const cv::Mat& img, const cv::Point& seed, int tol)
{
cv::Mat mask = cv::Mat::zeros(img.rows + 2, img.cols + 2, CV_8UC1);
cv::floodFill(img, mask, seed, 255, 0, cv::Scalar(tol, tol, tol), cv::Scalar(tol, tol, tol), 4 + (255 << 8) + cv::FLOODFILL_MASK_ONLY + cv::FLOODFILL_FIXED_RANGE);
std::vector<std::vector<cv::Point>> contours;
cv::findContours(mask, contours, CV_RETR_EXTERNAL, CV_CHAIN_APPROX_SIMPLE, cv::Point(-1, -1));
return contours;
}
int main()
{
cv::Mat img = cv::imread("Ph2PK.png");
auto start = boost::chrono::system_clock::now();
auto contours = segment_lane(img, cv::Point(21, 461), 40);
auto end = boost::chrono::system_clock::now();
boost::chrono::duration<double> elapsed_seconds = end - start;
std::cout << "elapsed time: " << elapsed_seconds.count() << "s\n";
cv::drawContours(img, contours, 0, cv::Scalar(255, 0, 0), 2);
cv::imshow("img", img);
cv::imwrite(out.png", img);
cv::waitKey(0);
}
次のコンターでは不十分な場合は、許容値を調整するか、省略してみてくださいFLOODFILL_FIXED_RANGE
。
