matplotlib:如何在图像上绘制矩形


139

如何在图像上绘制矩形,如下所示: 在此处输入图片说明

import matplotlib.pyplot as plt
from PIL import Image
import numpy as np
im = np.array(Image.open('dog.png'), dtype=np.uint8)
plt.imshow(im)

我不知道该如何进行。

Answers:


251

您可以将Rectangle补丁添加到matplotlib轴。

例如(在此处使用教程中的图像):

import matplotlib.pyplot as plt
import matplotlib.patches as patches
from PIL import Image
import numpy as np

im = np.array(Image.open('stinkbug.png'), dtype=np.uint8)

# Create figure and axes
fig,ax = plt.subplots(1)

# Display the image
ax.imshow(im)

# Create a Rectangle patch
rect = patches.Rectangle((50,100),40,30,linewidth=1,edgecolor='r',facecolor='none')

# Add the patch to the Axes
ax.add_patch(rect)

plt.show()

在此处输入图片说明


感谢您的回答!它可以工作,但似乎矩形是在轴上绘制的,而不是图片本身。如果我尝试将图像保存到文件中,则不会保存矩形。有没有办法让矩形替换图像上的像素值?再次感谢!
刘彦峰

没关系。我找到了此链接,它似乎正在起作用
Yanyan Liu

如果您仍在填充矩形,fill=FalseRectangle
请将

7
真奇怪 的文档patches.Rectangle说前两个数字是The bottom and left rectangle coordinates。我在这里看到前两个数字(50,100)对应于矩形的顶部和左坐标。我很困惑。
Monica Heddneck '18

1
不,矩形在正确的位置。在数据坐标中。如果需要轴坐标,则可以更改变换
tmdavison

20

您需要使用补丁。

import matplotlib.pyplot as plt
import matplotlib.patches as patches

fig2 = plt.figure()
ax2 = fig2.add_subplot(111, aspect='equal')

ax2.add_patch(
     patches.Rectangle(
        (0.1, 0.1),
        0.5,
        0.5,
        fill=False      # remove background
     ) ) 
fig2.savefig('rect2.png', dpi=90, bbox_inches='tight')

我喜欢将轴封装在图形对象中的方式:轴进行绘图,图形进行高级界面处理
Alex

19

不需要子图,并且pyplot可以显示PIL图像,因此可以进一步简化:

import matplotlib.pyplot as plt
from matplotlib.patches import Rectangle
from PIL import Image

im = Image.open('stinkbug.png')

# Display the image
plt.imshow(im)

# Get the current reference
ax = plt.gca()

# Create a Rectangle patch
rect = Rectangle((50,100),40,30,linewidth=1,edgecolor='r',facecolor='none')

# Add the patch to the Axes
ax.add_patch(rect)

或者,简短版本:

import matplotlib.pyplot as plt
from matplotlib.patches import Rectangle
from PIL import Image

# Display the image
plt.imshow(Image.open('stinkbug.png'))

# Add the patch to the Axes
plt.gca().add_patch(Rectangle((50,100),40,30,linewidth=1,edgecolor='r',facecolor='none'))

7

据我了解,matplotlib是一个绘图库。

如果要更改图像数据(例如,在图像上绘制矩形),则可以使用PIL的ImageDrawOpenCV或类似的东西。

这是PIL的ImageDraw方法来绘制矩形

这是OpenCV绘制矩形的方法之一

您的问题询问了有关Matplotlib的问题,但可能应该只是询问有关在图像上绘制矩形的问题。

这是另一个解决我想知道的问题的问题: 使用PIL在其中绘制一个矩形和一个文本

By using our site, you acknowledge that you have read and understand our Cookie Policy and Privacy Policy.
Licensed under cc by-sa 3.0 with attribution required.