matplotlib: come disegnare un rettangolo sull'immagine

Question

Come disegnare un rettangolo su un'immagine, come questo:

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) 

non so che cosa fare il prossimo.

Answer 1

È possibile aggiungere una patch Rectangle agli assi matplotlib.

Per esempio (utilizzando l'immagine dal tutorial here):

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

Answer 2

È necessario utilizzare le patch.

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

Altri esempi, si possono trovare qui: http://matthiaseisen.com/pp/patterns/p0203/