monemati / YOLOv8-DeepSORT-Streamlit

YOLOv8 Object Tracking and Counting using PyTorch, OpenCV and DeepSORT, deployed on Streamlit.
GNU Affero General Public License v3.0
35 stars 12 forks source link

Not supporting custom dataset #2

Closed Leanse1 closed 5 months ago

Leanse1 commented 10 months ago

Hi, I am trying to load best.pt trained through google colab. But streamlit is loading every other petrained models except best.pt. Can you explain why?

monemati commented 10 months ago

@Leanse1 Hi, you should add the model to config.py (# DL model config)