File size: 694 Bytes
e4545e2 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 |
#! /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')
|