tf_superpoint quantization scripts

Xilinx / Vitis-AI

Vitis AI is Xilinx’s development stack for AI inference on Xilinx hardware platforms, including both edge devices and Alveo cards.

https://www.xilinx.com/ai

Apache License 2.0

1.5k stars 635 forks source link

tf_superpoint quantization scripts #1489

Closed CroosJJSE closed 3 weeks ago

CroosJJSE commented 3 weeks ago

hi, i am an 4th year undergraduate, involved in an project where i am implementing customized superpoint (pytorch) based SLAM on KRIA KR260, I am struggling to do quantizing and pruning using Vitis AI, I saw the tf_superpoint model on model zoo, if i can get that quantizing and pruning python scripts of tf_superpoint, it will be helpful for my project, Your help is so appreciable Thank you

jimheaton commented 3 weeks ago

I don't think this model was pruned, so we don't have scripts for that. The training and quantization code should be here: https://www.xilinx.com/bin/public/openDownload?filename=tf_superpoint_3.5.zip

CroosJJSE commented 3 weeks ago

thank you, i will check it