Slike_slovenke_socialmediarip_vol.1.rar <Trusted ◆>

# Move model to GPU if available device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu") model.to(device) model.eval()

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) slike_SLOVENKE_socialMEDIArip_vol.1.rar

# Load pre-trained ResNet50 and remove the last layer model = resnet50(pretrained=True) model.fc = torch.nn.Identity() # Move model to GPU if available device = torch

# Define transformations for images transform = transforms.Compose([ transforms.Resize((224, 224)), transforms.ToTensor(), transforms.Normalize(mean=[0.485, 0.456, 0.406], std=[0.229, 0.224, 0.225]) ]) slike_SLOVENKE_socialMEDIArip_vol.1.rar