search

gengshan-y / VCN

Volumetric Correspondence Networks for Optical Flow, NeurIPS 2019.
https://gengshan-y.github.io/VCN/
MIT License
154 stars 25 forks source link

Correlation in PWC-net #17

Closed Guptajakala closed 4 years ago

Guptajakala commented 4 years ago

Hi, thanks for the great work! I have a question about your corr filter implementation in PWC-Net https://github.com/gengshan-y/VCN/blob/eb816cbc7887fd81fd80f36311e95eaddcb16f62/models/PWCNet.py#L191

The original PWC-Net repo has another cuda implementation, could you confirm if your implementation does the same thing as the implementation from PWC-Net original repo?

gengshan-y commented 4 years ago

Yea, they should give the same output. corr() might be slower though.