search

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

fix image color #4

Closed kobot30 closed 3 years ago

kobot30 commented 3 years ago

Hi. Thank you for the Pytorch version of the code. I tried running demo.py, but the color of the result image was strange. I tried to fix it, so please check it if you like.

before after
frame_1_out_before frame_1_out_after
lhwcv commented 3 years ago

Good job! Thanks a lot!