File size: 490 Bytes
846d481 aca891a 846d481 aca891a 846d481 aca891a 846d481 aca891a |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 |
import torch
import img2gray
from PIL import Image
import numpy as np
print(dir(img2gray))
img = Image.open("kernel-builder-logo-color.png").convert("RGB")
img = np.array(img)
img_tensor = torch.from_numpy(img).cuda()
print(img_tensor.shape) # HWC
gray_tensor = img2gray.img2gray(img_tensor).squeeze()
print(gray_tensor.shape) # HW
# save the output image
gray_img = Image.fromarray(gray_tensor.cpu().numpy().astype(np.uint8), mode="L")
gray_img.save("kernel-builder-logo-gray.png")
|