Spaces:

lnyan
/

stablediffusion-infinity

Runtime error

stablediffusion-infinity / PyPatchMatch /examples /cpp_example.cpp

Add files

5704551 about 4 years ago

946 Bytes

#include <iostream>
#include <opencv2/imgcodecs.hpp>
#include <opencv2/highgui.hpp>

#include "masked_image.h"
#include "nnf.h"
#include "inpaint.h"

int main() {
    auto source = cv::imread("./images/forest_pruned.bmp", cv::IMREAD_COLOR);

    auto mask = cv::Mat(source.size(), CV_8UC1);
    mask = cv::Scalar::all(0);
    for (int i = 0; i < source.size().height; ++i) {
        for (int j = 0; j < source.size().width; ++j) {
            auto source_ptr = source.ptr<unsigned char>(i, j);
            if (source_ptr[0] == 255 && source_ptr[1] == 255 && source_ptr[2] == 255) {
                mask.at<unsigned char>(i, j) = 1;
            }
        }
    }

    auto metric = PatchSSDDistanceMetric(3);
    auto result = Inpainting(source, mask, &metric).run(true, true);
    // cv::imwrite("./images/forest_recovered.bmp", result);
    // cv::imshow("Result", result);
    // cv::waitKey();

    return 0;
}

	#include <iostream>
	#include <opencv2/imgcodecs.hpp>
	#include <opencv2/highgui.hpp>

	#include "masked_image.h"
	#include "nnf.h"
	#include "inpaint.h"

	int main() {
	auto source = cv::imread("./images/forest_pruned.bmp", cv::IMREAD_COLOR);

	auto mask = cv::Mat(source.size(), CV_8UC1);
	mask = cv::Scalar::all(0);
	for (int i = 0; i < source.size().height; ++i) {
	for (int j = 0; j < source.size().width; ++j) {
	auto source_ptr = source.ptr<unsigned char>(i, j);
	if (source_ptr[0] == 255 && source_ptr[1] == 255 && source_ptr[2] == 255) {
	mask.at<unsigned char>(i, j) = 1;
	}
	}
	}

	auto metric = PatchSSDDistanceMetric(3);
	auto result = Inpainting(source, mask, &metric).run(true, true);
	// cv::imwrite("./images/forest_recovered.bmp", result);
	// cv::imshow("Result", result);
	// cv::waitKey();

	return 0;
	}