search

CVLAB-Unibo / Real-time-self-adaptive-deep-stereo

Code for "Real-time self-adaptive deep stereo" - CVPR 2019 (ORAL)
Apache License 2.0
419 stars 73 forks source link

how to integrate other network to this self adaptive framework? #49

Closed passion3394 closed 4 years ago

passion3394 commented 4 years ago

such as PSMNet, GANet?

AlessioTonioni commented 4 years ago

You should implement the network and wrap it in a class that inerhits from [Stereo_net.py].(https://github.com/CVLAB-Unibo/Real-time-self-adaptive-deep-stereo/blob/master/Nets/Stereo_net.py)

Something similar to what has been done for MadNet in MadNet.py and Dispnet in DispNet.py

Finally add the network class to the factory method in Nets/__init__.py

passion3394 commented 4 years ago

thanks for your reply.