В какую сторону мне стоит дальше изучать opencv, чтобы улучшить алгоритм?

@OrangeVagon · Регистрация: 24.12.2022

Author24 — интернет-сервис помощи студентам

Здравствуйте, дали задачку на стажировку - найти спичечные коробки на фото и пометить их красной границей. Можно использовать opencv, stl, boost. Обязательно использование примитивов opencv. Подскажите, в какую сторону мне рыть, чтобы оптимизировать следующий код, дабы он выдавал более точное позиционирование коробков на фото?

C++

#include <opencv2/opencv.hpp>
using namespace cv;
using namespace std;
 
int main(int argc, char* argv[])
{
    Mat image = imread("C:\\Users\\79\\Desktop\\j1.jpg");
    Mat gray;
    cvtColor(image, gray, COLOR_BGR2GRAY);
    GaussianBlur(gray, gray, Size(5, 5), 0);
    Mat bin_gray;
    threshold(gray, bin_gray, 100, 255, THRESH_BINARY);
 
    vector<vector<Point>> contours;
    vector<Vec4i> hierarchy;
    findContours(bin_gray, contours, hierarchy, RETR_EXTERNAL, CHAIN_APPROX_SIMPLE);
 
    vector<vector<Point>> filtered_contours;
    double minCountourArea = 180.0;
    for (size_t i = 0; i < contours.size(); i++)
    {
        approxPolyDP(contours[i], contours[i], 4, true);
        
        if (contourArea(contours[i]) >= minCountourArea) 
        {
            filtered_contours.push_back(contours[i]);
        }
    }
 
    drawContours(image, filtered_contours, -1, Scalar(0, 0, 255), 2);
    
    imwrite("C:\\Users\\79\\Desktop\\result.jpg", image);
    
    imshow("Result", image);
 
    waitKey(0);
}

Результат, который я получил:

В какую сторону мне стоит дальше изучать opencv, чтобы улучшить алгоритм?

Я понимаю, что в моем коде сейчас не достает проверок на существование загружаемого изображения, допилю их позже. Хотелось бы получить материалы, которые помогут мне более конкретно выделить спичечные коробки на фото. Сам в opencv неделю от силы, не стоит в целях сдать задачу вовремя и получить работу, сейчас задача разобраться в opencv на конкретных задачах.

@Laborant666 · 26.09.2023, 00:33

OrangeVagon, вы же понимаете что в данный момент у вас выделяются не коробки, а контуры с площадью больше 180 единиц? И попадают сюда не только коробки, а и чиркаш коробка сбоку, и человечек на одном из коробков? В данный момент здесь нет никакого детектирования коробка. Начните с малого - вместо отсеивания контуров по площади отсеивать начните контуры, которые имеют в себе градусы 90+- 20 - у коробка в любом случае есть углы. Уберите размытие, на мой взгляд, оно лишнее, ну или хотя бы сравните с ним и без него десяток изображений. С бинаризацией я бы повременил, перевода изображения в серый одноканал достаточно для скорости на небольших изображениях, да и такого рода бинаризация, как я вижу, вредит - у вас контуры с большой площадью нашлись не все. В общем для начала переберите варианты подготовки изображения, что бы добиться нахождения 90-95% всех контуров. Потом фильтруйте по признакам присущим форме коробка, потом может и по площади фильтровать уже, но мне кажется это не нужно, будет скипать коробки, которые лежат далеко и из-за перспективы в кадре по площади малы

@Laborant666 208 / 101 / 20 Регистрация: 30.06.2020 Сообщений: 848
	26.09.2023, 00:33	2
	OrangeVagon, вы же понимаете что в данный момент у вас выделяются не коробки, а контуры с площадью больше 180 единиц? И попадают сюда не только коробки, а и чиркаш коробка сбоку, и человечек на одном из коробков? В данный момент здесь нет никакого детектирования коробка. Начните с малого - вместо отсеивания контуров по площади отсеивать начните контуры, которые имеют в себе градусы 90+- 20 - у коробка в любом случае есть углы. Уберите размытие, на мой взгляд, оно лишнее, ну или хотя бы сравните с ним и без него десяток изображений. С бинаризацией я бы повременил, перевода изображения в серый одноканал достаточно для скорости на небольших изображениях, да и такого рода бинаризация, как я вижу, вредит - у вас контуры с большой площадью нашлись не все. В общем для начала переберите варианты подготовки изображения, что бы добиться нахождения 90-95% всех контуров. Потом фильтруйте по признакам присущим форме коробка, потом может и по площади фильтровать уже, но мне кажется это не нужно, будет скипать коробки, которые лежат далеко и из-за перспективы в кадре по площади малы 1