Viewing a single comment thread. View all comments

sudhanv99 t1_iwyhvfj wrote

it is doable but you have to know what images are in your folder. you can find models trained on coco dataset which can do this. but your images have to be in the coco dataset.

you can write a script which will move your files based on the predictions after that.

1

PegasusInvasion OP t1_iwyj8hb wrote

I was trying to avoid having to train on a coco dataset 🫤 Anyway, thanks for your help!

1

sudhanv99 t1_iwyjd7e wrote

you dont have to train. there are pretrained models, you can just run inference.

4

PegasusInvasion OP t1_iwyjhlh wrote

Sorry I misunderstood what you meant by coco dataset. I get it now. Do you know any good models that I can try?

1