From a66f9ac2208af3d9210da222fad6e0304fd45818 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E9=99=88=E5=A2=9E=E8=BE=89?= <3045316072@qq.com> Date: Mon, 22 Sep 2025 17:18:26 +0800 Subject: [PATCH] =?UTF-8?q?=E5=A2=9E=E5=8A=A0=E5=86=9B=E6=B0=91=E6=B8=AF?= =?UTF-8?q?=E5=8F=A3=E6=96=87=E4=BB=B6=E5=8C=BA=E5=88=86?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- .../PredictProcessRaster_AC.py | 2 - .../SpacetyTIFFDataStretch2PNG_AC.py | 60 +++++++++---------- 2 files changed, 29 insertions(+), 33 deletions(-) diff --git a/generatorRasterSlicesTools/PredictProcessRaster_AC.py b/generatorRasterSlicesTools/PredictProcessRaster_AC.py index e5d8183..3f360a1 100644 --- a/generatorRasterSlicesTools/PredictProcessRaster_AC.py +++ b/generatorRasterSlicesTools/PredictProcessRaster_AC.py @@ -16,8 +16,6 @@ spacetySliceEnvPathExEPath=r"d:\ProgramData\anaconda3\envs\spacetySliceEnv\pytho logPath=r"R:\TYSAR-德清院\A-预处理-未标注\A0-算法版本\AA\SpacetySliceDataTools\log\process.log" - - def existOrCreate(dirpath): if not os.path.exists(dirpath): os.makedirs(dirpath) diff --git a/generatorRasterSlicesTools/SpacetyTIFFDataStretch2PNG_AC.py b/generatorRasterSlicesTools/SpacetyTIFFDataStretch2PNG_AC.py index 84c6420..0ef2767 100644 --- a/generatorRasterSlicesTools/SpacetyTIFFDataStretch2PNG_AC.py +++ b/generatorRasterSlicesTools/SpacetyTIFFDataStretch2PNG_AC.py @@ -17,6 +17,11 @@ portsliceSize=5000 shipsliceSize=1024 BlockOverLayer=0.25 +def existOrCreate(dirpath): + if not os.path.exists(dirpath): + os.makedirs(dirpath) + + def get_filename_without_ext(path): base_name = os.path.basename(path) if '.' not in base_name or base_name.startswith('.'): @@ -212,20 +217,6 @@ def stretchProcess(infilepath,outfilepath,strechmethod): print("图像拉伸处理结束") -#切片模式 -def getSlicePoints(h): - n = int(math.floor((h - 1024) * 1.2 / sliceSize)) - step=int(math.ceil((h-1024)/n)) - ti=list(range(0,h-1024,step)) - ti.append(h-1024) - # 评价重叠率 - movelayer=[] - for i in range(len(ti)-1): - movelayer.append((ti[i] + 1024 - ti[i + 1]) / 1024 * 100.0) - print("重叠率：",movelayer) - return ti - - def getsliceGeotrans(GeoTransform,Xpixel,Ypixel): XGeo = GeoTransform[0]+GeoTransform[1]*Xpixel+GeoTransform[2]*Ypixel YGeo = GeoTransform[3]+GeoTransform[4]*Xpixel+GeoTransform[5]*Ypixel @@ -254,9 +245,9 @@ def getNextSliceNumber(n,sliceSize,overlap=0.25): return newN,ti def sliceShipDataset(rootname,im_data,src_im_data, im_Geotrans, im_proj, outfolder): - binfolder=os.path.join(outfolder,"unit8binfolder") - pngfolder=os.path.join(outfolder,"pngfolder") - tifffolder=os.path.join(outfolder,"tifffolder") + binfolder=os.path.join(outfolder,"舰船","unit8binfolder") + pngfolder=os.path.join(outfolder,"舰船","pngfolder") + tifffolder=os.path.join(outfolder,"舰船","tifffolder") h,w=im_data.shape @@ -340,9 +331,9 @@ def slicePortDataset(rootname,im_data,src_im_data, im_Geotrans, im_proj, outfold else: pass # 切片主流程 - binfolder=os.path.join(outfolder,"unit8binfolder") - pngfolder=os.path.join(outfolder,"pngfolder") - tifffolder=os.path.join(outfolder,"tifffolder") + binfolder=os.path.join(outfolder,"港口","unit8binfolder") + pngfolder=os.path.join(outfolder,"港口","pngfolder") + tifffolder=os.path.join(outfolder,"港口","tifffolder") h,w=im_data.shape nextH,ht=getNextSliceNumber(h,portsliceSize,BlockOverLayer) nextW,wt=getNextSliceNumber(w,portsliceSize,BlockOverLayer) @@ -371,18 +362,25 @@ def slicePortDataset(rootname,im_data,src_im_data, im_Geotrans, im_proj, outfold def stretchSliceProcess(infilepath, outfolder,portfilestr, strechmethod): - binfolder=os.path.join(outfolder,"unit8binfolder") - pngfolder=os.path.join(outfolder,"pngfolder") - tifffolder=os.path.join(outfolder,"tifffolder") + shipbinfolder=os.path.join(outfolder,"舰船","unit8binfolder") + shippngfolder=os.path.join(outfolder,"舰船","pngfolder") + shiptifffolder=os.path.join(outfolder,"舰船","tifffolder") allpngfolder = os.path.join(outfolder, "allpngfolder") - if not os.path.exists(binfolder): - os.makedirs(binfolder) - if not os.path.exists(pngfolder): - os.makedirs(pngfolder) - if not os.path.exists(tifffolder): - os.makedirs(tifffolder) - if not os.path.exists(allpngfolder): - os.makedirs(allpngfolder) + + portbinfolder=os.path.join(outfolder,"港口","unit8binfolder") + portpngfolder=os.path.join(outfolder,"港口","pngfolder") + porttifffolder=os.path.join(outfolder,"港口","tifffolder") + + existOrCreate(shipbinfolder) + existOrCreate(shippngfolder) + existOrCreate(shiptifffolder) + existOrCreate(allpngfolder) + existOrCreate(portbinfolder) + existOrCreate(portpngfolder) + existOrCreate(porttifffolder) + + + im_proj, im_Geotrans, im_data=read_tif(infilepath) src_im_data=im_data*1.0 im_data = DataStrech(im_data,strechmethod) # 拉伸