`demo.py` img_size hardcoded to incorrect value

[ThereforeGames]

Hi,

Thank you for sharing your work!

I noticed in demo.py, the get_transform() method does not respect the size argument. It is hardcoded to resize to (224,224):

https://github.com/fawnliu/TRIS/blob/b45f660783b0334f6110c3afe7d53c8b30eab418/demo.py#L22

I believe this should be (size,size), which will take 320px by default. I can confirm that this change results in a heatmap that is a lot closer to the example in your readme.

---

Before fix

After fix

I'm still not sure why my heatmap doesn't match your example 100%, but the results are impressive nonetheless. 🙂