64 lines
2.0 KiB
Python
64 lines
2.0 KiB
Python
|
import os
|
|||
|
|
import piexif
|
|||
|
|
from PIL import Image
|
|||
|
|
|
|||
|
|
def dms_to_decimal(dms):
|
|||
|
|
"""将DMS格式转换为十进制度"""
|
|||
|
|
if not dms:
|
|||
|
|
return None
|
|||
|
|
degrees = dms[0][0] / dms[0][1]
|
|||
|
|
minutes = dms[1][0] / dms[1][1] / 60
|
|||
|
|
seconds = dms[2][0] / dms[2][1] / 3600
|
|||
|
|
return degrees + minutes + seconds
|
|||
|
|
|
|||
|
|
def get_gps_info(image_path):
|
|||
|
|
"""获取图片的GPS信息"""
|
|||
|
|
try:
|
|||
|
|
image = Image.open(image_path)
|
|||
|
|
exif_data = piexif.load(image.info['exif'])
|
|||
|
|
|
|||
|
|
gps_info = exif_data.get("GPS", {})
|
|||
|
|
if not gps_info:
|
|||
|
|
return None, None, None
|
|||
|
|
|
|||
|
|
# 获取纬度
|
|||
|
|
lat = dms_to_decimal(gps_info.get(2))
|
|||
|
|
if lat and gps_info.get(1) and gps_info[1] == b'S':
|
|||
|
|
lat = -lat
|
|||
|
|
|
|||
|
|
# 获取经度
|
|||
|
|
lon = dms_to_decimal(gps_info.get(4))
|
|||
|
|
if lon and gps_info.get(3) and gps_info[3] == b'W':
|
|||
|
|
lon = -lon
|
|||
|
|
|
|||
|
|
# 获取高度
|
|||
|
|
alt = None
|
|||
|
|
if 6 in gps_info:
|
|||
|
|
alt = gps_info[6][0] / gps_info[6][1]
|
|||
|
|
|
|||
|
|
return lat, lon, alt
|
|||
|
|
|
|||
|
|
except Exception as e:
|
|||
|
|
print(f"读取文件 {image_path} 时出错: {str(e)}")
|
|||
|
|
return None, None, None
|
|||
|
|
|
|||
|
|
def main():
|
|||
|
|
# 设置输入输出路径
|
|||
|
|
image_dir = r"E:\datasets\UAV\134\project\images"
|
|||
|
|
output_path = r"E:\datasets\UAV\134\project\gps.txt"
|
|||
|
|
|
|||
|
|
with open(output_path, 'w', encoding='utf-8') as f:
|
|||
|
|
for filename in os.listdir(image_dir):
|
|||
|
|
if filename.lower().endswith(('.jpg', '.jpeg')):
|
|||
|
|
image_path = os.path.join(image_dir, filename)
|
|||
|
|
lat, lon, alt = get_gps_info(image_path)
|
|||
|
|
|
|||
|
|
if lat is not None and lon is not None:
|
|||
|
|
# 如果没有高度信息，使用0
|
|||
|
|
alt = alt if alt is not None else 0
|
|||
|
|
filename = filename.replace(".jpg", ".tif")
|
|||
|
|
f.write(f"{filename} {lat} {lon} {alt}\n")
|
|||
|
|
|
|||
|
|
if __name__ == '__main__':
|
|||
|
|
main()
|