Viewing a single comment thread. View all comments

PegasusInvasion OP t1_iwyhe5f wrote

Reply to comment by smsorin in [P] Any object detection library by PegasusInvasion

Thanks for the explanation.

I actually wanted to use a machine learning model to sort my pictures folder based on the objects in the image.

For example, I have 10 images of cars and 5 images of buildings. I wanted to move images of cars and buildings into separate folders.

Do you have any idea, how I can achieve that?

2

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