princeton-vl / RAFT

BSD 3-Clause "New" or "Revised" License
3.23k stars 629 forks source link

Code for Correlation Range of 128 pixels. #137

Open fupiao1998 opened 2 years ago

fupiao1998 commented 2 years ago

Hello, thank you very much for such a valuable work, after two years, it is still so excellent. In the table2 of the paper, when the Correlation Range is 128px, the same result as All-Pairs can be obtained. Could you please provide the code for this part, I would like to do some tests for this part of the experiment, thank you very much and look forward to your reply.

fupiao1998 commented 2 years ago

I try to implement it by https://github.com/ClementPinard/Pytorch-Correlation-extension, and get a local correlation matrix. Next I send it to correlation pooling and lookup module. But doing so directly causes the network to fail to converge. Therefore, I look forward to your open source related code. thank you very much