normantud/ailocate

Create an index of all your image files, get what's on them with yolo and allow you to do `ailocate image cat`

First, make sure the predict.py runs properly. Make sure torch is installed. It worked for me with torch==1.11.0+cu102.

Then, create an index with:

bash create_index.sh

This will look at all jpg, png, jpeg and gif files and try to find what is depicted on them.

Then you can do:

./ailocate sheep

(maybe you need to chmod +x ./ailocate.)

and you get

/usr/share/tuxpaint/templates/sheep.jpg -> sheep
/usr/share/scratch/Media/Backgrounds/Outdoors/hay_field.jpg -> sheep

This may help someone someday