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

How to merge the pred bboxs and labels into one output feature #23

Open
JPChen2000 opened this issue May 14, 2024 · 1 comment
Open

Comments

@JPChen2000
Copy link

1、deploy to onnx failed. cannot export pth to onnx.
05/14 15:10:10 - mmengine - WARNING - Failed to search registry with scope "mmyolo" in the "Codebases" registry tree. As a workaround, the current "Codebases" registry in "mmdeploy" is used to build instance. This may cause unexpected failure when running the built modules. Please check whether "mmyolo" is a correct scope, or whether the registry is initialized.
05/14 15:10:10 - mmengine - WARNING - Failed to search registry with scope "mmyolo" in the "backend_detectors" registry tree. As a workaround, the current "backend_detectors" registry in "mmdeploy" is used to build instance. This may cause unexpected failure when running the built modules. Please check whether "mmyolo" is a correct scope, or whether the registry is initialized.
The given version [15] is not supported, only version 1 to 14 is supported in this build.

2、How to merge the pred bboxs and labels into one output feature.
I couldn't find a way to modify the output nodes in this project. I don't know how the model is constructed. Does mmengine provide an API that allows modifying nodes during the model export to ONNX?

By the way, using mmengine seems has made this project less user-friendly.

@FishAndWasabi
Copy link
Owner

Sorry for the delay.

  1. According to the error message, the mmyolo may not be installed. We suggest reinstalling mmyolo.
  2. To merge the two outputs, we think the mmdepoly may not provide API to do this, and you can manage the output in the code of the head. The appendix is a simple deploy code (without postprocess) with only ONNX. Hope it can help.

The code of mmlab may be complicate. We will try to transfer our code into ultralytics.

Thanks for your interest in our work!
Best Wishes! 😊

simple_deploy.zip

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants