Non-iterative Deocders
Soft Regression
- class ezflow.decoder.noniterative.soft_regression.Soft4DFlowRegression(size, max_disp=4, entropy=False, factorization=1)[source]
Applies 4D soft argmax operation to regress flow.
- Parameters
size (List[int]) – List containing values of B, H, W
max_disp (int, default : 4) – Maximum displacement
entropy (bool, default : False) – If True, computes local and global entropy from matching cost
factorization (int, default : 1) – Max displacement factorization value
- class ezflow.decoder.noniterative.soft_regression.SoftArg2DFlowRegression(max_u=3, max_v=3, operation='argmax')[source]
Applies 2D soft argmin/argmax operation to regress flow. Used in DICL (https://arxiv.org/abs/2010.14851)
- Parameters
max_u (int, default : 3) – Maximum displacement in the horizontal direction
max_v (int, default : 3) – Maximum displacement in the vertical direction
operation (str, default : argmax) – The argmax/argmin operation for flow regression