Skip to main content

Posts

Showing posts from August, 2022

How to train a custom dataset and make inference with Official YOLOv7 on Kaggle?

Another post starts with you beautiful people! I hope you have already learned the state-of-the-art object detection technique Darknet's YOLOv4 from my previous posts. If you are new to my blog then I recommend you to once go through  this link  to learn about end-to-end implementation of YOLOv4. In this post we are going to learn the successor of the  official YOLOv4  :  YOLOv7: Trainable bag-of-freebies sets new state-of-the-art for real-time object detectors  ðŸ’¥. Since the official implementation of YOLOv7 is still not done in the Darknet framework, in this post we are going to use the Pytorch framework to train a custom dataset and then make inferences using the same Pytorch framework. But don't worry if you are new to PyTorch 👲, we are not going to reinvent the wheel in fact as a practitioner our goal should be to learn how to use it. For our learning, we will use a real-world dataset and the problem-  detecting starfish in underwater images . This challenge was held on K