C++でOpenCVをはじめる｜撮影したレシートを自動でトリミング

cmake_minimum_required(VERSION 3.28)
project(resitori_core)

set(CMAKE_CXX_STANDARD 11)

find_package(OpenCV REQUIRED) # 追加
include_directories(${OpenCV_INCLUDE_DIRS}) # 追加

add_executable(resitori_core main.cpp)

target_link_libraries(resitori_core ${OpenCV_LIBS}) # 追加

brew --prefix opencv

#include <opencv2/opencv.hpp>
#include <iostream>

int main() {
    cv::Mat image = cv::imread("/somewhere/sample.jpg");
    if (image.empty()) {
        std::cerr << "画像が読み込めませんでした。" << std::endl;
        return -1;
    }

    cv::imshow("表示テスト", image);
    cv::waitKey(0);
    return 0;
}

#include <opencv2/opencv.hpp>
#include <iostream>

int main() {
    // 入力画像ファイルの読み込み
    cv::Mat colorImage = cv::imread("/somewhere/sample.jpg");
    if (colorImage.empty()) {
        std::cerr << "画像が読み込めません。" << std::endl;
        return -1;
    }

    // グレースケール変換
    cv::Mat grayImage;
    cv::cvtColor(colorImage, grayImage, cv::COLOR_BGR2GRAY);

    // グレースケール画像を保存
    bool result = cv::imwrite("gray_sample.jpg", grayImage);
    if (!result) {
        std::cerr << "保存に失敗しました。" << std::endl;
        return -1;
    }

    std::cout << "グレースケール画像を保存しました。" << std::endl;
    cv::imshow("表示テスト", grayImage);
    cv::waitKey(0);
    return 0;
}

#include <opencv2/opencv.hpp>
#include <iostream>

#include <opencv2/opencv.hpp>
#include <iostream>

int main() {
    // 入力画像を読み込み
    cv::Mat colorImage = cv::imread("/somewhere/sample.jpg");
    if (colorImage.empty()) {
        std::cerr << "画像が読み込めません。" << std::endl;
        return -1;
    }

    // グレースケールに変換
    cv::Mat grayImage;
    cv::cvtColor(colorImage, grayImage, cv::COLOR_BGR2GRAY);

    // ノイズ除去（Canny前にぼかすと良い結果が得られる）
    cv::Mat blurred;
    cv::GaussianBlur(grayImage, blurred, cv::Size(5, 5), 1.4);

    // エッジ検出（Canny）
    cv::Mat edgeImage;
    cv::Canny(blurred, edgeImage, 50, 150);  // しきい値は適宜調整

    // エッジ画像を保存
    bool result = cv::imwrite("edges.jpg", edgeImage);
    if (!result) {
        std::cerr << "エッジ画像の保存に失敗しました。" << std::endl;
        return -1;
    }

    std::cout << "エッジ画像を保存しました。" << std::endl;

    // ウィンドウで表示（必要なら）
    cv::imshow("Edges", edgeImage);
    cv::waitKey(0);

    return 0;
}

#include <opencv2/opencv.hpp>
#include <iostream>

#include <opencv2/opencv.hpp>
#include <iostream>

int main() {
    // 入力画像を読み込み
    cv::Mat colorImage = cv::imread("/somewhere/sample.jpg");
    if (colorImage.empty()) {
        std::cerr << "画像が読み込めません。" << std::endl;
        return -1;
    }

    // グレースケール → ぼかし → 二値化（エッジ不要）
    cv::Mat gray, blurred, binary;
    cv::cvtColor(colorImage, gray, cv::COLOR_BGR2GRAY);
    cv::imwrite("1_gray.jpg", gray);
    cv::GaussianBlur(gray, blurred, cv::Size(5, 5), 0);
    cv::imwrite("2_blurred.jpg", blurred);
    cv::threshold(blurred, binary, 0, 255, cv::THRESH_BINARY + cv::THRESH_OTSU);
    cv::imwrite("3_binary.jpg", binary);

    // エッジ検出
    std::vector<std::vector<cv::Point>> contours;
    cv::findContours(binary, contours, cv::RETR_EXTERNAL, cv::CHAIN_APPROX_SIMPLE);

    std::vector<cv::Point> receiptContour;
    for (const auto& contour : contours) {
        double area = cv::contourArea(contour);
        if (area < 10000) continue; // 小さな輪郭を無視

        std::vector<cv::Point> approx;
        cv::approxPolyDP(contour, approx, cv::arcLength(contour, true) * 0.02, true);

        if (approx.size() == 4 && cv::isContourConvex(approx)) {
            receiptContour = approx;
            break;
        }
    }

    if (receiptContour.empty()) {
        std::cerr << "四角形の輪郭が見つかりませんでした。" << std::endl;
        return -1;
    }

    // 四隅マスク作成（白＝不透明、黒＝透明）
    cv::Mat mask = cv::Mat::zeros(colorImage.size(), CV_8UC1);
    cv::fillConvexPoly(mask, receiptContour, 255);
    cv::imwrite("5_mask.png", mask);


    // RGBA画像を生成（透明化）
    cv::Mat output(colorImage.size(), CV_8UC4);
    for (int y = 0; y < colorImage.rows; ++y) {
        for (int x = 0; x < colorImage.cols; ++x) {
            cv::Vec3b color = colorImage.at<cv::Vec3b>(y, x);
            uchar alpha = mask.at<uchar>(y, x);
            output.at<cv::Vec4b>(y, x) = cv::Vec4b(color[0], color[1], color[2], alpha);
        }
    }

    // PNG形式で保存（透過可能）
    bool saved = cv::imwrite("receipt_cropped.png", output);
    if (!saved) {
        std::cerr << "画像の保存に失敗しました。" << std::endl;
        return -1;
    }

    std::cout << "透過画像として保存しました。" << std::endl;


    return 0;
}