Mass prediction

Hi, I am currently working on my first classification task. However, I can’t seem to figure out how to predict on new, unlabeled images in a folder - only on single images. I have been stuck for days with this. Ideally, I want to use my trained model to classify all images in a folder, and then write the predictions to a csv file (as image-name, label). Any hints, ideas?

Thanks!!

you can either get a list of files as a list using os.listdir() and create a dataset

1 Like