How to inference with any images?

[snowmint]

I want to use this model to do inference with my own photo (just inference to get a prediction, no need to calculate accuracy), I've tried to modify test.py and try to make my own MYDatasets.py but I don't know how to get it correct.

Have anyone written the inference function? Or have any advice about how to modify test.py to inference use?