Spaces:
Starting
on
A10G
Starting
on
A10G
| #! /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') | |