matplotlib: cómo dibujar un rectángulo en la imagen

139

Cómo dibujar un rectángulo en una imagen, así: ingrese la descripción de la imagen aquí

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)

No se como proceder.

python image matplotlib KAI ZHAO
fuente

Respuestas:

251

Puede agregar un Rectangleparche a los ejes matplotlib.

Por ejemplo (usando la imagen del tutorial aquí ):

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

ingrese la descripción de la imagen aquí

tmdavison
fuente
¡Gracias por tu respuesta! Funciona, pero parece que el rectángulo se dibuja en el eje, no en la imagen en sí. Si intento guardar la imagen en un archivo, el rectángulo no se guardará. ¿Hay alguna manera de que el rectángulo reemplace los valores de píxeles en la imagen? ¡Gracias de nuevo!
Yanfeng Liu
No importa. Encontré este enlace y parece estar funcionando
:)
Si todavía está llenando un rectángulo, pase la fill=Falsebandera aRectangle
Ivan Talalaev el
77
Esto es raro La documentación de patches.Rectangledice que los dos primeros números son The bottom and left rectangle coordinates. Veo aquí que los dos primeros números (50,100) corresponden a la coordenada SUPERIOR e izquierda del rectángulo. Estoy confundido.
Monica Heddneck
1
No, el rectángulo está en el lugar correcto. Está en coordenadas de datos. Puede cambiar la transformación si lo desea en coordenadas de ejes
tmdavison el
20

Necesitas usar parches.

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')
Serenidad
fuente
Me gustó cómo encapsulaste los ejes dentro de un objeto de figura: los ejes hacen el trazado, la figura hace cosas de interfaz de alto nivel
Alex
19

No hay necesidad de subtramas, y pyplot puede mostrar imágenes PIL, por lo que esto se puede simplificar aún más:

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)

O la versión corta:

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
fuente
7

Desde mi entendimiento matplotlib es una biblioteca de trazado.

Si desea cambiar los datos de la imagen (por ejemplo, dibujar un rectángulo en una imagen), puede usar ImageDraw , OpenCV de PIL o algo similar.

Aquí está el método ImageDraw de PIL para dibujar un rectángulo .

Este es uno de los métodos de OpenCV para dibujar un rectángulo. .

Su pregunta se refería a Matplotlib, pero probablemente debería haberle preguntado sobre dibujar un rectángulo en una imagen.

Aquí hay otra pregunta que aborda lo que creo que querías saber: dibuja un rectángulo y un texto con PIL

usuario3731622
fuente