How to Prune YOLOv8 and Any PyTorch Model to Make It Faster

Опубликовано: 13 Август 2024
на канале: Nicolai Nielsen
5,752
144

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 go over how you can prune any PyTorch model which means that we reduce the number of weights in each layer and even whole layers. This will reduce the size of our models and make them faster without loosing much accuracy. Sometimes the models becomes better at generalisation since we are applying regularisation to the model and weights. In this video I show how to do it on a YOLOv8 object detection model but you can do the exact some on any PyTorch model.

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
👨‍💻 Learn Software Engineering instead with my good friend Tech with Tim: https://coursecareers.com/a/NicolaiAI...

📈 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:
00:00 Introduction
00:52 Documentation
02:07 Code Imports
02:47 Prune Function
05:27 Use YOLOv8
05:43 Validation
08:20 Run Code
08:50 Results
11:47 Outro

Tags: #yolov8pruning #Pruningpytorchmodel #PruningAI