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

Question

如何在图像上绘制矩形，如下所示：

 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)

我不知道该怎么做。

原文由 Kai ZHAO 发布，翻译遵循 CC BY-SA 4.0 许可协议

Answer 1

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

例如（使用 此处 教程中的图像）：

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

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

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

# 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()

原文由 tmdavison 发布，翻译遵循 CC BY-SA 4.0 许可协议

Answer 2

不需要子图，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'))

原文由 James ‘Fluffy’ Burton 发布，翻译遵循 CC BY-SA 4.0 许可协议