Training customer datasets

[Jackson-zhao97]

Hi,
I am interested in your work, but the weight files given are mainly used for road scenarios.
I want to retrain with other datasets, how should I construct the data? What is the exact process to train a customer dataset?
Looking forward to your reply, thanks!

[Lavreniuk]

Hi, @Jackson-zhao97 , I am happy that you find my model interesting.
At first I'd suggest to check how the data for cityscapes, or kitti are structured. I think it is better to check kitti, it is commonly used dataset, and easy structured. After that you will have 2 options, structure your custom data in same way as kitti (I'd suggest this one), so you don't need to change code, or use custom data structure, but write your own DataLoader class.

[Jackson-zhao97]

@Lavreniuk Thanks for your quick reply, I will try the structure of the kitti dataset first.