Spaces:
Runtime error
Runtime error
File size: 623 Bytes
58f667f |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 |
import numpy as np
class Inpainter:
def __call__(self, img, height_top_mask, height_down_mask, width_left_mask, width_right_mask):
h = img.shape[0]
w = img.shape[1]
h_top_mask = int(float(h) / 100.0 * float(height_top_mask))
h_down_mask = int(float(h) / 100.0 * float(height_down_mask))
w_left_mask = int(float(w) / 100.0 * float(width_left_mask))
w_right_mask = int(float(w) / 100.0 * float(width_right_mask))
img_new = img
img_new[h_top_mask:h_down_mask, w_left_mask:w_right_mask] = 0
img_new = img_new.astype('ubyte')
return img_new
|