update 更新打包模块

Author: jadehh

dataset_tools/jade_voc_datasets.py

@@ -10,6 +10,7 @@
 from opencv_tools import *
 from xml.dom import minidom
 from dataset_tools import *
+from opencv_tools import *
 def GetXmlClassesNames(xml_path):
     classnames = []
     # Read the XML annotation file.
@@ -261,6 +262,20 @@ def video_to_voc(self,save_path,detector=None,fps=5):
                     index = index + 1
             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))
+    image_path_list = GetAllImagesPath(os.path.join(image_path))
+    processBar = ProgressBar(len(image_path_list))
+    for image_path in image_path_list:
+        file_name = GetLastDir(image_path).split(".")[0]
+        image = ReadChinesePath(image_path)
+        result = detector.predict(image,threshold)
+        if result["labels"][0] == -1:
+            GenerateXml(file_name,image.shape,[],[],save_anno_path)
+        else:
+            GenerateXml(file_name,image.shape,result["boxes"],result["labels"],save_anno_path)
+        processBar.update()
 
 if __name__ == '__main__':
     jadeVOCDatasets = JadeVOCDatasets(r'F:\数据集\VOC数据集\定制版顶相机箱号检测数据集')

main.py

@@ -14,7 +14,7 @@
     #removeNolabelDatasets(r"F:\数据集\关键点检测数据集\定制版箱号关键点数据集\2022-03-09")
     #create_text_detection_datasets(r"F:\数据集\关键点检测数据集\定制版箱号关键点数据集",r"E:\Data\字符检测识别数据集\定制版箱号关键点数据集",0.95)
     #CreatePaddleOCRDatasets(root_path="E:\Data\字符检测识别数据集\镇江大港厂内车牌关键点检测数据集", save_path="E:\Data\OCR\镇江大港厂内车牌识别数据集",dataset_type="镇江厂内车牌数据集")
-    #removeNolabelVocDatasets(r"E:\Data\VOC数据集\箱门检测数据集")
-    CreateYearsDatasets(r"E:\Data\VOC数据集\箱门检测数据集\ContainVOC")
+    removeNolabelVocDatasets(r"E:\Data\VOC数据集\集装箱残损检测数据集")
+    CreateYearsDatasets(r"E:\Data\VOC数据集\集装箱残损检测数据集")
     #create_text_detection_datasets(r"F:\数据集\关键点检测数据集\箱号关键点数据集",r'E:\Data\字符检测识别数据集\箱号关键点数据集')
     #CreatePaddleOCRDatasets(r'E:\Data\字符检测识别数据集\箱号关键点数据集', save_path="E:\Data\OCR\箱号识别数据集",dataset_type="箱号数据集")

setup.py

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