Преобразование 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.