Da — (3).mp4

# Add batch dimension tensor_frame = tensor_frame.unsqueeze(0)

# Transform to apply to frames transform = transforms.Compose([ transforms.ToTensor(), transforms.Normalize(mean=[0.485, 0.456, 0.406], std=[0.229, 0.224, 0.225]) ]) da (3).mp4

while True: ret, frame = video_capture.read() if not ret: break # Convert to RGB and apply transform rgb_frame = cv2.cvtColor(frame, cv2.COLOR_BGR2RGB) tensor_frame = transform(rgb_frame) # Add batch dimension tensor_frame = tensor_frame

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

# Display or save frame if needed # ...