JiaRenChang/PSMNet

Health v2 · maintenance only

Flags: no_releases

Adoption not part of the score

1548 stars · 424 forks observed · 2026-08-28

What it is AI-extracted, prompt v1, taxonomy v1, 2026-08-30, confidence not recorded

PSMNet is the official PyTorch implementation of the CVPR 2018 paper 'Pyramid Stereo Matching Network' for stereo depth estimation. It uses spatial pyramid pooling and a 3D CNN with stacked hourglass networks to compute disparity maps from stereo image pairs.

Use cases

• estimate depth from stereo image pairs
• compute disparity maps with a CNN
• train a stereo matching model on Scene Flow
• evaluate stereo models on KITTI
• reproduce PSMNet research results
• benchmark stereo matching networks

When to choose

• you need a well-cited baseline for stereo depth estimation research
• you want to train or evaluate stereo matching on KITTI or Scene Flow
• you need a PyTorch reference implementation of PSMNet

When to avoid

• you need real-time stereo matching on embedded hardware
• you want a maintained production-ready library with recent updates
• you work with monocular (single-camera) depth estimation

Facets

library · maturity maintenance

machine-learning computer-vision deep-learning computer-vision deep-learning machine-learning python stereo-matching stereo-vision depth-estimation pytorch disparity cvpr-2018 research-code linux gpu

Member repositories

For agents

markdown · JSON · MCP: product_card(name="JiaRenChang/PSMNet")