File size: 1,110 Bytes
434b57f |
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 29 30 31 32 33 34 35 36 37 38 39 40 |
import io
import numpy as np
import PIL.Image
import requests
def load_image(url, size):
r = requests.get(url)
img = PIL.Image.open(io.BytesIO(r.content))
img.thumbnail((40, 40), PIL.Image.ANTIALIAS)
img = np.float32(img) / 255.0
# premultiply RGB by Alpha
img[..., :3] *= img[..., 3:]
# pad to self.h, self.h
diff = size - 40
img = np.pad(img, ((diff // 2, diff // 2), (diff // 2, diff // 2), (0, 0)))
return img
def load_emoji(emoji, size, code=None):
if code is None:
code = hex(ord(emoji))[2:].lower()
url = (
"https://github.com/googlefonts/noto-emoji/blob/main/png/128/emoji_u%s.png?raw=true"
% code
)
return load_image(url, size)
def make_circle_masks(n, h, w, r=None):
x = np.linspace(-1.0, 1.0, w)[None, None, :]
y = np.linspace(-1.0, 1.0, h)[None, :, None]
center = np.random.uniform(-0.5, 0.5, size=[2, n, 1, 1])
if r is None:
r = np.random.uniform(0.1, 0.4, size=[n, 1, 1])
x, y = (x - center[0]) / r, (y - center[1]) / r
mask = x * x + y * y < 1.0
return mask.astype(float)
|