Skip to content

About Image transform #31

Open
Open
@wanxinzzz

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

No one assigned

    Labels

    No labels
    No labels

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions