Open
Description
Hello! I tried to use your released p3d pytorch model. But i got a wrong result, every class got a low score under 0.01.
Is it my way to transform image into tensor result in this?
model = P3D199(pretrained=True)
model = model.eval().to(device)
transforms = Compose([Resize(160), CenterCrop(160), ToTensor(),
Normalize(mean=(0.485,0.456,0.406),std=(0.229, 0.224, 0.255),inplace=True)])
clip = [transforms(img) for img in clip] # 16 frames
clip = torch.stack(clip, 0).permute(1, 0, 2, 3).unsqueeze(0).to(device)
out=model(clip)
Thank you for your advice
Metadata
Assignees
Labels
No labels