Spaces:
Runtime error
Runtime error
| import numpy as np | |
| def get_mask(W, H, ratio_w=0.9, ratio_h=0.9): | |
| w = int(W * ratio_w) | |
| h = int(H * ratio_h) | |
| x1 = (W - w) // 2 | |
| x2 = x1 + w | |
| y1 = (H - h) // 2 | |
| y2 = y1 + h | |
| mask = np.ones((H, W), dtype=np.float32) | |
| # top | |
| row = np.linspace(0, 0, w)[None, :] # (1, w) | |
| col = np.linspace(0, 1, y1)[:, None] # (y1, 1) | |
| grad_t = np.sqrt(row**2 + col**2).astype(np.float32) # (y1, w) | |
| mask[0:y1, x1:x2] = grad_t | |
| # bottom | |
| row = np.linspace(0, 0, w)[None, :] # (1, w) | |
| col = np.linspace(1, 0, H - y2)[:, None] # (H-y2, 1) | |
| grad_b = np.sqrt(row**2 + col**2).astype(np.float32) # (H-y2, w) | |
| mask[y2:H, x1:x2] = grad_b | |
| # left | |
| row = np.linspace(0, 1, x1)[None, :] # (1, x1) | |
| col = np.linspace(0, 0, h)[:, None] # (h, 1) | |
| grad_l = np.sqrt(row**2 + col**2).astype(np.float32) # (h, x1) | |
| mask[y1:y2, 0:x1] = grad_l | |
| # right | |
| row = np.linspace(1, 0, W - x2)[None, :] # (1, W-x2) | |
| col = np.linspace(0, 0, h)[:, None] # (h, 1) | |
| grad_r = np.sqrt(row**2 + col**2).astype(np.float32) # (h, W-x2) | |
| mask[y1:y2, x2:W] = grad_r | |
| # top left | |
| row = np.linspace(1, 0, x1)[None, :] # (1, w) | |
| col = np.linspace(1, 0, y1)[:, None] # (y1, 1) | |
| grad_tl = np.sqrt(row**2 + col**2).astype(np.float32) # (y1, x1) | |
| grad_tl = 1 - np.clip(grad_tl, 0, 1) | |
| mask[0:y1, 0:x1] = grad_tl | |
| # top right | |
| row = np.linspace(0, 1, W - x2)[None, :] # (1, W-x2) | |
| col = np.linspace(1, 0, y1)[:, None] # (y1, 1) | |
| grad_tr = np.sqrt(row**2 + col**2).astype(np.float32) # (y1, W-x2) | |
| grad_tr = 1 - np.clip(grad_tr, 0, 1) | |
| mask[0:y1, x2:W] = grad_tr | |
| # bottom left | |
| row = np.linspace(1, 0, x1)[None, :] # (1, x1) | |
| col = np.linspace(0, 1, H - y2)[:, None] # (H-y2, 1) | |
| grad_bl = np.sqrt(row**2 + col**2).astype(np.float32) # (H-y2, x1) | |
| grad_bl = 1 - np.clip(grad_bl, 0, 1) | |
| mask[y2:H, 0:x1] = grad_bl | |
| # bottom right | |
| row = np.linspace(0, 1, W - x2)[None, :] # (1, W-x2) | |
| col = np.linspace(0, 1, H - y2)[:, None] # (H-y2, 1) | |
| grad_br = np.sqrt(row**2 + col**2).astype(np.float32) # (H-y2, W-x2) | |
| grad_br = 1 - np.clip(grad_br, 0, 1) | |
| mask[y2:H, x2:W] = grad_br | |
| return mask[:, :, None] | |