matplotlib: como desenhar um retângulo na imagem

Como desenhar um retângulo em uma imagem, desta forma:

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)

Eu não sei como proceder.

Você pode adicionar um Rectanglepatch aos eixos do matplotlib.

Por exemplo (usando a imagem do tutorial aqui ):

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

Você precisa usar patches.

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')

Não há necessidade de subtramas, e o pyplot pode exibir imagens PIL, portanto, isso pode ser simplificado ainda mais:

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)

Ou, a versão curta:

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'))

Pelo meu entendimento, o matplotlib é uma biblioteca de plotagem.

Se você quiser alterar os dados da imagem (por exemplo, desenhar um retângulo em uma imagem), use o ImageDraw , OpenCV ou algo semelhante do PIL .

Aqui está o método ImageDraw do PIL para desenhar um retângulo .

Aqui está um dos métodos do OpenCV para desenhar um retângulo .

Sua pergunta foi feita sobre o Matplotlib, mas provavelmente deveria ter sido perguntada sobre o desenho de um retângulo em uma imagem.

Aqui está outra pergunta que aborda o que eu acho que você queria saber: Desenhe um retângulo e um texto usando PIL