training based on bdd100k?

InhwanBae / ENet-SAD_Pytorch

Pytorch implementation of "Learning Lightweight Lane Detection CNNs by Self Attention Distillation (ICCV 2019)"

MIT License

76 stars 18 forks source link

training based on bdd100k? #18

Closed huoguangdiandian closed 2 years ago

huoguangdiandian commented 2 years ago

Hi, I want to trian the net based on bdd100k, What should I pay attention to? For example, setting hyperparameters or Is there anything that needs to be changed in train.py. Looking forward to your reply

InhwanBae commented 2 years ago

Hi @huoguangdiandian ! You don't need to change train.py. The only thing you need to do is to modify cfg.json.

"dataset_name": "BDD100K",
"resize_shape": [640, 360]

See here for more detail.

InhwanBae commented 2 years ago

I'm closing this issue for now. Please feel free to open another issue for other questions.