update 数据集操作新增删除voc ImageSets文件夹

Author: jadehh

dataset_tools/jade_voc_datasets.py

@@ -262,6 +262,16 @@ def video_to_voc(self,save_path,detector=None,fps=5):
                     index = index + 1
             processBar.update()
 
+def remove_voc_imagesets(root_path):
+    processBar = ProgressBar(len(os.listdir(root_path)))
+    for year in os.listdir(root_path):
+        if os.path.isdir(os.path.join(root_path,year)):
+            year_path = os.path.join(root_path,year)
+            for file in os.listdir(year_path):
+                if  file == "ImageSets":
+                    shutil.rmtree(os.path.join(year_path,file))
+        processBar.update()
+
 def paddle_pretrain_detection_dataset(root_path,detector,threshold=0.6):
     image_path = os.path.join(root_path,DIRECTORY_IMAGES)
     save_anno_path = CreateSavePath(os.path.join(root_path,DIRECTORY_ANNOTATIONS))

setup.py

@@ -13,7 +13,7 @@
 
     setup(
     name="dataset_tools",
-    version="1.0.4",
+    version="1.0.5",
     keywords=("pip", "dataset_tools", ""),
     description="dataset_tools",
     long_description="",