web-dev-qa-db-ja.com

matplotlib:画像上に四角形を描く方法

このように、画像上に四角形を描画する方法: enter image description here

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)

次に何をすべきかわかりません。

87
KAI ZHAO

Matplotlib Axesに Rectangle パッチを追加することができます。

例えば(チュートリアルの画像を使う こちら ):

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

enter image description here

152
tmdavison

あなたはパッチを使う必要があります。

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

サブプロットは不要で、pyplotはPILイメージを表示することができるので、これはさらに単純化することができます。

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)

または、ショートバージョン:

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

私の理解 からmatplotlib はプロットライブラリです。

画像データを変更したい場合(例えば、画像上に長方形を描く)、 PILのImageDrawを使うことができます。 )OpenCV 、または似たようなもの。

これは 四角形を描画するPILのImageDrawメソッドです

これは OpenCVの四角形の描画方法の1つです

あなたの質問はMatplotlibについて尋ねましたが、おそらく画像上に長方形を描くことについて尋ねたはずです。

これは私があなたが知りたいと思うことに対処するもう一つの質問です: PILを使って長方形とその中にテキストを描きます

5
user3731622