How to cast fastai Image to a Pytorch Tensor?

Part 1 (2019) UTC+5 India
1

Hello everyone!

I am trying to extract features of an image from ResNet50 model. Kindly refer the image below

compress
compress1260×916 127 KB

I have removed the top layer of classification and the ReLU, BatchNorm1D and Dropout layers just preceding that so that I can obtain a 512 length feature vector for my images.

But when I do that and try to make a forward pass, I need to convert the datatype of the image from a fastai.vision.image.Image to a tensor. Otherwise I cannot make the forward pass as seen in the screenshot below.

convert_from_image_to_tensor
convert_from_image_to_tensor1229×925 201 KB

Can someone please explain a way to cast a fastai Image to a PyTorch tensor?

Thanks for your time.

With Regards,
Vinayak.

2 Likes
2

try x.px or x.data

1 Like
3

Thanks Julian! It worked…