`AttributeError: 'JpegImageFile' object has no attribute 'shape'`

I was trying the code with the random image but I need to print the image I am passing nottouch.rand

my_image = load_img("127074.jpg")
v = Vit()
img = torch.randn(1, 3, 256, 256)
preds = v(img)
print(preds)
p=v(my_image)
print(p)

the code worked with using random but when i pass my_image i got AttributeError: 'JpegImageFile' object has no attribute 'shape'

[lessw2020]

Hi @ersamo Yes exactly. In the above, from the error…your image is on gpu, your model is on cpu. To fix model.to(pred_device)

Then both are on gpu and you should be cruising.

[lessw2020]

dinner was delayed…so here you go:

import torch
import torchvision
from PIL import Image
from torch import Tensor
import torchvision.transforms as T


inference_transform = T.Compose(
    [
        T.Resize(256),
        T.ToTensor(),
        T.Normalize([0.485, 0.456, 0.406], [0.229, 0.224, 0.225]),  # Imagenet
    ]
)

pred_device=torch.device('cuda' if torch.cuda.is_available() else 'cpu')

# load image, prep for inference and run it
orig_img = Image.open("myimage.jpg")

img_tensor = inference_transform(orig_img)

img_tensor = img_tensor.unsqueeze(0)

tensor_on_device = img_tensor.to(pred_device)

output = model(tensor_on_device)