Преобразование OpenCV-Python в С++/Objective C
Я нашел этот очень хороший код из @abid-rahman-k, чтобы обнаружить прямоугольник из изображения:
OpenCV С++/Obj-C: расширенное обнаружение квадратов
Теперь код находится в Python, и вот он:
import cv2
import numpy as np
img = cv2.imread('sof.jpg')
img = cv2.resize(img,(500,500))
gray = cv2.cvtColor(img,cv2.COLOR_BGR2GRAY)
ret,thresh = cv2.threshold(gray,127,255,0)
contours,hier = cv2.findContours(thresh,cv2.RETR_LIST,cv2.CHAIN_APPROX_SIMPLE)
for cnt in contours:
if cv2.contourArea(cnt)>5000: # remove small areas like noise etc
hull = cv2.convexHull(cnt) # find the convex hull of contour
hull = cv2.approxPolyDP(hull,0.1*cv2.arcLength(hull,True),True)
if len(hull)==4:
cv2.drawContours(img,[hull],0,(0,255,0),2)
cv2.imshow('img',img)
cv2.waitKey(0)
cv2.destroyAllWindows()
Я хотел бы преобразовать его в Objective C/С++. Это то, что я сделал, но не работал, что я пропустил?
- (void)processImage2:(cv::Mat&)image;
{
cv::Mat img,gray,thresh;
std::vector<std::vector<cv::Point> > contours;
// cv::resize(image.clone(), image, cv::Size(500,500) );
cvtColor(image, gray, cv::COLOR_BGR2GRAY );
cv::threshold(gray, thresh, 127, 255, 0);
findContours(thresh, contours, CV_RETR_LIST, CV_CHAIN_APPROX_SIMPLE);
std::vector<cv::Point> hull;
for (size_t i = 0; i < contours.size(); i++)
{
if (cv::contourArea(contours[i])>5000){
cv::convexHull(contours[i],hull);
approxPolyDP(hull, hull, 0.1*arcLength(hull, true), true);
if (hull.size() ==4)
cv::drawContours(image,hull,0,cv::Scalar(0,255,0),2);
}
}
}
Update:
Программа запускается, но после того, как я выберем изображение, которое он сработает, и я получаю эту ошибку:
Nov 28 10:26:52 Anas-Basalamahs-MacBook-Air.local OpenCV Tutorial[18861] <Error>: CGBitmapContextCreate: unsupported parameter combination: 0 integer bits/component; 0 bits/pixel; 0-component color space; kCGImageAlphaNone; 0 bytes/row.
Nov 28 10:26:52 Anas-Basalamahs-MacBook-Air.local OpenCV Tutorial[18861] <Error>: CGContextConcatCTM: invalid context 0x0
Nov 28 10:26:52 Anas-Basalamahs-MacBook-Air.local OpenCV Tutorial[18861] <Error>: CGContextSetInterpolationQuality: invalid context 0x0
Nov 28 10:26:52 Anas-Basalamahs-MacBook-Air.local OpenCV Tutorial[18861] <Error>: CGContextDrawImage: invalid context 0x0
Nov 28 10:26:52 Anas-Basalamahs-MacBook-Air.local OpenCV Tutorial[18861] <Error>: CGBitmapContextCreateImage: invalid context 0x0
2012-11-28 10:26:52.963 OpenCV Tutorial[18861:c07] resized image size: NSSize: {0, 0}
OpenCV Error: Assertion failed (i < 0) in getMat, file /Users/bloodaxe/Develop/opencv/modules/core/src/matrix.cpp, line 957
terminate called throwing an exception(lldb)
Ответы
Ответ 1
Я не вижу в вашем коде ошибки, поэтому я предполагаю, что вы читаете изображение вне этой функции и передаете его. Пробовали ли вы делать imshow сразу после прочтения, чтобы обеспечить правильную загрузку изображения.
Первая строка указывает мне, что это может быть проблемой.
Ответ 2
Проблема в
cv::drawContours(image,hull,0,cv::Scalar(0,255,0),2);
по какой-то причине ему не нравится массив hull
...
Поэтому я предполагаю, что что-то не так происходит в одной из этих строк:
cv::convexHull(contours[i],hull);
approxPolyDP(hull, hull, 0.1*arcLength(hull, true), true);
(вы не говорите Шерлоку!)
Я попытался найти решение, играющее с кодом из этих ссылок:
Я надеюсь, что это поможет вам больше, чем со мной.
Кроме того, извините за отправку его в качестве ответа... но моя репутация была низкой, и я не мог добавить комментарий.
Ответ 3
Здесь это работает:
cv::Mat oResult, oToShow;
// roMat is your input image
cv::cvtColor(roMat, oToShow, CV_GRAY2BGR);
cv::threshold(roMat, oResult, 127, 255, 0);
std::vector<std::vector<cv::Point>> contours;
std::vector<cv::Vec4i> hierarchy;
cv::findContours( oResult, contours, hierarchy, CV_RETR_LIST, CV_CHAIN_APPROX_SIMPLE );
std::vector<std::vector<cv::Point> > oHull(contours.size());
for( int i = 0; i != contours.size(); i++)
{
if( cv::contourArea(contours[i]) > 5000 )
{
cv::convexHull(contours[i], oHull[i]);
cv::approxPolyDP(oHull[i], oHull[i], 0.1*cv::arcLength(oHull[i], true), true);
if( oHull[i].size() == 4 )
{
cv::drawContours(oToShow, oHull, -1, cv::Scalar(0, 255,0), 2, 8);
}
}
}
cv::imshow("hull", oToShow);
cv::waitKey(0);
Я думаю, что ошибка состояла в том, что у вас действительно был массив Array, а не только один Hull.