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

wrong many bounding box and wrong predict untrained class id #594

Open
MehmetOKUYAR opened this issue Feb 9, 2023 · 2 comments
Open

wrong many bounding box and wrong predict untrained class id #594

MehmetOKUYAR opened this issue Feb 9, 2023 · 2 comments

Comments

@MehmetOKUYAR
Copy link

MehmetOKUYAR commented Feb 9, 2023

Camera TensorRT YOLO Demo_screenshot_09 02 2023

It's predicting the class id that I haven't trained, and it's predicting a lot of random bounding grays. I didn't know where there was an error. It works fine when I try it on another pc.

weights work fine if i don't do tensorrt transform with opencv darknet library

@xiaoyaoguai
Copy link

Camera TensorRT YOLO Demo_screenshot_09 02 2023

It's predicting the class id that I haven't trained, and it's predicting a lot of random bounding grays. I didn't know where there was an error. It works fine when I try it on another pc.

weights work fine if i don't do tensorrt transform with opencv darknet library

You can try modifying the py file of class_id.
I encountered a similar issue on Nano, where TensorRT's inference was entirely incorrect, and the generated Tiny model was 32mb (compared to the correct model generated on other devices, which is only 13mb). Notably, Jetpack 4.6 comes pre-configured with Python 3.6 and Protoc 3.0.0, and upgrading Protoc from 3.0.0 to 3.8 renders ONNX 1.9 unimportable. I attempted upgrading Protoc after importing ONNX, but this still did not resolve the issue.

@xiaoyaoguai
Copy link

try this:rebuild the "yolo_layer.h" in plugin,
line70: bool supportsFormatCombination(int pos, const PluginTensorDesc* inOut, int nbInputs, int nbOutputs) const NOEXCEPT override { return inOut[pos].format == PluginFormat::kLINEAR && inOut[pos].type == DataType::kFLOAT; }

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