Spaces:

lnyan
/

stablediffusion-infinity

Sleeping

Update files

09c675d over 1 year ago

No virus

915 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;
	}