Show the image and label of query data

[ivanwu0404]

Hi,
I want to use my own dataset on this model,
but i also want to show which the query data is running now[ex. image.show()].
did you have any idea to show the image and the label because i can't find the place that query data is processing now, thank you.

[zilunzhang]

Hi, I want to use my own dataset on this model, but i also want to show which the query data is running now[ex. image.show()]. did you have any idea to show the image and the label because i can't find the place that query data is processing now, thank you.

Please refer to line 91 of main.py