cfzd / Ultra-Fast-Lane-Detection

Ultra Fast Structure-aware Deep Lane Detection (ECCV 2020)
MIT License
1.76k stars 491 forks source link

增强学习 #323

Open qiyuefun opened 1 year ago

qiyuefun commented 1 year ago

如果在已训练好的模型上,增加其他情景的数据,然后重新训练,目前工程版本是否支持,如果支持,应该如何修改

cfzd commented 1 year ago

@qiyuefun 可以把已训好的模型作为预训练参数载入进去然后在新的数据上进行迁移学习。你可以直接使用fintune部分的逻辑: https://github.com/cfzd/Ultra-Fast-Lane-Detection/blob/85eda710482eebe1a48f90316de7cfec7399bbf3/train.py#L122-L129

qiyuefun commented 1 year ago

好的,感谢。