Slike_slovenke_socialmediarip_vol.1.rar

# Load pre-trained ResNet50 and remove the last layer model = resnet50(pretrained=True) model.fc = torch.nn.Identity()

# Now 'features' is a list of feature vectors, you can convert it to a numpy array features = np.array(features) slike_SLOVENKE_socialMEDIArip_vol.1.rar

features = [] for filename in os.listdir(images_dir): img_path = os.path.join(images_dir, filename) image = Image.open(img_path) image = transform(image) image = image.unsqueeze(0).to(device) feature = model(image) feature = feature.detach().cpu().numpy().squeeze() features.append(feature) # Load pre-trained ResNet50 and remove the last

import torch import torchvision import torchvision.transforms as transforms from torchvision.models import resnet50 from PIL import Image import os import numpy as np slike_SLOVENKE_socialMEDIArip_vol.1.rar