search

AaronCIH / APGCC

ECCV24 - Improving Point-based Crowd Counting and Localization Based on Auxiliary Point Guidance
https://apgcc.github.io/
MIT License
51 stars 9 forks source link

APGCC ONNX Conversion #11

Closed junhui-ng closed 2 months ago

junhui-ng commented 2 months ago

Nice work! I would like to generate a TensorRT engine for this model, but I encountered an issue when converting the model to ONNX. Could you please provide the ONNX conversion script? Thanks!

junhui-ng commented 2 months ago

I managed to convert the model to ONNX.

For those interested, you can find the ONNX model and a sample inference script here: https://drive.google.com/drive/folders/1sqICjpxYajFj1_2tSfyCwz55unag6vCN?usp=sharing