How to run inference on custom dataset?

[somccrt]

Hello,

First of all, this is an excellent work! thank you for sharing your work with us.
I was just tying to run it on a custom image dataset, for example, I have a set of 100 person images stored in a folder, now how to modify your inference code so that with a input text based query i can extract some matching images from those 100 images ?

Basically I want to do this:

1. choose one sentence as the query
2. use this query to retrieve person images
3. rank the images according to the similarities

Thanks