predict.py on custom weights

MuhammadMoinFaisal / YOLOv8-DeepSORT-Object-Tracking

YOLOv8 Object Tracking using PyTorch, OpenCV and DeepSORT

666 stars 197 forks source link

predict.py on custom weights #16

Closed arhamlet closed 1 year ago

arhamlet commented 1 year ago

I have trained a yolov8 model which has two labels. How can I pass the labels to predict.py file for the model to output classes for the custom weights. I have already tried changing the coco.yaml and coco128.yaml files but didn't work

MuhammadMoinFaisal commented 1 year ago

Hi, Please check my YouTube here i have made multiple videos regarding this https://www.youtube.com/@muhammadmoinfaisal/videos Thanks