Test the trained model on new data

[AlxJude]

Hello,

I was trying to test the trained model with new data. Could you please help me with the pipeline involved in doing so?
I ran the generate_mask.py with the test data and then ran test.py, but I was getting a size mismatch error.

Kindly help. Thanks!