如何使用opencv在鱼眼校正图像中绘制内接矩形？
这就是我裁剪图像所需要的：矩形的 4 个角点或边的中点。
最终手动更正图像：lh3.googleusercontent.com/-KAylBUEtYO0/U7KJhDEhVWI/AAAAAAAQDY/UA1NRjCRYNk/w742-h434-no/rect.jpg
问问题
1037 次
1 回答
1
使用我自己的代码
int main()
{
//cv::Mat input = cv::imread("interiorBBExtraction.png");
//cv::Mat input = cv::imread("LenaWithBG.png");
cv::Mat input = cv::imread("fisheye_interior.png");
cv::Mat gray;
cv::cvtColor(input,gray,CV_BGR2GRAY);
cv::imshow("gray", gray);
cv::Mat mask = gray>0;
cv::imshow("mask", mask);
std::vector<std::vector<cv::Point> > contours;
std::vector<cv::Vec4i> hierarchy;
cv::findContours(mask,contours,hierarchy, CV_RETR_EXTERNAL, CV_CHAIN_APPROX_NONE, cv::Point(0,0));
std::cout << "found contours: " << contours.size() << std::endl;
cv::Mat contourImage = cv::Mat::zeros( input.size(), CV_8UC3 );;
//find contour with max elements
// remark: in theory there should be only one single outer contour surrounded by black regions!!
unsigned int maxSize = 0;
unsigned int id = 0;
for(unsigned int i=0; i<contours.size(); ++i)
{
if(contours.at(i).size() > maxSize)
{
maxSize = contours.at(i).size();
id = i;
}
}
std::cout << "chosen id: " << id << std::endl;
std::cout << "max size: " << maxSize << std::endl;
/// Draw contour
cv::Scalar color = cv::Scalar( 255, 255,255);
//cv::drawContours( contourImage, contours, id, color, 1, 8, hierarchy, 0, cv::Point() );
cv::drawContours( contourImage, contours, id, color, 1, 8, hierarchy, 0, cv::Point() );
cv::Mat contourMask = cv::Mat::zeros( input.size(), CV_8UC1 );
cv::drawContours( contourMask, contours, id, cv::Scalar(255), -1, 8, hierarchy, 0, cv::Point() );
cv::imshow("contour mask", contourMask);
cv::imshow("contours",contourImage);
std::vector<cv::Point> usedContour = contours.at(id);
std::vector<cv::Point> cSortedX = contours.at(id);
//std::cout << cSortedX.size() << std::endl;
std::sort(cSortedX.begin(), cSortedX.end(), sortX);
std::cout << cSortedX.front().x << ",..., " << cSortedX.back().x << std::endl;
std::vector<cv::Point> cSortedY = contours.at(id);
//std::cout << cSortedY.size() << std::endl;
std::sort(cSortedY.begin(), cSortedY.end(), sortY);
std::cout << cSortedY.front().y << ",..., " << cSortedY.back().y << std::endl;
unsigned int minXId = 0;
unsigned int maxXId = cSortedX.size()-1;
unsigned int minYId = 0;
unsigned int maxYId = cSortedY.size()-1;
cv::Rect interiorBB;
while( (minXId<maxXId)&&(minYId<maxYId) )
{
cv::Point min(cSortedX[minXId].x, cSortedY[minYId].y);
cv::Point max(cSortedX[maxXId].x, cSortedY[maxYId].y);
//std::cout << min << " ... " << max << std::endl;
interiorBB = cv::Rect(min.x,min.y, max.x-min.x, max.y-min.y);
//std::cout << interiorBB << std::endl;
int ocTop = 0;
int ocBottom = 0;
int ocLeft = 0;
int ocRight = 0;
bool finished = checkInteriorExterior(contourMask, interiorBB, ocTop, ocBottom,ocLeft, ocRight);
if(finished)
{
break;
}
if(ocLeft)++minXId;
if(ocRight) --maxXId;
if(ocTop) ++minYId;
if(ocBottom)--maxYId;
}
std::cout << "done! : " << interiorBB << std::endl;
cv::Mat mask2 = cv::Mat::zeros(input.rows, input.cols, CV_8UC1);
cv::rectangle(mask2,interiorBB, cv::Scalar(255),-1);
cv::Mat maskedImage;
input.copyTo(maskedImage);
for(unsigned int y=0; y<maskedImage.rows; ++y)
for(unsigned int x=0; x<maskedImage.cols; ++x)
{
maskedImage.at<cv::Vec3b>(y,x)[2] = 255;
}
input.copyTo(maskedImage,mask2);
cv::imshow("masked image", maskedImage);
cv::imwrite("interiorBoundingBoxFisheyeResult.png", maskedImage);
cv::waitKey(-1);
return 0;
}
和
bool sortX(cv::Point a, cv::Point b)
{
bool ret = false;
if(a.x == a.x)
if(b.x==b.x)
ret = a.x < b.x;
return ret;
}
bool sortY(cv::Point a, cv::Point b)
{
bool ret = false;
if(a.y == a.y)
if(b.y == b.y)
ret = a.y < b.y;
return ret;
}
bool checkInteriorExterior(const cv::Mat&mask, const cv::Rect&interiorBB, int&top, int&bottom, int&left, int&right)
{
bool returnVal = true;
//std::cout << interiorBB << std::endl;
cv::Mat sub = mask(interiorBB);
//cv::imshow("sub",sub); cv::waitKey(10);
unsigned int max = 0;
// top row:
unsigned int x=0;
unsigned int y=0;
unsigned int cTop=0;
unsigned int cBottom=0;
unsigned int cLeft=0;
unsigned int cRight=0;
for(y=0, x=0 ; x<sub.cols; ++x)
{
// if there is an exterior part in the interior we have to move the top side of the rect a bit to the bottom
if(sub.at<unsigned char>(y,x) == 0)
{
// std::cout << "top: " << cv::Point(x,y) << std::endl;
returnVal = false;
++cTop;
}
}
for(y=sub.rows-1, x=0; x<sub.cols; ++x)
{
// if there is an exterior part in the interior we have to move the bottom side of the rect a bit to the top
if(sub.at<unsigned char>(y,x) == 0)
{
// std::cout << "bottom: " << cv::Point(x,y) << std::endl;
returnVal = false;
++cBottom;
}
}
for(y=0, x=0 ; y<sub.rows; ++y)
{
// if there is an exterior part in the interior
if(sub.at<unsigned char>(y,x) == 0)
{
// std::cout << "left: " << cv::Point(x,y) << std::endl;
returnVal = false;
++cLeft;
}
}
for(x=sub.cols-1, y=0; y<sub.rows; ++y)
{
// if there is an exterior part in the interior
if(sub.at<unsigned char>(y,x) == 0)
{
// std::cout << "right: " << cv::Point(x,y) << std::endl;
returnVal = false;
++cRight;
}
}
if(cTop > cBottom)
{
if(cTop > cLeft)
if(cTop > cRight)
top = 1;
}
else
if(cBottom > cLeft)
if(cBottom > cRight)
bottom = 1;
if(cLeft >= cRight)
{
if(cLeft >= cBottom)
if(cLeft >= cTop)
left = 1;
}
else
if(cRight >= cTop)
if(cRight >= cBottom)
right = 1;
return returnVal;
}
并使用此图像作为输入：
我得到这个输出：
不确定如果绿色边框消失会发生什么（没有这样的图像），因为原始图像也包含一些黑色部分。
于 2014-07-02T10:02:39.647 回答