Fixed relative paths in darknet_images.py (AlexeyAB#7566)

Tossy0423 · Apr 25, 2021 · 98fa47c · 98fa47c
1 parent 5c92722
commit 98fa47c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/darknet_images.py b/darknet_images.py
@@ -162,7 +162,7 @@ def save_annotations(name, image, detections, class_names):
     """
     Files saved with image_name.txt and relative coordinates
     """
-    file_name = name.split(".")[:-1][0] + ".txt"
+    file_name = os.path.splitext(name)[0] + ".txt"
     with open(file_name, "w") as f:
         for label, confidence, bbox in detections:
             x, y, w, h = convert2relative(image, bbox)