From e98ab44bf530da6eb94ab9b6dbb21bb39bce3abe Mon Sep 17 00:00:00 2001
From: Glenn Jocher <glenn.jocher@ultralytics.com>
Date: Mon, 21 Jun 2021 22:50:56 +0200
Subject: [PATCH] Fix `img2label_paths()` order (#3720)

* Fix `img2label_paths()` order

* fix, 1
---
 utils/datasets.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/utils/datasets.py b/utils/datasets.py
index 93d6511ac658..25a7b2f67355 100755
--- a/utils/datasets.py
+++ b/utils/datasets.py
@@ -348,7 +348,7 @@ def __len__(self):
 def img2label_paths(img_paths):
     # Define label paths as a function of image paths
     sa, sb = os.sep + 'images' + os.sep, os.sep + 'labels' + os.sep  # /images/, /labels/ substrings
-    return ['txt'.join(x.replace(sa, sb, 1).rsplit(x.split('.')[-1], 1)) for x in img_paths]
+    return [sb.join(x.rsplit(sa, 1)).rsplit('.', 1)[0] + '.txt' for x in img_paths]
 
 
 class LoadImagesAndLabels(Dataset):  # for training/testing