search

TRI-ML / packnet-sfm

TRI-ML Monocular Depth Estimation Repository
https://tri-ml.github.io/packnet-sfm/
MIT License
1.21k stars 241 forks source link

Custom data set training #223

Closed luda1013 closed 2 years ago

luda1013 commented 2 years ago

Hallo, finaly i can get to this point where i can train with kitty tiny dataset and save the model. But once i want to train my own dataset, i have this error saying that ZeroDivisionError: division by zero

Btw i dont use the docker, had no success with it. So i installed on computer

The dataset that i use in config yaml is dataset: ['Image']

Thank you very much for your help!

ArminBaz commented 2 years ago

Hey have you figured out this issue? If not you might need to employ the same solution done in this issue.

luda1013 commented 2 years ago

Hey have you figured out this issue? If not you might need to employ the same solution done in this issue.

Hallo ArminBaz, thanks for replying, yeah i already figured it out. with the split like mentioned in that issue. But i must tweak some lines in the packnet_sfm\datasets\image_dataset.py