Fix FP32 TensorRT model export

[SahilChachra]

Problem :- While exporting model as TensorRT, the code was using FP16 precision even without using "--half" flag with the export command. Hence, was unable to export FP32 precision models with the export command.

Fix :- Passed "half" variable to the if condition which means we will get FP16 model only if user has passed the "half" flag and the hardware supports FP16. Also modified the logger code to print appropriate precision being used. Just adding "half" variable, the bug has been resolved. Thanks 😄

Have tested the code for FP16 and FP32 both. It's working.

Attaching screenshots.

After fixing the code, we get,

🛠️ PR Summary

_{Made with ❤️ by Ultralytics Actions}

🌟 Summary

Improved TensorRT export flexibility regarding precision modes.

📊 Key Changes

• Adjusted the logging message to reflect the precision of the engine being built (FP16 or FP32).
• Conditional use of FP16 mode now also depends on the half flag.

🎯 Purpose & Impact

• Purpose: To ensure that the TensorRT engine uses the appropriate precision based on the model's requirements and the user's choice, providing better control over the trade-off between speed and accuracy.
• Impact: Users will have clearer information about the precision mode used during the engine build process and can influence the precision mode with the half setting, potentially improving performance on compatible hardware. 🚀

[github-actions]

👋 Hello @SahilChachra, 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 upstream/master. If your PR is behind upstream/master an automatic GitHub Actions merge may be attempted by writing /rebase in a new comment, or by running the following code, replacing 'feature' with the name of your local branch:

git remote add upstream https://github.com/ultralytics/yolov5.git
git fetch upstream
# git checkout feature  # <--- replace 'feature' with local branch name
git merge upstream/master
git push -u origin -f

• ✅ Verify all 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]

@SahilChachra I'd defaulted all TRT to FP16, but you're right, this is probably more aligned with the expected behavior here.

[glenn-jocher]

@SahilChachra PR is merged. Thank you for your contributions to YOLOv5 🚀 and Vision AI ⭐

[SahilChachra]

Thankyou @glenn-jocher 😃

[glenn-jocher]

@SahilChachra you're welcome! The YOLO community and the Ultralytics team have put in a lot of effort to make YOLOv5 a success. Thank you for being a part of it! If you have any more questions or feedback, feel free to ask.