The problem encountered during training with my own dataset.
Closed this issue · 2 comments
zza1617 commented
May I ask how to fine-tune a pre-trained model using my own dataset, which includes three subfolders under a main folder, one for left views, one for right views, and another one for disparity maps? I'm not sure how to proceed with the training process, so I would appreciate any help. Thank you very much.
lahavlipson commented
You'll need to create a new subclass of StereoDataset
in stereo_datasets.py, similar to what is done for Middlebury, KITTI, etc. Also under def fetch_dataloader(args):
you will need to add a case for your particular dataset.
zza1617 commented
Thanks