Get an image into feature vector

image-into-clip-feature-vector.py

import clip
import torch
from PIL import Image

#image_pil = Image.open("path_to_image.jpg")
image_pil = Image.fromarray(image)
device = 'cuda'
image_input = preprocess(image_pil).unsqueeze(0).to(device)
with torch.no_grad():
     image_features = model.encode_image(image_input)

image-into-feature-vector.py

import cv2
from torchvision.models import resnet50
from torchvision.transforms import functional as TF

model = resnet50(pretrained=True)
image = cv2.imread('shaolin-monk-test.jpg')
input_tensor = TF.to_tensor(image).unsqueeze(0)
vector  = model(input_tensor)