search

knazeri / edge-connect

EdgeConnect: Structure Guided Image Inpainting using Edge Prediction, ICCV 2019 https://arxiv.org/abs/1901.00212
http://openaccess.thecvf.com/content_ICCVW_2019/html/AIM/Nazeri_EdgeConnect_Structure_Guided_Image_Inpainting_using_Edge_Prediction_ICCVW_2019_paper.html
Other
2.5k stars 528 forks source link

Export models as `ONNX` files #193

Open Luni-4 opened 11 months ago

Luni-4 commented 11 months ago

This PR exports models as ONNX files.

It it possible to optimize those models using https://pypi.org/project/onnxoptimizer/ as:

onnxoptimizer input-model.onnx optimized-model.onnx