| import torch | |
| import img2gray | |
| from PIL import Image | |
| import numpy as np | |
| print(dir(img2gray)) | |
| img = Image.open("/home/ubuntu/Projects/img2gray/kernel-builder-logo-color.png").convert("RGB") | |
| img = np.array(img) | |
| img_tensor = torch.from_numpy(img) | |
| print(img_tensor.shape) # HWC | |
| img_tensor = img_tensor.permute(2, 0, 1).unsqueeze(0).contiguous().cuda() # BCHW | |
| print(img_tensor.shape) # BCHW | |
| gray_tensor = img2gray.img2gray(img_tensor).squeeze() | |
| print(gray_tensor.shape) # B1HW | |
| # save the output image | |
| gray_img = gray_tensor.cpu().numpy() # 1HW -> HW | |
| gray_img = Image.fromarray(gray_img.astype(np.uint8), mode="L") | |
| gray_img.save("/home/ubuntu/Projects/img2gray/kernel-builder-logo-gray.png") |