Bläddra i källkod

Merge pull request #498 from SunAhong1993/develop

remove delete file
Jason 4 år sedan
förälder
incheckning
33c50100b0
3 ändrade filer med 4 tillägg och 4 borttagningar
  1. 1 1
      paddlex/tools/x2coco.py
  2. 1 1
      paddlex/tools/x2seg.py
  3. 2 2
      paddlex/tools/x2voc.py

+ 1 - 1
paddlex/tools/x2coco.py

@@ -71,7 +71,7 @@ class X2COCO(object):
         # Convert the image files.
         new_image_dir = osp.join(dataset_save_dir, "JPEGImages")
         if osp.exists(new_image_dir):
-            shutil.rmtree(new_image_dir)
+            raise Exception("The directory {} is already exist, please remove the directory first".format(new_image_dir))
         os.makedirs(new_image_dir)
         for img_name in os.listdir(image_dir):
             if is_pic(img_name):

+ 1 - 1
paddlex/tools/x2seg.py

@@ -110,7 +110,7 @@ class X2Seg(object):
         # Convert the image files.
         new_image_dir = osp.join(dataset_save_dir, "JPEGImages")
         if osp.exists(new_image_dir):
-            shutil.rmtree(new_image_dir)
+            raise Exception("The directory {} is already exist, please remove the directory first".format(new_image_dir))
         os.makedirs(new_image_dir)
         for img_name in os.listdir(image_dir):
             if is_pic(img_name):

+ 2 - 2
paddlex/tools/x2voc.py

@@ -41,7 +41,7 @@ class X2VOC(object):
         # Convert the image files.
         new_image_dir = osp.join(dataset_save_dir, "JPEGImages")
         if osp.exists(new_image_dir):
-            shutil.rmtree(new_image_dir)
+            raise Exception("The directory {} is already exist, please remove the directory first".format(new_image_dir))
         os.makedirs(new_image_dir)
         for img_name in os.listdir(image_dir):
             if is_pic(img_name):
@@ -51,7 +51,7 @@ class X2VOC(object):
         # Convert the json files.
         xml_dir = osp.join(dataset_save_dir, "Annotations")
         if osp.exists(xml_dir):
-            shutil.rmtree(xml_dir)
+            raise Exception("The directory {} is already exist, please remove the directory first".format(xml_dir))
         os.makedirs(xml_dir)
         self.json2xml(new_image_dir, json_dir, xml_dir)