Need help with one project, im learning YOLO, so im doing a project, one guy did the same way as i did, and it was correct.
this is the code
and with videos, i dont think its about the code, the problem is in the model or some config i dont know, i tested in two computers, one with GPU and other with CPU
When i run in google collab, everything is fine
So what exactly is your question?
I resolved, there is some bug with PYTORCH 2.4.0 i have the 2.3.1 now
1 Like
solution:
pip install torch==2.3.1 torchvision==0.18.1 torchaudio==2.3.1 --index-url https://download.pytorch.org/whl/cpu