绘制效果如下:
左图为原图,右图为局部放大图。选中原图中的一小部分矩形区域框出,然后放大,将原图和放大图都绘制在同一个画布上,并将对角线采用直线连接。
代码可以直接使用,只需要替换一些参数即可,部分参数设置解释:
color = (255, 0, 0) # 颜色
folder_path = r"originals_path" # 待处理图像文件目录
folder_path2 = r"rectangles_path" # 处理后图像保存目录
resize = (700,457) # 请根据原图像素大小自行设置
x1, y1 = 110, 140 # 原图中局部放大区域左上角坐标
x2, y2 = 210, 210 # 原图中局部放大区域右下角坐标
linewidth = 2 # 绘制线粗细
zoom_array = [0.55, 0.10, 0.4, 0.4] # 即[A,B,C,D]
如下图所示:A、B用于固定局部放大图左下角的位置,C、D指定局部放大图大小。这里的A、B、C、D都是相对于整个画布的比例。A越大,局部放大图越靠近右边;B越大,越靠近上方;C、D越大,局部放大图相对于整个画布越大。请根据实际情况多多尝试,以找出合适的取值。
绘制两条连接线的参数与代码:
# 绘制线连接原始图像和放大图
line = ConnectionPatch(xyA=(x1, y2), xyB=(0, 0), coordsA="data", coordsB="axes fraction", axesA=ax, axesB=zoom_ax, color=color, linewidth=linewidth)
fig.add_artist(line)# xyA:原图中局部放大区域的顶点坐标
# xyB:局部放大图的顶点坐标
# 这里的意思是连接原图中的(x1, y2)点与局部放大图的(0, 0)点
根据局部放大图的位置,这部分参数可能需要调整。具体取值和连接效果可参考下图:
完整代码如下,可以根据需求省略或补充部分代码:
import matplotlib.pyplot as plt
import matplotlib.patches as patches
from matplotlib.patches import ConnectionPatch
import numpy as np
from PIL import Image
import os
# 待修改参数
color = (255, 0, 0)
folder_path = r"originals_path"
folder_path2 = r"rectangles_path"
resize = (700,457)
# 定义局部放大区域的坐标
x1, y1 = 110, 140
x2, y2 = 210, 210
linewidth = 2
zoom_array = [0.55, 0.10, 0.4, 0.4] # x y 子图左下角相对于整个画幅的位置, 后两位子图大小
def normalize_color(color):
normalized_color = tuple(channel / 255.0 for channel in color)
return normalized_color
color = normalize_color(color)
# 遍历文件夹中的每个图像文件
for filename in os.listdir(folder_path):
if filename.endswith(".png"):
image_path = os.path.join(folder_path, filename)
image = Image.open(image_path)
image = image.resize(resize)
image_array = np.array(image)
# 获取局部放大区域的图像数据
zoomed_region = image_array[y1:y2, x1:x2]
fig, ax = plt.subplots()
# 绘制原始图像
ax.imshow(image_array)
ax.axis('off')
# 绘制局部放大区域的边界框
rect = patches.Rectangle((x1, y1), x2-x1, y2-y1, linewidth=linewidth, edgecolor=color, facecolor='none')
ax.add_patch(rect)
# 创建放大图
zoom_ax = fig.add_axes(zoom_array)
zoom_ax.imshow(zoomed_region)
# 设置放大图的边框,并增加线宽
zoom_ax.spines['top'].set_color(color)
zoom_ax.spines['top'].set_linewidth(linewidth)
zoom_ax.spines['bottom'].set_color(color)
zoom_ax.spines['bottom'].set_linewidth(linewidth)
zoom_ax.spines['left'].set_color(color)
zoom_ax.spines['left'].set_linewidth(linewidth)
zoom_ax.spines['right'].set_color(color)
zoom_ax.spines['right'].set_linewidth(linewidth)
# 隐藏放大图的坐标轴
zoom_ax.set_xticks([])
zoom_ax.set_yticks([])
# 绘制线连接原始图像和放大图
line = ConnectionPatch(xyA=(x1, y2), xyB=(0, 0), coordsA="data", coordsB="axes fraction", axesA=ax, axesB=zoom_ax, color=color, linewidth=linewidth-1)
fig.add_artist(line)
line2 = ConnectionPatch(xyA=(x2, y1), xyB=(1, 1), coordsA="data", coordsB="axes fraction", axesA=ax, axesB=zoom_ax, color=color, linewidth=linewidth-1)
fig.add_artist(line2)
# plt.show()
# 保存图像
save_path = os.path.join(folder_path2, f"zoomed_{filename}")
plt.savefig(save_path,transparent=True,dpi=300,bbox_inches='tight')