Resample2D function to CPU compatible code

NVIDIA / flownet2-pytorch

Pytorch implementation of FlowNet 2.0: Evolution of Optical Flow Estimation with Deep Networks

Other

3.09k stars 739 forks source link

Resample2D function to CPU compatible code #190

Open ShrutheeshIR opened 4 years ago

ShrutheeshIR commented 4 years ago

What do the resample2d, correlation and channel_norm cuda codes do. Is it possible to write CPU equivalent code, so that inference can be run on CPU devices? Thanks

fa9r commented 4 years ago

Yeah I'd also be interested in that.