search

Alibaba-MIIL / TResNet

Official Pytorch Implementation of "TResNet: High-Performance GPU-Dedicated Architecture" (WACV 2021)
Apache License 2.0
471 stars 63 forks source link

Standford Car Test #42

Closed Janspiry closed 2 years ago

Janspiry commented 2 years ago

Hi，thanks for this great works. I used the pretrained model to test on Standford Car dataset, but get the same results in most images. I used the transform function on below:

    transforms.Resize(int(368 / 1)),
    transforms.CenterCrop(368),
    transforms.ToTensor()

and read the image by:

img = Image.open(img_path).convert("RGB")
img = self.tfs(img)

Is there anything I need to pay extra attention to？

mrT23 commented 2 years ago

the "transforms.CenterCrop(368)" is wrong, remove it (squish resizing)

other than that, triple check to make sure you are loading the model correctly