Training Custom dataset

Beniko95J / MLF-VO

Multi-Layer Fusion Visual Odometry

GNU General Public License v3.0

20 stars 3 forks source link

Training Custom dataset #2

Open arkinrc opened 2 years ago

arkinrc commented 2 years ago

@Beniko95J Hello, I would like to appreciate your efforts. I have successfully implemented pretrained model with the kitti dataset but when applied with my own dataset, there is huge difference in results. So i would like to know regarding the custom dataset training? do you have any plan to opensource the training file? my results are given below.

combine_images

looking forward to your feedback.... regards arkin

Beniko95J commented 1 year ago

Sorry for the late reply. I have uploaded the files for training in the dev branch although they still looks messy... For reference, the codes for training are mainly based on the Monodepth2, and we mainly change the architecture of PoseNet and the calculation of training loss. I will find some time to clean up the codes in the dev branch for easy use.