52 lines
1.8 KiB
Python
52 lines
1.8 KiB
Python
import os
|
|
import sys
|
|
sys.path.append(os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
|
|
|
|
import matplotlib.pyplot as plt
|
|
from preprocess.gps_extractor import GPSExtractor
|
|
|
|
DATASET = r'E:\datasets\UAV\502\project\images'
|
|
|
|
if __name__ == '__main__':
|
|
extractor = GPSExtractor(DATASET)
|
|
gps_points = extractor.extract_all_gps()
|
|
|
|
# 创建两个子图
|
|
fig, (ax1, ax2) = plt.subplots(1, 2, figsize=(20, 8))
|
|
|
|
# 左图：原始散点图
|
|
ax1.scatter(gps_points['lon'], gps_points['lat'],
|
|
color='blue', marker='o', label='GPS Points')
|
|
ax1.set_title("GPS Coordinates of Images", fontsize=14)
|
|
ax1.set_xlabel("Longitude", fontsize=12)
|
|
ax1.set_ylabel("Latitude", fontsize=12)
|
|
ax1.grid(True)
|
|
ax1.legend()
|
|
|
|
# # 右图：按时间排序的轨迹图
|
|
# gps_points_sorted = gps_points.sort_values('date')
|
|
|
|
# # 绘制飞行轨迹线
|
|
# ax2.plot(gps_points_sorted['lon'][300:600], gps_points_sorted['lat'][300:600],
|
|
# color='blue', linestyle='-', linewidth=1, alpha=0.6)
|
|
|
|
# # 绘制GPS点
|
|
# ax2.scatter(gps_points_sorted['lon'][300:600], gps_points_sorted['lat'][300:600],
|
|
# color='red', marker='o', s=30, label='GPS Points')
|
|
|
|
# 标记起点和终点
|
|
# ax2.scatter(gps_points_sorted['lon'].iloc[0], gps_points_sorted['lat'].iloc[0],
|
|
# color='green', marker='^', s=100, label='Start')
|
|
# ax2.scatter(gps_points_sorted['lon'].iloc[-1], gps_points_sorted['lat'].iloc[-1],
|
|
# color='purple', marker='s', s=100, label='End')
|
|
|
|
ax2.set_title("UAV Flight Trajectory", fontsize=14)
|
|
ax2.set_xlabel("Longitude", fontsize=12)
|
|
ax2.set_ylabel("Latitude", fontsize=12)
|
|
ax2.grid(True)
|
|
ax2.legend()
|
|
|
|
# 调整子图之间的间距
|
|
plt.tight_layout()
|
|
plt.show()
|