Training Custom Dataset

[rak7045]

Hello @IrohXu ,
Is this repo supports to train the network on custom dataset? If not so, do you have any idea what parameters should change while training on custom dataset? What I should keep in my mind while using custom dataset?

Thanks and Kind Regards
Ravi Teja

[IrohXu]

It can support training on other public dataset or custom dataset. Please change some hyper-parameters here: https://github.com/IrohXu/lanenet-lane-detection-pytorch/blob/main/model/utils/cli_helper.py

Also, you need define the dataloader. Example is here:
https://github.com/IrohXu/lanenet-lane-detection-pytorch/blob/main/dataloader/data_loaders.py

Note: the input image should be RGB images, but you can also change the input channel of the network by yourself.