-
Notifications
You must be signed in to change notification settings - Fork 101
Open
Description
I installed opencv with CUDA with version 4.5.4

I run the file yolo.py

I get 8 - 10 fps, which is too low. I removed the imshow and the fps stayed at the same level. I have noticed that the video card has almost no load. The load is 3 - 6%. I also tried running a smaller model, but fps did not rise above 16.

I ran this on the CPU as well, and got about 4 fps. I tried to run the second script, but it didn't work.
Traceback (most recent call last):
File "C:\Users\test\Desktop\yolov5-opencv-cpp-python-main\yolo-tiny.py", line 19, in <module>
input_image = format_yolov5(image) # making the image square
File "C:\Users\test\Desktop\yolov5-opencv-cpp-python-main\yolo-tiny.py", line 12, in format_yolov5
row, col, _ = frame.shape
AttributeError: 'NoneType' object has no attribute 'shape'
How can I improve the speed?
Metadata
Metadata
Assignees
Labels
No labels