Classify a single image

[VivianaFigueira]

Hello :)
How could I predict the label of a single new image after the SwAV training and linear evaluation ?
Thank you!

[ayulockin]

Hey @VivianaFigueira, you will take the backbone model you pretrained using SwAV and do something like model.predict(image). This should give you a prediction on a single image.

However, the data (image) should be preprocessed as per the model's requirement.

[VivianaFigueira]

Thank you so much :)