It seems like the issue of converting Centernet models to ONNX is not yet resolved and I cannot figure out any other way to get a CoreML model of CenterNet for iOS application.
Please provide instructions on how to convert or better yet, provide the converted files if possible.
Thank you for your interest in our project! I also tried to export our model to onnx, the main issue is the DCN layers are not supported in onnx as far as I know. I will be happy to provide the onnx model when DCN is supported in onnx.
Is there any other way to get an mlmodel file of CentreNet?
I'm looking to run it on an IPhone. If you've already tested the network on mobile devices please let me know the results.
Unfortunately, I haven't tested.
@ShreshthSaxena
I've written the conversion process from ctdet_coco_dla_2x.pth to onnx model. You can refer to here.
https://github.com/CaoWGG/TensorRT-CenterNet is TensorRT conversion code for CenterNet.
@CaoWGG Thanks a lot!
Is there any other way to get an mlmodel file of CentreNet?
I'm looking to run it on an IPhone. If you've already tested the network on mobile devices please let me know the results.
hey guys,Can you share the swift/OC code about CenterNet
Is there any other way to get an mlmodel file of CentreNet?
I'm looking to run it on an IPhone. If you've already tested the network on mobile devices please let me know the results.
thanks a lot
Most helpful comment
@ShreshthSaxena
I've written the conversion process from ctdet_coco_dla_2x.pth to onnx model. You can refer to here.
https://github.com/CaoWGG/TensorRT-CenterNet is TensorRT conversion code for CenterNet.