miccunifi/ARNIQA

Minimal example does not work

Chuudy opened this issue · 1 comments

Chuudy commented

I've been trying to run the minimal example, however I got following error:

TypeError: img should be Tensor Image. Got <class 'PIL.Image.Image'>

when trying to execute this line:

img = preprocess(img).unsqueeze(0).to(device)

LorenzoAgnolucci commented

Hi @Chuudy!

You are right, I accidentally inverted the two transformations in the preprocessing pipeline. Try again now and let me know.