Problems with expansion

[mrfh12]

Hi, at first thanks for sharing your code, it worked like a charm!
I am using your work to use your work to track the process of contraction and expansion for various processes.
Tracking an object which contracts itself (e.g a balloon which loses its air) works perfectly!
However in contrast, trying to track the expansion, when filling it with air doesnt work as well. Only half of the object is captured at maximum expansion.
I already tried increasing the radius but the problem is, that it somehow selects features next to the object as most similar.
Do you have any idea how to circumvent this problem? (Training with smaller patches e.g?)
Thank you!

[b4shy]

I could solve this problem by increasing the size of the feature map.
Either try removing some strides or upscale the image