utils.h

#pragma once
#include <opencv2/opencv.hpp>

// Colors
const std::vector<cv::Scalar> CITYSCAPES_COLORS = {
    cv::Scalar(128, 64, 128),
    cv::Scalar(232, 35, 244),
    cv::Scalar(70, 70, 70),
    cv::Scalar(156, 102, 102),
    cv::Scalar(153, 153, 190),
    cv::Scalar(153, 153, 153),
    cv::Scalar(30, 170, 250),
    cv::Scalar(0, 220, 220),
    cv::Scalar(35, 142, 107),
    cv::Scalar(152, 251, 152),
    cv::Scalar(180, 130, 70),
    cv::Scalar(60, 20, 220),
    cv::Scalar(0, 0, 255),
    cv::Scalar(142, 0, 0),
    cv::Scalar(70, 0, 0),
    cv::Scalar(100, 60, 0),
    cv::Scalar(90, 0, 0),
    cv::Scalar(230, 0, 0),
    cv::Scalar(32, 11, 119),
    cv::Scalar(0, 74, 111),
    cv::Scalar(81, 0, 81)
};

// Structure to hold clicked point coordinates
struct PointData {
    cv::Point point;
    bool clicked;
};

// Overlay mask on the image
void overlay(Mat& image, Mat& mask, Scalar color = Scalar(128, 64, 128), float alpha = 0.8f, bool showEdge = true)
{
    // Draw mask
    Mat ucharMask(image.rows, image.cols, CV_8UC3, color);
    image.copyTo(ucharMask, mask <= 0);
    addWeighted(ucharMask, alpha, image, 1.0 - alpha, 0.0f, image);

    // Draw contour edge
    if (showEdge)
    {
        vector<vector<Point>> contours;
        vector<Vec4i> hierarchy;
        findContours(mask <= 0, contours, hierarchy, RETR_TREE, CHAIN_APPROX_NONE);
        drawContours(image, contours, -1, Scalar(255, 255, 255), 2);
    }
}

// Function to handle mouse events
void onMouse(int event, int x, int y, int flags, void* userdata) {
    PointData* pd = (PointData*)userdata;
    if (event == cv::EVENT_LBUTTONDOWN) {
        // Save the clicked coordinates
        pd->point = cv::Point(x, y);
        pd->clicked = true;
    }
}