stablediffusion-infinity/PyPatchMatch/examples/py_example_global_mask.py

#! /usr/bin/env python3
# -*- coding: utf-8 -*-
# File   : test.py
# Author : Jiayuan Mao
# Email  : [email protected]
# 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')