Skip to content

Da (3).mp4 Direct

# Load a pre-trained model model = torchvision.models.resnet50(pretrained=True) model.eval() # Set to evaluation mode

# Add batch dimension tensor_frame = tensor_frame.unsqueeze(0) da (3).mp4

# Move to GPU if available device = torch.device('cuda' if torch.cuda.is_available() else 'cpu') tensor_frame = tensor_frame.to(device) model.to(device) # Load a pre-trained model model = torchvision

Volver arriba