Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Please provide ONNX or coreml model file for ctdet_coco_dla_2x.pth #493

Open
ShreshthSaxena opened this issue Nov 28, 2019 · 7 comments
Open
Labels
help wanted Extra attention is needed

Comments

@ShreshthSaxena
Copy link

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.

@xingyizhou
Copy link
Owner

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.

@ShreshthSaxena
Copy link
Author

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.

@xingyizhou
Copy link
Owner

Unfortunately, I haven't tested.

@xingyizhou xingyizhou added the help wanted Extra attention is needed label Dec 7, 2019
@CaoWGG
Copy link

CaoWGG commented Dec 22, 2019

@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.

@xingyizhou
Copy link
Owner

@CaoWGG Thanks a lot!

@JohnRen1
Copy link

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

@JohnRen1
Copy link

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

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
help wanted Extra attention is needed
Projects
None yet
Development

No branches or pull requests

4 participants