search

rampal-punia / yolov8-streamlit-detection-tracking

Object detection and tracking algorithm implemented for Real-Time video streams and static images.
https://codingmantras-yolov8-streamlit-detection-tracking-app-njcqjg.streamlit.app/
279 stars 119 forks source link

About YOLOv8 weights #8

Open Usman2217 opened 11 months ago

Usman2217 commented 11 months ago

Hello Sir, Im new to ML and trying to train a model(object detedtion) on Google Colab and make a streamlit web app of it in colab bcz my local machine has not the GPU. My question is: After training i need to load the model.Im confused which file should i load the best.pt or the yolov8s.pt?image And after that i can make a function to upload and detect images with bounding boxes and labels on it. I will be very grateful to you.

quillbit8 commented 10 months ago

You should load the trained best.pt model :)