matplotlib: comment dessiner un rectangle sur l'image

139

Comment dessiner un rectangle sur une image, comme ceci:

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)

Je ne sais pas comment procéder.

python image matplotlib

— KAI ZHAO
source

251

Vous pouvez ajouter un Rectanglepatch aux axes matplotlib.

Par exemple (en utilisant l'image du tutoriel ici ):

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
source

Merci pour votre réponse! Cela fonctionne, mais il semble que le rectangle est dessiné sur l'axe, pas l'image elle-même. Si j'essaye d'enregistrer l'image dans un fichier, le rectangle ne sera pas enregistré. Existe-t-il un moyen pour que le rectangle remplace les valeurs de pixel sur l'image? Merci encore!

— Yanfeng Liu

Ça ne fait rien. J'ai trouvé ce lien et il semble fonctionner

— :)

Si vous obtenez toujours un rectangle rempli, passez le fill=Falsedrapeau àRectangle

— Ivan Talalaev

7

C'est bizarre. La documentation pour patches.Rectangledit que les deux premiers nombres sont The bottom and left rectangle coordinates. Je vois ici que les deux premiers nombres, (50,100), correspondent à la coordonnée HAUT et à gauche du rectangle. Je suis confus.

— Monica Heddneck

1

Non, le rectangle est au bon endroit. C'est en coordonnées de données. Vous pouvez modifier la transformation si vous le souhaitez en coordonnées d'axes

— tmdavison

20

Vous devez utiliser des correctifs.

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

— Sérénité
source

J'ai aimé la façon dont vous encapsulez les axes dans un objet figure: les axes font le tracé, la figure fait des trucs d'interface de haut niveau

— Alex

19

Il n'y a pas besoin de sous-graphiques et pyplot peut afficher des images PIL, cela peut donc être simplifié davantage:

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, la version courte:

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 Burton 'Fluffy'
source

7

D'après ma compréhension matplotlib est une bibliothèque de traçage.

Si vous souhaitez modifier les données de l'image (par exemple dessiner un rectangle sur une image), vous pouvez utiliser ImageDraw , OpenCV de PIL ou quelque chose de similaire.

Voici la méthode ImageDraw de PIL pour dessiner un rectangle .

Voici l'une des méthodes d' OpenCV pour dessiner un rectangle .

Votre question portait sur Matplotlib, mais aurait probablement dû simplement poser une question sur le dessin d'un rectangle sur une image.

Voici une autre question qui répond à ce que je pense que vous vouliez savoir: dessinez un rectangle et un texte dedans en utilisant PIL

— utilisateur3731622
source