search

WongKinYiu / yolov7

Implementation of paper - YOLOv7: Trainable bag-of-freebies sets new state-of-the-art for real-time object detectors
GNU General Public License v3.0
13.34k stars 4.21k forks source link

CUDA is shown but no detection occurs? #1246

Open haamaam opened 1 year ago

haamaam commented 1 year ago

Hi, i have no problem with detection my model in yolov5 or any other yolo version but yolov7. When i start the detect.py it shows the video, my gpu but no detection occurs?

yulin010101 commented 1 year ago

Does running on CPU get the correct result?

haamaam commented 1 year ago

yes, it is detecting with cpu, and detecting with gpu for other yolo versions

yulin010101 commented 1 year ago

Maybe it's because of half precision. You can try setting "half = False" in detect.py.

chrislill commented 1 year ago

Thanks @yulin010101, this workaround worked for me. I'm running Windows 10 with a NVIDIA GeForce GTX 1650 GPU