Границы водосбора тесно окружают одну область

Я пытаюсь сделать в среднем две капли в OpenCV. Для этого я планировал использовать алгоритм водораздела на изображении, предварительно обработанном следующим образом:

cv::Mat common, diff, processed, result;
cv::bitwise_and(blob1, blob2, common); //calc common area of the two blobs
cv::absdiff(blob1, blob2, diff);       //calc area where they differ

cv::distanceTransform(diff, processed, CV_DIST_L2, 3); //idea here is that the highest intensity
//will be in the middle of the differing area
cv::normalize(processed, processed, 0, 255, cv::NORM_MINMAX, CV_8U); //convert floats to bytes

cv::Mat watershedMarkers, watershedOutline;
common.convertTo(watershedMarkers, CV_32S, 1. / 255, 1); //change background to label 1, common area to label 2
watershedMarkers.setTo(0, processed); //set 0 (unknown) for area where blobs differ

cv::cvtColor(processed, processed, CV_GRAY2RGB); //watershed wants 3 channels
cv::watershed(processed, watershedMarkers);
cv::rectangle(watershedMarkers, cv::Rect(0, 0, watershedMarkers.cols, watershedMarkers.rows), 1); //remove the outline

//draw the boundary in red (for debugging)
watershedMarkers.convertTo(watershedOutline, CV_16S);
cv::threshold(watershedOutline, watershedOutline, 0, 255, CV_THRESH_BINARY_INV);
watershedOutline.convertTo(watershedOutline, CV_8U);
processed.setTo(cv::Scalar(CV_RGB(255, 0, 0)), watershedOutline);

//convert computed labels back to mask (blob), less relevant but shows my ultimate goal
watershedMarkers.convertTo(watershedMarkers, CV_8U);
cv::threshold(watershedMarkers, watershedMarkers, 1, 0, CV_THRESH_TOZERO_INV);
cv::bitwise_not(watershedMarkers * 255, result);

Моя проблема с результатами заключается в том, что расчетная граница (почти) всегда прилегает к области, общей для обоих объектов. Вот картинки:

Входные маркеры (черный = 0, серый = 1, белый = 2)
Входные маркеры

Исходное изображение водораздела (результат преобразования расстояния) с результирующим контуром, нарисованным красным:
входное изображение водораздела с нарисованным контуром

Я ожидаю, что граница пройдет вдоль области максимальной интенсивности ввода (то есть вдоль середины отличающейся области). Вместо этого (как вы можете видеть) он в основном проходит вокруг области, обозначенной как 2, с небольшим смещением, чтобы коснуться фона (отмечен как 1). Я делаю что-то здесь не так, или я неправильно понял, как работает водораздел?

1

Решение

Начиная с этого изображения:

введите описание изображения здесь

Вы можете получить правильный результат, просто передав все-ноль алгоритм изображения к водоразделу. «Бассейн» затем в равной степени заполнены «воды» начиная с каждого «боковая сторона» (затем просто не забудьте удалить внешнюю границу, которая по умолчанию установлена ​​на -1 по алгоритму водораздела):

введите описание изображения здесь

Код:

#include <opencv2\opencv.hpp>

using namespace cv;
using namespace std;

int main()
{
Mat1b img = imread("path_to_image", IMREAD_GRAYSCALE);

Mat1i markers(img.rows, img.cols, int(0));
markers.setTo(1, img == 128);
markers.setTo(2, img == 255);

Mat3b image(markers.rows, markers.cols, Vec3b(0,0,0));
markers.convertTo(markers, CV_32S);
watershed(image, markers);

Mat3b result;
cvtColor(img, result, COLOR_GRAY2BGR);
result.setTo(Scalar(0, 0, 255), markers == -1);

imshow("Result", result);
waitKey();

return(0);
}
1

Другие решения

Других решений пока нет …

По вопросам рекламы [email protected]