OpenCV обнаружил неправильный оператор знака равенства

Question

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

import cv2
import numpy as np
from PIL import Image, ImageOps



img = cv2.imread("sum.png")
gray = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)

#Gaussian Blurring to reduce noise
blur = cv2.GaussianBlur(gray, (5,5), 0)

#Adaptive Thresholding to account for different light/shadows
threshed = cv2.adaptiveThreshold(gray,255,cv2.ADAPTIVE_THRESH_GAUSSIAN_C,cv2.THRESH_BINARY,11,2)


morphed = cv2.morphologyEx(threshed, cv2.MORPH_OPEN, np.ones((3,3)))

conturs_lst = cv2.findContours(morphed, cv2.RETR_TREE, cv2.CHAIN_APPROX_SIMPLE)[-2]


for cnt in conturs_lst:
    x,y,w,h = cv2.boundingRect(cnt)
    cv2.rectangle(img, (x,y), (x+w, y+h), (255, 0, 255), 1, cv2.LINE_AA)

Этот код приводит к такому поведению: две части знака равенства ограничены в отдельных полях.

Исходное изображение:

Фактический вывод:

Кто-нибудьзнаете, как можно было бы изменить мой код, чтобы весь знак равенства находился в одной ограничительной рамке?

Answer 1

Вы можете попробовать что-то вроде этого:

cv::namedWindow("result1", cv::WINDOW_FREERATIO);
cv::namedWindow("result2", cv::WINDOW_FREERATIO);
cv::Mat img = cv::imread(R"(FRfLi.png)");

// to gray
cv::Mat gray;
cv::cvtColor(img, gray, cv::COLOR_BGR2GRAY);
cv::threshold(gray, gray, 80, 255, cv::THRESH_BINARY_INV);

cv::morphologyEx(gray, gray, cv::MORPH_CLOSE, cv::getStructuringElement(cv::MORPH_CROSS, cv::Size(3, 27)));
cv::imshow("result1", gray);

std::vector<std::vector<cv::Point> > contours;
cv::findContours(gray, contours, cv::RETR_EXTERNAL, cv::CHAIN_APPROX_SIMPLE);

for (int i = 0; i < contours.size(); ++i) {
    cv::Rect rect = cv::boundingRect(contours[i]);
    rect.x -= 3;
    rect.y -= 3;
    rect.width += 6;
    rect.height += 6;
    cv::rectangle(img, rect, cv::Scalar(0, 0, 255), 2);
}

cv::imshow("result2", img);
cv::waitKey();

После морфинга вы получите такой результат:

И обнаружив контуры вы получите такой результат:

ПРИМЕЧАНИЕ: Код написан на C ++, вы можете просто рассмотреть этапы его реализации.

Надеюсь, это поможет вам!