gautam678 / Pix2Depth

DEPRECATED: Depth Map Estimation from Monocular Images
GNU General Public License v3.0
354 stars 83 forks source link

Where should I download these models? #22

Closed YuehWu1994 closed 5 years ago

YuehWu1994 commented 6 years ago

Hello, thanks for developing such a amazing project. I have a question that where I should download these models in main.py. You've provided links to search for 'pix2pix', 'CycleGAN' and 'CNN', but I still can't find the corresponding name in either of these links.

Thanks,

model_list = {  
        'pix2depth':{ 
            'pix2pix' : load_model('weights/p2d_pix2pix.h5'),
            'CycleGAN':load_model('weights/p2d_cycle.h5'),
            'CNN': load_model('weights/p2d_cnn.h5'),
            },
        'depth2pix':{ 
            'pix2pix' : load_model('weights/d2p_pix2pix.h5'),
            'CycleGAN':load_model('weights/d2p_cycle.h5'),
            }
         }
shreyashk09 commented 6 years ago

where can I find model to 'pix2pix_depth.h5' weights?

gautam678 commented 5 years ago

Hey! the weights are corrupt and I didn't have a backup. I'm planning on retraining the models this winter and publish it here. I'll keep this issue open till I upload the models back into the repo

gautam678 commented 5 years ago

I have the models trained for pix2pix

https://drive.google.com/drive/u/1/folders/1DzFIh9MkS-CjZYn1Mqcw1qt_0Jk9tSNe