search

cherubicXN / hawp

Holistically-Attracted Wireframe Parsing [TPAMI'23] & [CVPR' 20]
MIT License
298 stars 52 forks source link

Export hawp to onnx #34

Open TanyaChutani opened 2 years ago

TanyaChutani commented 2 years ago

Export hawp to onnx

cherubicXN commented 2 years ago

Thanks for your contribution for the exporting of onnx

AndyJado commented 10 months ago

apart from repo structure changes, I tried a onnx export simply adding few lines in ssl/predict.py and returned a hawp.onnx size in 306Byte