百度街景瓦片拼接

爱被打了一巴掌 2022-02-28 14:06 323阅读 0赞

import glob, os  
import re  
from PIL import Image  
import time

# 3328 1664 #

# 获取指定目录下的文件列表 类似os.listdir #

files = glob.glob(‘D:\\fgq\\baiduimgs\\*\_**\**\*.png’)

# 最后大图的输出目录 #

# files\_big\_pic\_path = “D:\\fgq\\baiduimgsPic\\” #

files\_big\_pic\_path = “D:\\fgq\\ddd\\”

# 定义一个set集合不可出现重复元素 记录panoid #

set\_rep = set()

# int(i) for i in re.findall(’\\d+’, x)\[:2\] #

# 根据panoid进行一个排序 #

files.sort(key=lambda x: tuple(\[set\_rep.add(i) for i in re.findall(r’\\d+D’, x)\]))

# print(len(set\_rep)) #

# print(len(files)) #

# 形成以panoid为key 当前panoid所有的行、列的瓦片集合为value的一个不重复key的字典集合 #

imagefiles = \{\}  
for item in files:  
match = re.search(r’\\d+D’, item)  
pre = match.group()  
if not imagefiles.get(pre):  
imagefiles\[pre\] = \[\]  
imagefiles\[pre\].append(item)

# 将字典的key集合和对应的value集合 形成一个个一一对应的元组的集合并排序 #

imagefiles = sorted(zip(imagefiles.keys(), imagefiles.values()));

# 每个小图都是512\*512 一张大图是 8列 \* 4行=32张小图 #

total\_width = 8 \* 512  
total\_height = 4 \* 512

# \# 创建一张大图的画布 #

# new\_image = Image.new(‘RGB’, (total\_width, total\_height)) #

time1 = time.time()  
for item in imagefiles:  
x\_offset = 0  
y\_offset = 0  
count = 1  
images = map(Image.open, item\[1\])  
new\_image = Image.new(‘RGB’, (total\_width, total\_height))  
for subitem in images:  
new\_image.paste(subitem, (x\_offset, y\_offset))  
x\_offset += subitem.size\[1\]  
if count % 8 == 0:  
y\_offset += subitem.size\[0\]  
x\_offset = 0  
count = 1  
else:  
count += 1  
\# 将图像改变尺寸、压缩大小后保存到文件夹 切记压缩方法只适用于jpg/JPEG格式  
out = new\_image.resize((3328, 1664), Image.ANTIALIAS)  
out.save(files\_big\_pic\_path + item\[0\] + “.jpg”,quality=75, subsampling=0)  
new\_image.close()  
out.close()  
time2 = time.time()  
print(str(time2-time1))