search

cardwing / Codes-for-Lane-Detection

Learning Lightweight Lane Detection CNNs by Self Attention Distillation (ICCV 2019)
MIT License
1.05k stars 333 forks source link

About the ERFNet trained model #263

Closed zy1817104 closed 4 years ago

zy1817104 commented 4 years ago

the ERFNet trained model is invalid Please provide another Thanks!!!

cardwing commented 4 years ago

It runs well on my local server. What error do you get when loading the trained model?

zy1817104 commented 4 years ago

Thanks for your reply.I have solved this problem.And I have 2 questions.

  1. what does the parameter(npb) mean in test_erfnet.sh?
  2. what does 'from models import sync_bn' mean in test_erfnet.py? Thanks!
cardwing commented 4 years ago

The parameter is to control whether using sync bn. The original code is based on PyTorch 0.3.0 (not support sync bn) and we have to write the sync bn module by ourselves and import it. You can find the sync bn module in 8a04dd4.