taking too long - Githubissues

rigvedrs / YOLO-V8-CAM

Wanna know what your model sees? Here's a package for applying EigenCAM on the new YOLO V8 model

MIT License

100 stars 17 forks source link

taking too long #12

Closed erminkev1 closed 1 month ago

erminkev1 commented 4 months ago

Hello :)

Thank you for this awesome tool!

I've ran a "m" model which I have trained on a custom dataset, and I've waited over 3 minutes each time I ran it, without receiving any output. I assume this is some weird behavior that isn't supposed to happen. I do have gpu, and I did try to send my model to gpu but it didn't help somehow. Any tips would be appreciated.

Best, Ermin

rigvedrs commented 4 months ago

The first run for running YOLO models takes long, but the next runs will be faster. Can you try passing an image to the model one more time to the same model and see how long it takes?