Help on visualizing Correlation pyramid

princeton-vl / RAFT-Stereo

MIT License

721 stars 139 forks source link

Help on visualizing Correlation pyramid #63

Open Mythili-kannan opened 2 years ago

Mythili-kannan commented 2 years ago

Hi can you help me on how can i obtain correlation pyramid features and visualize them, so that i can see the common region between two images

lahavlipson commented 2 years ago

In the forward pass, fmap1, fmap2 contain the correlation features for the left and right images, respectively. You can visualization the magnitude of their outer-product in matplotlib to see the similar regions.