From ec972b72b96e3a2ab67296f1b8004e483d9695c7 Mon Sep 17 00:00:00 2001
From: Morgan <30824839+x12901@users.noreply.github.com>
Date: Thu, 17 Dec 2020 20:26:40 +0800
Subject: [PATCH] =?UTF-8?q?=E4=BF=AE=E5=A4=8D=E5=9B=BE=E5=83=8F=E5=90=8D?=
 =?UTF-8?q?=E4=B8=AD=E4=B8=8D=E8=83=BD=E5=8C=85=E5=90=AB=E7=9B=AE=E5=BD=95?=
 =?UTF-8?q?=E5=90=8D=E7=9A=84=E9=97=AE=E9=A2=98?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

原本的程序如果图像名有目录名会被删除掉
---
 pdseg/tools/create_dataset_list.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/pdseg/tools/create_dataset_list.py b/pdseg/tools/create_dataset_list.py
index 8dd4c7e9a3..a33bfaad51 100644
--- a/pdseg/tools/create_dataset_list.py
+++ b/pdseg/tools/create_dataset_list.py
@@ -128,12 +128,12 @@ def generate_list(args):
         file_list = os.path.join(dataset_root, dataset_split + '.txt')
         with open(file_list, "w") as f:
             for item in range(num_images):
-                left = image_files[item].replace(dataset_root, '')
+                left = image_files[item].replace(dataset_root, '', 1)
                 if left[0] == os.path.sep:
                     left = left.lstrip(os.path.sep)
 
                 try:
-                    right = label_files[item].replace(dataset_root, '')
+                    right = label_files[item].replace(dataset_root, '', 1)
                     if right[0] == os.path.sep:
                         right = right.lstrip(os.path.sep)
                     line = left + separator + right + '\n'