Create a Pytorch Tensor of an image

createimagetensor.py

import torch
import numpy as np
import cv2
import torchvision.transforms as transforms
from PIL import Image 

# Read the image with openCV
# image = cv2.imread('sachadee.jpg')

# Convert BGR image to RGB image
#image = cv2.cvtColor(image, cv2.COLOR_BGR2R

#or with PIL
image = Image.open('sachadee.jpg')

GB)
  
# Define a transform to convert
# the image to torch tensor
transform = transforms.Compose([
    transforms.ToTensor()
])
  
# Convert the image to Torch tensor
tensor = transform(image)
  
# print the converted image tensor
print(tensor)