61 lines
1.8 KiB
Python
61 lines
1.8 KiB
Python
|
import os
|
|||
|
|
import argparse
|
|||
|
|
from osgeo import ogr,gdal
|
|||
|
|
from matplotlib import pyplot as plt
|
|||
|
|
from osgeo import gdal
|
|||
|
|
import matplotlib
|
|||
|
|
import matplotlib.patches as patches
|
|||
|
|
from osgeo import gdal
|
|||
|
|
from PIL import Image
|
|||
|
|
from scipy.spatial import cKDTree
|
|||
|
|
import numpy as np
|
|||
|
|
from tools.DotaOperator import DotaObj,createDota,readDotaFile,writerDotaFile
|
|||
|
|
import argparse
|
|||
|
|
import math
|
|||
|
|
from math import ceil, floor
|
|||
|
|
|
|||
|
|
pngpath=r"D:\港口\切片结果\Geo_bc2-sm-org-vv-20231016t135315-008424-0020e8-01_3.png"
|
|||
|
|
txtpath=r"D:\港口\切片结果\Geo_bc2-sm-org-vv-20231016t135315-008424-0020e8-01_3.txt"
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
pngdata=np.array(Image.open(pngpath))
|
|||
|
|
dotalist=readDotaFile(txtpath)
|
|||
|
|
|
|||
|
|
plt.figure()
|
|||
|
|
plt.imshow(pngdata)
|
|||
|
|
# 绘制每个目标的矩形框并标注坐标
|
|||
|
|
for i in range(len(dotalist)):
|
|||
|
|
# 提取x和y坐标
|
|||
|
|
x_coords = [dotalist[i].x1, dotalist[i].x2, dotalist[i].x3, dotalist[i].x4]
|
|||
|
|
y_coords = [dotalist[i].y1, dotalist[i].y2, dotalist[i].y3, dotalist[i].y4]
|
|||
|
|
|
|||
|
|
# 计算最小外接矩形（AABB）
|
|||
|
|
x_min, x_max = min(x_coords), max(x_coords)
|
|||
|
|
y_min, y_max = min(y_coords), max(y_coords)
|
|||
|
|
width = x_max - x_min
|
|||
|
|
height = y_max - y_min
|
|||
|
|
|
|||
|
|
# 绘制无填充矩形框（仅红色边框）
|
|||
|
|
rect = patches.Rectangle(
|
|||
|
|
(x_min, y_min), width, height,
|
|||
|
|
linewidth=2, edgecolor='red', facecolor='none' # 关键：facecolor='none'
|
|||
|
|
)
|
|||
|
|
plt.gca().add_patch(rect)
|
|||
|
|
|
|||
|
|
# ax.annotate(f'({x},{y})', xy=(x, y), xytext=(5, 5),
|
|||
|
|
# textcoords='offset points', fontsize=10,
|
|||
|
|
# bbox=dict(boxstyle='round,pad=0.5', fc='white', alpha=0.8))
|
|||
|
|
|
|||
|
|
# 在矩形中心标注目标编号
|
|||
|
|
center_x = sum(x_coords) / 4
|
|||
|
|
center_y = sum(y_coords) / 4
|
|||
|
|
plt.text(center_x, center_y, str(i),
|
|||
|
|
ha='center', va='center', fontsize=6, color='red')
|
|||
|
|
|
|||
|
|
|
|||
|
|
plt.show()
|
|||
|
|
|
|||
|
|
|
|||
|
|
|