修改交互式删除点
This commit is contained in:
along
parent
a3b03de856
commit
ab82c81192
235
gps_selector.py
Normal file
235
gps_selector.py
Normal file
@ -0,0 +1,235 @@
|
||||
from utils.gps_extractor import GPSExtractor
|
||||
import os
|
||||
import sys
|
||||
import shutil
|
||||
from pathlib import Path
|
||||
import matplotlib.pyplot as plt
|
||||
from matplotlib.widgets import RectangleSelector
|
||||
import pandas as pd
|
||||
from matplotlib.font_manager import FontProperties
|
||||
|
||||
|
||||
class GPSSelector:
|
||||
def __init__(self, image_dir: str, output_dir: str = None):
|
||||
# 移除中文字体设置
|
||||
self.image_dir = image_dir
|
||||
self.output_dir = output_dir
|
||||
self.gps_points = None
|
||||
self.selected_points = []
|
||||
self.fig, self.ax = plt.subplots(figsize=(12, 8))
|
||||
self.scatter = None
|
||||
self.rs = None
|
||||
self.setup_plot()
|
||||
|
||||
def extract_gps(self):
|
||||
"""提取GPS数据"""
|
||||
extractor = GPSExtractor(self.image_dir)
|
||||
self.gps_points = extractor.extract_all_gps()
|
||||
print(f"成功提取 {len(self.gps_points)} 个GPS点")
|
||||
|
||||
def setup_plot(self):
|
||||
"""设置绘图"""
|
||||
self.ax.set_title('GPS Points - Use mouse to drag and select points to delete')
|
||||
self.ax.set_xlabel('Longitude')
|
||||
self.ax.set_ylabel('Latitude')
|
||||
self.ax.grid(True)
|
||||
|
||||
# 设置坐标轴使用相同的比例
|
||||
self.ax.set_aspect('equal')
|
||||
|
||||
# 设置矩形选择器
|
||||
self.rs = RectangleSelector(
|
||||
self.ax, self.on_select,
|
||||
interactive=True,
|
||||
useblit=True,
|
||||
button=[1], # 只响应左键
|
||||
props=dict(facecolor='red', alpha=0.3)
|
||||
)
|
||||
|
||||
# 添加按钮回调
|
||||
self.fig.canvas.mpl_connect('key_press_event', self.on_key_press)
|
||||
|
||||
# 添加缩放和平移功能
|
||||
self.fig.canvas.mpl_connect('scroll_event', self.on_scroll)
|
||||
self.fig.canvas.mpl_connect('button_press_event', self.on_press)
|
||||
self.fig.canvas.mpl_connect('button_release_event', self.on_release)
|
||||
self.fig.canvas.mpl_connect('motion_notify_event', self.on_motion)
|
||||
|
||||
# 用于平移功能的变量
|
||||
self._pan_start = None
|
||||
|
||||
def plot_gps_points(self):
|
||||
"""绘制GPS点"""
|
||||
if self.scatter is not None:
|
||||
self.scatter.remove()
|
||||
|
||||
# 计算经纬度的范围
|
||||
lon_range = self.gps_points['lon'].max() - self.gps_points['lon'].min()
|
||||
lat_range = self.gps_points['lat'].max() - self.gps_points['lat'].min()
|
||||
|
||||
# 设置合适的图形大小，保持经纬度的真实比例
|
||||
aspect_ratio = lon_range / lat_range
|
||||
fig_width = 12
|
||||
fig_height = fig_width / aspect_ratio
|
||||
self.fig.set_size_inches(fig_width, fig_height)
|
||||
|
||||
self.scatter = self.ax.scatter(
|
||||
self.gps_points['lon'],
|
||||
self.gps_points['lat'],
|
||||
c='blue',
|
||||
s=20,
|
||||
alpha=0.6
|
||||
)
|
||||
|
||||
# 设置适当的显示范围，添加一些边距
|
||||
margin = 0.1
|
||||
x_margin = lon_range * margin
|
||||
y_margin = lat_range * margin
|
||||
|
||||
self.ax.set_xlim([
|
||||
self.gps_points['lon'].min() - x_margin,
|
||||
self.gps_points['lon'].max() + x_margin
|
||||
])
|
||||
self.ax.set_ylim([
|
||||
self.gps_points['lat'].min() - y_margin,
|
||||
self.gps_points['lat'].max() + y_margin
|
||||
])
|
||||
|
||||
# 关闭自动缩放
|
||||
self.ax.autoscale(False)
|
||||
|
||||
# 使用更精确的刻度
|
||||
self.ax.ticklabel_format(useOffset=False, style='plain')
|
||||
|
||||
self.fig.canvas.draw_idle()
|
||||
|
||||
def on_select(self, eclick, erelease):
|
||||
"""矩形选择回调"""
|
||||
x1, y1 = eclick.xdata, eclick.ydata
|
||||
x2, y2 = erelease.xdata, erelease.ydata
|
||||
|
||||
# 获取选中区域内的点
|
||||
mask = (
|
||||
(self.gps_points['lon'] >= min(x1, x2)) &
|
||||
(self.gps_points['lon'] <= max(x1, x2)) &
|
||||
(self.gps_points['lat'] >= min(y1, y2)) &
|
||||
(self.gps_points['lat'] <= max(y1, y2))
|
||||
)
|
||||
|
||||
selected = self.gps_points[mask]
|
||||
self.selected_points.extend(selected['file'].tolist())
|
||||
|
||||
# 从数据中移除选中的点
|
||||
self.gps_points = self.gps_points[~mask]
|
||||
|
||||
# 更新绘图
|
||||
self.plot_gps_points()
|
||||
print(f"选中 {len(selected)} 个点，剩余 {len(self.gps_points)} 个点")
|
||||
|
||||
def on_key_press(self, event):
|
||||
"""键盘事件回调"""
|
||||
if event.key == 'enter':
|
||||
self.save_results()
|
||||
plt.close()
|
||||
elif event.key == 'escape':
|
||||
plt.close()
|
||||
|
||||
def save_results(self):
|
||||
"""保存结果"""
|
||||
if not self.output_dir:
|
||||
return
|
||||
|
||||
# 创建输出目录
|
||||
os.makedirs(self.output_dir, exist_ok=True)
|
||||
removed_dir = os.path.join(self.output_dir, "removed_images")
|
||||
os.makedirs(removed_dir, exist_ok=True)
|
||||
|
||||
# 移动被删除的图像
|
||||
for img_name in self.selected_points:
|
||||
src = os.path.join(self.image_dir, img_name)
|
||||
dst = os.path.join(removed_dir, img_name)
|
||||
shutil.move(src, dst)
|
||||
|
||||
# 保存剩余点的信息
|
||||
self.gps_points.to_csv(
|
||||
os.path.join(self.output_dir, "remaining_points.csv"),
|
||||
index=False
|
||||
)
|
||||
|
||||
print(f"已移动 {len(self.selected_points)} 张图片到 {removed_dir}")
|
||||
print(f"保留 {len(self.gps_points)} 个点")
|
||||
|
||||
def run(self):
|
||||
"""运行选择器"""
|
||||
self.extract_gps()
|
||||
self.plot_gps_points()
|
||||
plt.show()
|
||||
|
||||
def on_scroll(self, event):
|
||||
"""鼠标滚轮缩放"""
|
||||
if event.inaxes != self.ax:
|
||||
return
|
||||
|
||||
# 获取当前视图范围
|
||||
cur_xlim = self.ax.get_xlim()
|
||||
cur_ylim = self.ax.get_ylim()
|
||||
|
||||
# 缩放因子
|
||||
base_scale = 1.1
|
||||
xdata = event.xdata
|
||||
ydata = event.ydata
|
||||
|
||||
if event.button == 'up':
|
||||
# 放大
|
||||
scale_factor = 1/base_scale
|
||||
else:
|
||||
# 缩小
|
||||
scale_factor = base_scale
|
||||
|
||||
# 设置新的视图范围
|
||||
new_width = (cur_xlim[1] - cur_xlim[0]) * scale_factor
|
||||
new_height = (cur_ylim[1] - cur_ylim[0]) * scale_factor
|
||||
|
||||
self.ax.set_xlim([xdata - new_width * (xdata - cur_xlim[0]) / (cur_xlim[1] - cur_xlim[0]),
|
||||
xdata + new_width * (cur_xlim[1] - xdata) / (cur_xlim[1] - cur_xlim[0])])
|
||||
self.ax.set_ylim([ydata - new_height * (ydata - cur_ylim[0]) / (cur_ylim[1] - cur_ylim[0]),
|
||||
ydata + new_height * (cur_ylim[1] - ydata) / (cur_ylim[1] - cur_ylim[0])])
|
||||
|
||||
self.fig.canvas.draw_idle()
|
||||
|
||||
def on_press(self, event):
|
||||
"""鼠标按下事件"""
|
||||
if event.inaxes != self.ax or event.button != 3: # 只响应右键
|
||||
return
|
||||
self._pan_start = (event.xdata, event.ydata)
|
||||
|
||||
def on_release(self, event):
|
||||
"""鼠标释放事件"""
|
||||
self._pan_start = None
|
||||
|
||||
def on_motion(self, event):
|
||||
"""鼠标移动事件"""
|
||||
if self._pan_start is None or event.inaxes != self.ax:
|
||||
return
|
||||
|
||||
# 计算移动距离
|
||||
dx = event.xdata - self._pan_start[0]
|
||||
dy = event.ydata - self._pan_start[1]
|
||||
|
||||
# 更新视图范围
|
||||
cur_xlim = self.ax.get_xlim()
|
||||
cur_ylim = self.ax.get_ylim()
|
||||
|
||||
self.ax.set_xlim(cur_xlim - dx)
|
||||
self.ax.set_ylim(cur_ylim - dy)
|
||||
|
||||
self.fig.canvas.draw_idle()
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
# 使用示例
|
||||
selector = GPSSelector(
|
||||
image_dir=r"C:\datasets\134\code\images",
|
||||
output_dir=r"C:\datasets\ODM_output\134"
|
||||
)
|
||||
selector.run()
|
||||
@ -1,134 +0,0 @@
|
||||
import os
|
||||
import sys
|
||||
import shutil
|
||||
from pathlib import Path
|
||||
import matplotlib.pyplot as plt
|
||||
from matplotlib.widgets import RectangleSelector
|
||||
import pandas as pd
|
||||
|
||||
# 添加项目根目录到系统路径
|
||||
project_root = str(Path(__file__).parent.parent)
|
||||
sys.path.append(project_root)
|
||||
|
||||
from utils.gps_extractor import GPSExtractor
|
||||
|
||||
class GPSSelector:
|
||||
def __init__(self, image_dir: str, output_dir: str = None):
|
||||
self.image_dir = image_dir
|
||||
self.output_dir = output_dir
|
||||
self.gps_points = None
|
||||
self.selected_points = []
|
||||
self.fig, self.ax = plt.subplots(figsize=(12, 8))
|
||||
self.scatter = None
|
||||
self.rs = None
|
||||
self.setup_plot()
|
||||
|
||||
def extract_gps(self):
|
||||
"""提取GPS数据"""
|
||||
extractor = GPSExtractor(self.image_dir)
|
||||
self.gps_points = extractor.extract_all_gps()
|
||||
print(f"成功提取 {len(self.gps_points)} 个GPS点")
|
||||
|
||||
def setup_plot(self):
|
||||
"""设置绘图"""
|
||||
self.ax.set_title('GPS Points - 使用鼠标拖动选择要删除的点')
|
||||
self.ax.set_xlabel('Longitude')
|
||||
self.ax.set_ylabel('Latitude')
|
||||
self.ax.grid(True)
|
||||
|
||||
# 设置矩形选择器
|
||||
self.rs = RectangleSelector(
|
||||
self.ax, self.on_select,
|
||||
interactive=True,
|
||||
useblit=True,
|
||||
button=[1], # 只响应左键
|
||||
props=dict(facecolor='red', alpha=0.3)
|
||||
)
|
||||
|
||||
# 添加按钮回调
|
||||
self.fig.canvas.mpl_connect('key_press_event', self.on_key_press)
|
||||
|
||||
def plot_gps_points(self):
|
||||
"""绘制GPS点"""
|
||||
if self.scatter is not None:
|
||||
self.scatter.remove()
|
||||
|
||||
self.scatter = self.ax.scatter(
|
||||
self.gps_points['lon'],
|
||||
self.gps_points['lat'],
|
||||
c='blue',
|
||||
s=20,
|
||||
alpha=0.6
|
||||
)
|
||||
self.fig.canvas.draw_idle()
|
||||
|
||||
def on_select(self, eclick, erelease):
|
||||
"""矩形选择回调"""
|
||||
x1, y1 = eclick.xdata, eclick.ydata
|
||||
x2, y2 = erelease.xdata, erelease.ydata
|
||||
|
||||
# 获取选中区域内的点
|
||||
mask = (
|
||||
(self.gps_points['lon'] >= min(x1, x2)) &
|
||||
(self.gps_points['lon'] <= max(x1, x2)) &
|
||||
(self.gps_points['lat'] >= min(y1, y2)) &
|
||||
(self.gps_points['lat'] <= max(y1, y2))
|
||||
)
|
||||
|
||||
selected = self.gps_points[mask]
|
||||
self.selected_points.extend(selected['file'].tolist())
|
||||
|
||||
# 从数据中移除选中的点
|
||||
self.gps_points = self.gps_points[~mask]
|
||||
|
||||
# 更新绘图
|
||||
self.plot_gps_points()
|
||||
print(f"选中 {len(selected)} 个点，剩余 {len(self.gps_points)} 个点")
|
||||
|
||||
def on_key_press(self, event):
|
||||
"""键盘事件回调"""
|
||||
if event.key == 'enter':
|
||||
self.save_results()
|
||||
plt.close()
|
||||
elif event.key == 'escape':
|
||||
plt.close()
|
||||
|
||||
def save_results(self):
|
||||
"""保存结果"""
|
||||
if not self.output_dir:
|
||||
return
|
||||
|
||||
# 创建输出目录
|
||||
os.makedirs(self.output_dir, exist_ok=True)
|
||||
removed_dir = os.path.join(self.output_dir, "removed_images")
|
||||
os.makedirs(removed_dir, exist_ok=True)
|
||||
|
||||
# 移动被删除的图像
|
||||
for img_name in self.selected_points:
|
||||
src = os.path.join(self.image_dir, img_name)
|
||||
dst = os.path.join(removed_dir, img_name)
|
||||
shutil.move(src, dst)
|
||||
|
||||
# 保存剩余点的信息
|
||||
self.gps_points.to_csv(
|
||||
os.path.join(self.output_dir, "remaining_points.csv"),
|
||||
index=False
|
||||
)
|
||||
|
||||
print(f"已移动 {len(self.selected_points)} 张图片到 {removed_dir}")
|
||||
print(f"保留 {len(self.gps_points)} 个点")
|
||||
|
||||
def run(self):
|
||||
"""运行选择器"""
|
||||
self.extract_gps()
|
||||
self.plot_gps_points()
|
||||
plt.show()
|
||||
|
||||
|
||||
if __name__ == "__main__":
|
||||
# 使用示例
|
||||
selector = GPSSelector(
|
||||
image_dir=r"E:\datasets\UAV\1619\project\images",
|
||||
output_dir=r"E:\datasets\UAV\1619\filtered"
|
||||
)
|
||||
selector.run()
|
||||
Loading…
Reference in New Issue
Block a user