How to export and optimize YOLO-NAS object detection model for real-time with ONNX and TensorRT

Опубликовано: 01 Февраль 2024
на канале: Nicolai Nielsen
6,973
62

Inside my school and program, I teach you my system to become an AI engineer or freelancer. Life-time access, personal help by me and I will show you exactly how I went from below average student to making $250/hr. Join the High Earner AI Career Program here 👉 https://www.nicolai-nielsen.com/aicareer (PRICES WILL INCREASE SOON)

You will also get access to all the technical courses inside the program, also the ones I plan to make in the future! Check out the technical courses below 👇
___________________________________________________________

In this video 📝 we are going to cover how to export and optimize YOLO-NAS models to ONNX and TensorRT for real-time performance. We will cover the whole pipeline step by step so you can follow along. The speedups we can get from optimizing our models are crazy without really loosing accuracy. It's a ton of information to ingest, but very important in the world of deep learning and computer vision. This is how good real-world applications and projects are made.

Code: https://github.com/niconielsen32/YOLO...
YOLO-NAS GitHub: https://github.com/Deci-AI/super-grad...
Supergradient documentation: https://docs.deci.ai/super-gradients/...
Deci's models: https://deci.ai/foundation-models/

If you enjoyed this video, be sure to press the 👍 button so that I know what content you guys like to see.

___________________________________________________________

🛠 Freelance Work: https://www.nicolai-nielsen.com/nncode

___________________________________________________________

💻💰🛠 High Earner AI Career Program: https://www.nicolai-nielsen.com/aicareer

📈 Investment Course: https://www.nicos-school.com/p/invest...
⚙ Real-world AI Technical Courses: (https://www.nicos-school.com)
📗 OpenCV GPU in Python: https://www.nicos-school.com/p/opencv...
📕 YOLOv7 Object Detection: https://www.nicos-school.com/p/yolov7...
📒 Transformer & Segmentation: https://www.nicos-school.com/p/transf...
📙 YOLOv8 Object Tracking: https://www.nicos-school.com/p/yolov8...
📘 Research Paper Implementation: https://www.nicos-school.com/p/resear...
📔 CustomGPT: https://www.nicos-school.com/p/custom...

___________________________________________________________

📞 Connect with Me:
🌳 https://linktr.ee/nicolainielsen
🌍 My Website: https://www.nicolai-nielsen.com/
🤖 GitHub: https://github.com/niconielsen32
👉 LinkedIn:   / nicolaiai  
🐦 X/Twitter:   / nielsencv_ai  
🌆 Instagram:   / nicolaicodes  

___________________________________________________________

🇦🇪 Want to move to Dubai and get visa? - Book a free consultation: https://www.genzoneconsulting.com/mee...

___________________________________________________________

📷 My camera calibration software, CharuCo Boards, and Checker boards
Link to webshop: https://camera-calibrator.com

___________________________________________________________

Timestamps:
0:00 Intro
0:44 YOLO-NAS Documentation
3:59 Install Cuda & TensortRT
4:55 Optimize to ONNX
11:36 Results ONNX
14:33 Convert to TensorRT
21:45 TensorRT Results
22:51 Outro

tags:
#YOLO-NAS #ONNX #TensorRT #supergradients