search

haofeixu / aanet

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

Train script for StereoNet feature extractor #66

Closed a-akram-98 closed 3 years ago

a-akram-98 commented 3 years ago

Can You please provide the script you used for training the StereoNet-based model and PSMNet model? thank you, @haofeixu

haofeixu commented 3 years ago

Hi @a-akram-98 , I can't find the exact training scripts but you can refer to AANet's training script (https://github.com/haofeixu/aanet/blob/master/scripts/aanet_train.sh) and modify the model configs according to https://github.com/haofeixu/aanet/blob/master/scripts/stereonet-aa_inference.sh and https://github.com/haofeixu/aanet/blob/master/scripts/psmnet-aa_inference.sh