Zero Shot Classification on my own Dataset

[deadpipe]

Hello,

I am trying to fine tune CLIP on my own dataset for Zero Shot Classification.
My question is - is there a way to load a CSV containing all the file paths and their corresponding labels? OR a Folder which contains all the images in subfolders?

[gabrielilharco]

Hi @deadpipe. The easiest way to do this in this codebase is the last option you mentioned. You'll need to write a dataset class for this, see https://github.com/mlfoundations/wise-ft/blob/master/src/datasets/imagenet.py#L8 for an example

[deadpipe]

So i have to create a new .py file which consists of my class or i should write another class within that file?