cvg / DeepLSD

Implementation of the paper "DeepLSD: Line Segment Detection and Refinement with Deep Image Gradients"
MIT License
486 stars 62 forks source link

export the DeepLSD model to ONNX #42

Closed rangingwaves closed 1 month ago

rangingwaves commented 1 month ago

How can I export the DeepLSD model to ONNX? Can you teach me?

rpautrat commented 1 month ago

Hi, I never did it, but the network is pretty simple, so you should be able to follow any good tutorial to export to ONNX.