fix deprecation warning (#199)

fcakyon · Dec 31, 2022 · f5a176d · f5a176d
1 parent 2ee349f
commit f5a176d
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/yolov5/helpers.py b/yolov5/helpers.py
@@ -1,4 +1,5 @@
 from pathlib import Path
+import warnings
 
 from yolov5.models.common import AutoShape, DetectMultiBackend
 from yolov5.models.experimental import attempt_load
@@ -65,7 +66,8 @@ def load_model(
 
 class YOLOv5:
     def __init__(self, model_path, device=None, load_on_init=True):
-        raise DeprecationWarning('YOLOv5 class is deprecated and will be removed in future release. Use "model = yolov5.load()" instead.')
+        warnings.warn("YOLOv5 class is deprecated and will be removed in future release. Use 'model = yolov5.load()' instead.", DeprecationWarning)
+
         self.model_path = model_path
         self.device = device
         if load_on_init: