matplotlib: como desenhar um retângulo na imagem

139

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.

python image matplotlib

— KAI ZHAO
fonte

251

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

— tmdavison
fonte

Obrigado pela sua resposta! Funciona, mas parece que o retângulo é desenhado no eixo, não a imagem em si. Se eu tentar salvar a imagem em um arquivo, o retângulo não será salvo. Existe uma maneira de o retângulo substituir os valores de pixel na imagem? Obrigado novamente!

— Yanfeng Liu

Deixa pra lá. Eu encontrei esta ligação e parece estar funcionando:)

— Yanfeng Liu

Se você ainda está obtendo um retângulo preenchido, passe a fill=Falsebandeira paraRectangle

— Ivan Talalaev 22/10

7

Isso é estranho. A documentação para patches.Rectanglediz que os dois primeiros números são The bottom and left rectangle coordinates. Vejo aqui que os dois primeiros números (50.100) correspondem à coordenada TOP e à esquerda do retângulo. Estou confuso.

— Monica Heddneck

1

Não, o retângulo está no lugar certo. Está em coordenadas de dados. Você pode alterar a transformar se quiser que ele em coordenadas eixos

— tmdavison

20

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

— Serenidade
fonte

Eu gostei de como você encapsulou eixos dentro de um objeto de figura: os eixos fazem a plotagem, a figura faz coisas de interface de alto nível

— Alex

19

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

— James 'Fofo' Burton
fonte

7

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

— user3731622
fonte