Pooling module as proposed in our NMMP

Pooling module as proposed in our paper

Balance between reconstruction task and prediction task.

Balance between reconstruction task and prediction task.

A Geometric Map is a int tensor of shape [layers, x, y]. The homography must transform a point in scene coordinates to the respective point i

r"""Warps image patches or tensors by homographies. .. math:: X_{dst} = H_{src}^{\{dst\}} * X_{src} Args: height (int): The he

Computes and stores the average and current value