Update darknet_images.py (#8559)

Safer and better type check.
AlexeyAB · Aug 25, 2023 · 2bb27c1 · 2bb27c1
1 parent 38a5b48
commit 2bb27c1
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/darknet_images.py b/darknet_images.py
@@ -101,7 +101,7 @@ def image_detection(image_or_path, network, class_names, class_colors, thresh):
     height = darknet.network_height(network)
     darknet_image = darknet.make_image(width, height, 3)
 
-    if type(image_or_path) == "str":
+    if isinstance(image_or_path, str):
         image = cv2.imread(image_or_path)
     else:
         image = image_or_path