【Python入门教程】读取图片信息(经纬度、偏转角、无人机POS数据、大疆影像等)

        通常读取图片的属性信息(如经纬度、拍摄时间、IMU数据等)都是通过exifread库进行读取,但是有些图片用这个库读取的效果不好。所以今天我就和大家分享一下如何使用Python读取图片属性信息的三种方法。

原创作者:RS迷途小书童

博客地址:https://blog.csdn.net/m0_56729804?type=blog

1 GDAL读取

        GDAL库是用来处理卫星影像的库,它同样可以读取普通的jpg、png格式的图片,可以查看图片的行列数、波段数、坐标系等信息。

# -*- coding: utf-8 -*-
"""
@Time : 2023/06/21 11:52
@Auth : RS迷途小书童
@File :Read Image.py
@IDE :PyCharm
@Purpose:读取图片信息
@Web:博客地址:https://blog.csdn.net/m0_56729804
"""
import exifread
from osgeo import gdal


def Get_data(filepath):
    print("------------------------------------影像基础信息-----------------------------------")
    ds = gdal.Open(filepath)  # 打开数据集dataset
    ds_width = ds.RasterXSize  # 获取数据宽度
    ds_height = ds.RasterYSize  # 获取数据高度
    ds_bands = ds.RasterCount  # 获取波段数
    ds_geo = ds.GetGeoTransform()  # 获取仿射地理变换参数
    ds_prj = ds.GetProjection()  # 获取投影信息
    print("影像的宽度为:" + str(ds_width))
    print("影像的高度为:" + str(ds_height))
    print("波段数为:" + str(ds_bands))
    print("仿射地理变换参数为:" + str(ds_geo))
    print("投影坐标系为:" + str(ds_prj))
    # data = ds.ReadAsArray(0, 0, ds_width, ds_height)  # 以数组的形式读取整个数据集

2 exifread读取

        这个就是最普遍的读取图片信息的库了,读取的属性值也有很多。如经纬度、时间、相机参数等等。

# -*- coding: utf-8 -*-
"""
@Time : 2023/06/21 11:52
@Auth : RS迷途小书童
@File :Read Image.py
@IDE :PyCharm
@Purpose:读取图片信息
@Web:博客地址:https://blog.csdn.net/m0_56729804
"""
import exifread
from osgeo import gdal


def Get_LatLon(path_image):
    """
    :param path_image: 输入图片路径
    :return: 返回图片中心点经纬度
    """
    print("----------------------------------图片exifread信息---------------------------------")
    # 获取图片的经纬度信息
    f = open(path_image, 'rb')
    contents = exifread.process_file(f)
    for key, value in contents.items():
        if key in ["GPS GPSLongitude", "GPS GPSLatitude"]:
            value = contents[key].values[0].num/contents[key].values[0].den + \
                    (contents[key].values[1].num/contents[key].values[1].den/60) + \
                    (contents[key].values[2].num/contents[key].values[2].den/3600)
            # longitude = contents["GPS GPSLongitude"].values
            # longitude_f = longitude[0].num / longitude[0].den + (longitude[1].num / longitude[1].den / 60) + (
            #             longitude[2].num / longitude[2].den / 3600)
            # latitude = contents["GPS GPSLatitude"].values
            # latitude_f = latitude[0].num / latitude[0].den + (latitude[1].num / latitude[1].den / 60) + (
            #             latitude[2].num / latitude[2].den / 3600)
        elif key == "JPEGThumbnail":
            break
        print(f"{key}: {value}")
    f.close()

3 二进制模式读取

        这个方法仅适用于部分的相机拍摄的图片,如大疆、长光禹辰等。因为它是通过二进制编码进行读取,所以如果图片的二进制编码中不包含代码中的编码就读取不出来。

# -*- coding: utf-8 -*-
"""
@Time : 2023/06/21 11:52
@Auth : RS迷途小书童
@File :Read Image.py
@IDE :PyCharm
@Purpose:读取图片信息
@Web:博客地址:https://blog.csdn.net/m0_56729804
"""
import exifread
from osgeo import gdal



def Get_Image_Yaw_angle(file_path):
    """
    :param file_path: 输入图片路径
    :return: 图片的偏航角
    """
    # 获取图片偏航角
    print("----------------------------------大疆exifread信息---------------------------------")
    # 定义字节模式 b 和 a,用于查找大疆EXIF数据的起始和结束标记
    b = b"\x3c\x2f\x72\x64\x66\x3a\x44\x65\x73\x63\x72\x69\x70\x74\x69\x6f\x6e\x3e"
    a = b"\x3c\x72\x64\x66\x3a\x44\x65\x73\x63\x72\x69\x70\x74\x69\x6f\x6e\x20"
    # 打开图片文件,以二进制模式读取
    img = open(file_path, 'rb')
    # 初始化一个字节数组用于存储EXIF数据
    data = bytearray()
    # 初始化一个标志,用于判断是否已经找到EXIF数据的起始标记
    flag = False
    # 逐行读取图片文件内容
    for line in img.readlines():
        # 如果当前行包含EXIF数据的起始标记,则设置标志为True
        if a in line:
            flag = True
            # 如果标志为True,则将当前行添加到EXIF数据中
        if flag:
            data += line
            # 如果当前行包含EXIF数据的结束标记,则跳出循环
        if b in line:
            break
            # 如果提取到的EXIF数据不为空
    dj_data_dict = {}
    # 遍历过滤后的行,并提取键值对存入字典中
    if len(data) > 0:
        # 将字节数据解码为ASCII字符串
        data = str(data.decode('ascii'))
        # 过滤出包含drone-dji的行,并分割每行为键值对
        lines = list(filter(lambda x: 'drone-dji:' in x, data.split("\n")))
        # 初始化一个空字典用于存储提取到的数据
        for d in lines:
            d = d.strip()[10:]  # 去除每行的前后空格和'\n'字符,并从第10个字符开始处理(因为drone-dji:占据了前9个字符)
            k, v = d.split("=")  # 将当前行分割为键和值两部分
            print(f"{k} : {v}")  # 打印键和值
            dj_data_dict[k] = v  # 将键值对存入字典中
    return dj_data_dict.get('YawAngle')  # 返回偏航角的值。如果未找到偏航角,则返回None。

4 总结

        上面就是三种读取图片属性信息的方法,基本覆盖了市面上的所有相机拍摄的图片。我平时最常用它的就是用其读取无人机影像的POS数据。

        本文章主要是分享个人在学习Python过程中写过的一些代码。有些部分参考了前人以及官网的教程,如有侵权请联系作者删除,大家有问题可以随时留言交流,博主会及时回复。

这个问题比较复杂,需要用到多个模块和库来实现。以下是一个基本的程序框架,你可以根据自己的需求进行修改。 1. 导入必要的模块和库 ```python import matplotlib.pyplot as plt from mpl_toolkits.mplot3d import Axes3D import numpy as np import pandas as pd from datetime import datetime ``` 2. 读取数据 ```python data = pd.read_csv('flight_data.csv') # 将字符串格式的时间转换为datetime格式 data['datetime'] = data['time'].apply(lambda x: datetime.strptime(x, '%Y-%m-%d %H:%M:%S')) # 将经纬度高度转换为xyz坐标 R_earth = 6371000 # 地球半径,单位m data['x'] = (R_earth + data['altitude']) * np.cos(np.radians(data['latitude'])) * np.cos(np.radians(data['longitude'])) data['y'] = (R_earth + data['altitude']) * np.cos(np.radians(data['latitude'])) * np.sin(np.radians(data['longitude'])) data['z'] = (R_earth + data['altitude']) * np.sin(np.radians(data['latitude'])) ``` 3. 绘制三维轨迹 ```python fig = plt.figure(figsize=(10, 10)) ax = fig.add_subplot(111, projection='3d') # 绘制轨迹 ax.plot(data['x'], data['y'], data['z'], color='blue') # 设置坐标轴标签 ax.set_xlabel('X') ax.set_ylabel('Y') ax.set_zlabel('Z') # 显示图像 plt.show() ``` 4. 绘制姿态 ```python # 将弧度转换为角度 data[['pitch', 'roll', 'yaw']] = np.degrees(data[['pitch', 'roll', 'yaw']]) # 绘制姿态 fig, ax = plt.subplots(1, 3, figsize=(10, 5), subplot_kw=dict(projection='3d')) ax[0].set_title('Pitch') ax[0].view_init(elev=0, azim=0) ax[0].plot(data['x'], data['y'], data['z'], color='blue') ax[0].quiver(data['x'], data['y'], data['z'], np.ones_like(data['x']), np.zeros_like(data['y']), np.zeros_like(data['z']), length=100, color='red') ax[1].set_title('Roll') ax[1].view_init(elev=0, azim=90) ax[1].plot(data['x'], data['y'], data['z'], color='blue') ax[1].quiver(data['x'], data['y'], data['z'], np.zeros_like(data['x']), np.ones_like(data['y']), np.zeros_like(data['z']), length=100, color='red') ax[2].set_title('Yaw') ax[2].view_init(elev=0, azim=0) ax[2].plot(data['x'], data['y'], data['z'], color='blue') ax[2].quiver(data['x'], data['y'], data['z'], np.zeros_like(data['x']), np.zeros_like(data['y']), np.ones_like(data['z']), length=100, color='red') # 显示图像 plt.show() ``` 这个程序框架只是一个基本的示例,你可以根据自己的需求进行修改和优化。例如,你可以添加交互式控件来控制飞机的姿态,或者使用其他库来实现更复杂的可视化效果。
评论 12
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

RS迷途小书童

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值