search

haofeixu / aanet

[CVPR'20] AANet: Adaptive Aggregation Network for Efficient Stereo Matching
Apache License 2.0
521 stars 100 forks source link

how to fine-tuned model to predict the disparity map, no additional training is done on Middlebury #48

Closed lawo123 closed 3 years ago

lawo123 commented 3 years ago

how to fine-tuned model to predict the disparity map, no additional training is done on Middlebury ? how to show image have color like below? image

haofeixu commented 3 years ago

You can use the prediction script to generate the disparity map on any image pair: https://github.com/haofeixu/aanet/blob/master/scripts/aanet_predict.sh

The disparity map is visualizaed with plasma colormap, which can be implemented with matplotlib.