From the course: Deep Learning and Computer Vision: Object Detection with PyTorch
Join today to access over 24,900 courses taught by industry experts.
Solution: Implementing object detection in GitHub Codespaces using PyTorch - PyTorch Tutorial
From the course: Deep Learning and Computer Vision: Object Detection with PyTorch
Solution: Implementing object detection in GitHub Codespaces using PyTorch
(upbeat electronic music) - [Instructor] Welcome back. In this video, we'll implement a solution to our object detection problem using GitHub Codespaces. We'll set up the dataset, load a pre-trained YOLO version five model and train it using PyTorch. Let's start by setting up our environment and loading the global wheat head data set in GitHub Codespaces. We'll first download the dataset and place it in our dataset folder within our GitHub Codespaces environment. Next, we'll create a custom dataset class to handle loading and pre-processing the data. Here we have defined a custom dataset class to load the images and corresponding bonding boxes from the dataset. This class will help us easily manage the data while training our model. Now let's load a pre-trained YOLO version five model using Torch.Hub and set it up for training. We'll fine tune this model to adapt it to our dataset. We have successfully loaded the YOLO version five model. Now let's set up the training loop using a…
Download courses and learn on the go
Watch courses on your mobile device without an internet connection. Download courses using your iOS or Android LinkedIn Learning app.