在处理图片过程中,可能会导致图片中携带的GPS信息丢失
但是GPS信息对于工作项目又十分重要。这里我们将原图片的GPS
信息,写回处理完的图片
的GPS信息。
GPS信息如下:
经纬度的信息需要调整为:
{2: ((36, 1), (42, 1), (198789, 10000)), 4: ((116, 1), (19, 1), (477041, 10000)), 6: (57128, 1000)}
元组格式
代码如下:
from PIL import Image
from PIL.ExifTags import TAGS
import piexif
from fractions import Fraction
# 获取图片的GPS信息
def get_gps_info(image_path):
image = Image.open(image_path)
exif_data = image._getexif()
if exif_data is not None:
for tag, value in exif_data.items():
tag_name = TAGS.get(tag, tag)
if tag_name == 'GPSInfo':
return value
return None
# 向文件中写入GPS信息 path、 经度 、纬度 、 高度
def write_gps_info(image_path, gps_longitude,gps_latitude, gps_altitude):
# 打开图片
image = Image.open(image_path)
# 创建一个空的Exif数据
exif_dict = {
"GPS": {
}}
# 构建GPS信息
gps_info = {
piexif.GPSIFD.GPSLatitude: _convert_to_dms(gps_longitude),
piexif.GPSIFD.GPSLongitude: _convert_to_dms(gps_latitude),
piexif.GPSIFD.GPSAltitude: (gps_altitude,1000), # 将高度转换为以毫米为单位
}
print(gps_info)
# 将GPS信息插入到Exif数据中
exif_dict["GPS"] = gps_info
# # 将修改后的Exif数据重新写入图片
exif_bytes = piexif.dump(exif_dict)
image.save(image_path, exif=exif_bytes)
def _convert_to_dms(coordinate):
degrees = int(coordinate[0])
minutes = int(coordinate[1])
seconds = int(coordinate[2] * 10000)
dms = ((degrees, 1), (minutes, 1), (seconds, 10000))
return dms
if __name__ == '__main__':
image_path = r'C:\projects\img_enhance\IMG\DJI_0001.JPG'
wait_image_path = r'C:\projects\yolov8\datasets\solar\images\Undistort_IMGDJI_0001.JPG'
gps_info = get_gps_info(image_path)
# 经度 纬度 高度
print("图片的GPS信息:", gps_info[2],gps_info[4],gps_info[6])
gps_altitude = int(gps_info[6]*1000)
write_gps_info(wait_image_path,gps_info[2],gps_info[4],gps_altitude)
print("{} 图片写入完成".format(wait_image_path))