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

Update common.py #12985

Open
wants to merge 10 commits into
base: master
Choose a base branch
from
Open

Update common.py #12985

wants to merge 10 commits into from

Conversation

NeKoooo233
Copy link

@NeKoooo233 NeKoooo233 commented May 5, 2024

Fixed a bug. When using a higher version of tensorrt and val.py to verify the model, the following problems may occur: getPluginCreator could not find plugin: InstanceNormalization_TRT version: 1
Serialization assertion creator failed.Cannot deserialize plugin since corresponding IPluginCreator not found in Plugin Registry
in common.py context = model.create_execution_context()
AttributeError: 'NoneType' object has no attribute 'create_execution_context'

I have read the CLA Document and I sign the CLA

🛠️ PR Summary

Made with ❤️ by Ultralytics Actions

🌟 Summary

Enhanced YOLOv5 model loading support for NVIDIA's TensorRT.

📊 Key Changes

  • Added initialization for TensorRT plugins within the model loading process.

🎯 Purpose & Impact

  • Purpose: This change ensures that all necessary TensorRT plugins are initialized when a model is loaded, which is crucial for compatibility and performance when using NVIDIA's optimization tools.
  • Impact: Users leveraging YOLOv5 with NVIDIA TensorRT for inference can expect smoother deployments and potentially improved performance, as all required plugins are correctly initialized. This update is especially beneficial for those working in environments where advanced optimizations are necessary for speed and efficiency. 🚀

Fixed a bug. When using a higher version of tensorrt and val.py to verify the model, the following problems may occur:
getPluginCreator could not find plugin: InstanceNormalization_TRT version: 1
Serialization assertion creator failed.Cannot deserialize plugin since corresponding IPluginCreator not found in Plugin Registry

Signed-off-by: Melt <60842553+NeKoooo233@users.noreply.github.com>
Copy link
Contributor

github-actions bot commented May 5, 2024


Thank you for your submission, we really appreciate it. Like many open-source projects, we ask that you all sign our Contributor License Agreement before we can accept your contribution. You can sign the CLA by just posting a Pull Request Comment same as the below format.


I have read the CLA Document and I sign the CLA


2 out of 3 committers have signed the CLA.
✅ (glenn-jocher)[https://github.com/glenn-jocher]
✅ (UltralyticsAssistant)[https://github.com/UltralyticsAssistant]
@NeKoooo233
You can retrigger this bot by commenting recheck in this Pull Request. Posted by the CLA Assistant Lite bot.

Copy link
Contributor

@github-actions github-actions bot left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

👋 Hello @NeKoooo233, thank you for submitting a YOLOv5 🚀 PR! To allow your work to be integrated as seamlessly as possible, we advise you to:

  • ✅ Verify your PR is up-to-date with ultralytics/yolov5 master branch. If your PR is behind you can update your code by clicking the 'Update branch' button or by running git pull and git merge master locally.
  • ✅ Verify all YOLOv5 Continuous Integration (CI) checks are passing.
  • ✅ Reduce changes to the absolute minimum required for your bug fix or feature addition. "It is not daily increase but daily decrease, hack away the unessential. The closer to the source, the less wastage there is." — Bruce Lee

@glenn-jocher
Copy link
Member

@NeKoooo233 please accept the CLA by writing a new issue with:

I have read the CLA Document and I sign the CLA

@NeKoooo233
Copy link
Author

@NeKoooo233 please accept the CLA by writing a new issue with:

I have read the CLA Document and I sign the CLA

OK

@NeKoooo233
Copy link
Author

I have read the CLA Document and I sign the CLA.

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

Successfully merging this pull request may close these issues.

None yet

3 participants