KDTree indexing before applying average pooling

[RahulSajnani]

I see that in the demo you do not perform a kdtree pooling. This will result in improperly extracted TFN features and will not give optimal performance. You need to perform a kdtree pooling here.

[vanhalen42]

Hey, thanks for this! This was a known issue, but I might have missed it in the demo code. I will update the code and the checkpoints soon.