1. 增加批处理工具

LabelPortShipRasterSlice/shpfile2dotaBatch.py

@@ -0,0 +1,79 @@
+import numpy as np
+import os
+from glob import glob
+from pathlib import Path
+# from multiprocessing import Pool
+########################################################
+# 函数区
+########################################################
+
+spacetySliceEnvPathExEPath=r"D:\ProgramData\miniconda3\envs\spacetySliceEnv\python.exe"
+
+def find_tif_files_pathlib(directory):
+    """
+    使用pathlib.Path.rglob递归查找指定目录下所有.tif和.tiff文件
+
+    参数:
+        directory (str): 要搜索的根目录路径
+
+    返回:
+        list: 包含所有找到的.tif/.tiff文件完整路径的列表
+    """
+    path = Path(directory)
+    # 使用rglob递归匹配所有.tif和.tiff文件
+    tif_files = list(path.rglob('*.tif')) + list(path.rglob('*.tiff'))
+    # 将Path对象转换为字符串路径
+    return [str(file) for file in tif_files]
+
+
+def find_shp_files_pathlib(directory):
+    """
+    使用pathlib.Path.rglob递归查找指定目录下所有.tif和.tiff文件
+
+    参数:
+        directory (str): 要搜索的根目录路径
+
+    返回:
+        list: 包含所有找到的.tif/.tiff文件完整路径的列表
+    """
+    path = Path(directory)
+    # 使用rglob递归匹配所有.tif和.tiff文件
+    tif_files = list(path.rglob('*.shp'))
+    # 将Path对象转换为字符串路径
+    return [str(file) for file in tif_files]
+
+
+def preProcessLabelshapefile(shpfilepath,preFolderPath):
+    if not os.path.exists(preFolderPath):
+        os.makedirs(preFolderPath)
+    file_path = Path(shpfilepath)
+    directory_path = str(file_path.resolve().parent.name)
+
+    rootname = Path(shpfilepath).stem
+    txtpath=os.path.join(preFolderPath,rootname+'.txt')
+    programpath=r"R:\TYSAR-德清院\A-预处理-未标注\A0-算法版本\AA\SpacetySliceDataTools\LabelPortShipRasterSlice\Portshapefile2dota_AA.py"
+    cmdtxt=r"{} {} -i {} -o {}".format(spacetySliceEnvPathExEPath,programpath,shpfilepath,txtpath)
+    if os.system(cmdtxt) ==2:
+        print("sucess:",cmdtxt)
+        return "sucess:  {}".format(cmdtxt)
+    else:
+        print("failed:",cmdtxt)
+        return "failed:  {}".format(cmdtxt)
+    pass
+
+
+########################################################
+# 流程执行区
+########################################################
+if __name__ == '__main__':
+    srcFolderPath = r"D:\Annotation_Y\港口\聚束模式"
+    preFolderPath= r"D:\Annotation_Y\港口\聚束模式"
+    shpPaths = find_shp_files_pathlib(srcFolderPath)
+    for shppath in shpPaths:
+        preProcessLabelshapefile(shppath, preFolderPath)
+
+
+
+
+
+