Skip to content

Latest commit

 

History

History
18 lines (10 loc) · 1.72 KB

training.md

File metadata and controls

18 lines (10 loc) · 1.72 KB

YOLOv5 training on a custom dataset on OpenShift AI

Now that our RHOAI environment is set up and our data has been prepared, we can begin model training and make some predictions. We will head back to the RHOAI Dashboard to view our Data Science Projects. We'll start by opening the pokedex workbench.

Screenshot 2023-11-30 at 9 36 48 AM

Log with your cluster credentials and you will be promted to allow permissions, keep everything checked and click allow selected permissions.

Screenshot 2023-11-30 at 9 41 43 AM

Now we're in our Jupyter environment, where we will be doing all experimentation, training and testing. First we will need to import our Pokedex repository from GitHub. First click the icon featured below.

Screenshot 2023-11-30 at 9 39 31 AM

From there, you click Clone a repository and paste the Pokedex repository to clone it in your Jupyter environment.

Screenshot 2023-11-30 at 9 40 22 AM

After a few moments, you should see the pokedex-demo directory in your environment. Under pokedex-demo/Notebooks/, open Pokedex_YOLO_v8_new.ipynb and follow the notebook for further instructions.