How train the model with gray image

lhwcv / mlsd_pytorch

Pytorch implementation of "M-LSD: Towards Light-weight and Real-time Line Segment Detection"

Apache License 2.0

189 stars 37 forks source link

Open Chrisspf opened 8 months ago

Chrisspf commented 8 months ago

Hi, thank you for your code, and I want to know how to train the model with gray image. Does the loss function need to be adjusted?thank you!