search

manurare / 360monodepth

Code release for 360monodepth. With our framework we achieve monocular depth estimation for high resolution 360° images based on aligning and blending perspective depth maps.
https://manurare.github.io/360monodepth/
MIT License
159 stars 33 forks source link

Fine-tune/train model on custom dataset #9

Closed HieuPhan33 closed 2 years ago

HieuPhan33 commented 2 years ago

Hi, I want to fine-tune or train the model on the custom dataset.

Could you please give some guidance?

Thanks, Hieu

manurare commented 2 years ago

Hello,

Our method has not been train in any dataset. We use pre-trained MiDaS as perspective monocular depth estimator. If you want, you could fine-tune MiDaS in your own dataset. Make sure though that MiDaS is designed for perspective images so, if you have 360 images to train you would need to extract perspective images from those.