search

antocad / FocusOnDepth

A Monocular depth-estimation for in-the-wild AutoFocus application.
MIT License
139 stars 31 forks source link

Training using SUNRGBD dataset #13

Open sreejithpai opened 2 years ago

sreejithpai commented 2 years ago

Hi, Thank you very much for the training script. I am currenlty trying to train using SUNRGBD dataset for both depth prediction and semantic segmentation. I have added multiple classes to config file ( chairs, floor, tables ). While training, program stopped execution due to nan values after 3 epochs. I am using 'vit_base', batch size 8, and default values for all other parameters. I have following doubts.

  1. Should I update/modify any code sections to train for multiple classes segmentation?
  2. I have passes ground truth depth values as such. Should I convert it to meters ?

Thanks in advance...

realTravisYou commented 4 months ago

Hi, Thank you very much for the training script. I am currenlty trying to train using SUNRGBD dataset for both depth prediction and semantic segmentation. I have added multiple classes to config file ( chairs, floor, tables ). While training, program stopped execution due to nan values after 3 epochs. I am using 'vit_base', batch size 8, and default values for all other parameters. I have following doubts.

  1. Should I update/modify any code sections to train for multiple classes segmentation?
  2. I have passes ground truth depth values as such. Should I convert it to meters ?

Thanks in advance...

Hello, have you successfully completed your work? Can you tell me how to do it, thank you very much.