radames's picture
backend + frontend
96c927a unverified
raw
history blame
694 Bytes
#! /usr/bin/env python3
# -*- coding: utf-8 -*-
# File : test.py
# Author : Jiayuan Mao
# Email : maojiayuan@gmail.com
# Date : 01/09/2020
#
# Distributed under terms of the MIT license.
import numpy as np
from PIL import Image
import sys
sys.path.insert(0, '../')
import patch_match
if __name__ == '__main__':
patch_match.set_verbose(True)
source = Image.open('./images/forest_pruned.bmp')
source = np.array(source)
source[:100, :100] = 255
global_mask = np.zeros_like(source[..., 0])
global_mask[:100, :100] = 1
result = patch_match.inpaint(source, global_mask=global_mask, patch_size=3)
Image.fromarray(result).save('./images/forest_recovered.bmp')