ValueError: Expected tensor to be a tensor image of size (C, H, W). Got tensor.size() = torch.Size([8, 8])

vision
2

Can you try adding

landmarks.unsqueeze_(0)

before the line:

1 Like