diff --git a/projects/python/perception/object_detection_2d/yolov5/webcam_demo.py b/projects/python/perception/object_detection_2d/yolov5/webcam_demo.py index 280e3c25d1..79b8fa37e9 100644 --- a/projects/python/perception/object_detection_2d/yolov5/webcam_demo.py +++ b/projects/python/perception/object_detection_2d/yolov5/webcam_demo.py @@ -16,6 +16,8 @@ import cv2 import time +# Importing torch to avoid "RuntimeError: cuDNN error: CUDNN_STATUS_NOT_INITIALIZED" error +import torch # NOQA from opendr.engine.data import Image from opendr.perception.object_detection_2d import YOLOv5DetectorLearner