matplotlib:画像に長方形を描く方法


139

次のように、画像に長方形を描画する方法: ここに画像の説明を入力してください

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)

どうすればいいのかわかりません。

回答:


251

Rectanglematplotlib Axesにパッチを追加できます。

例(ここのチュートリアルの画像を使用):

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

ここに画像の説明を入力してください


ご回答有難うございます!動作しますが、長方形は軸自体に描かれているように見えますが、画像自体ではありません。画像をファイルに保存しようとすると、長方形が保存されません。長方形が画像のピクセル値を置き換える方法はありますか?再度、感謝します!
Yanfeng Liu

気にしないで。私はこのリンクを見つけ、それは機能しているようです:)
Liu

それでも長方形が塗りつぶされている場合は、fill=Falseフラグを渡し てくださいRectangle
Ivan Talalaev

7
これは奇妙です。のドキュメントにpatches.Rectangleは、最初の2つの数値はと書かれていますThe bottom and left rectangle coordinates。ここで、最初の2つの数値(50,100)が長方形の上部座標と左座標に対応していることがわかります。よくわかりません。
モニカヘドネック2018年

1
いいえ、長方形は正しい場所にあります。データ座標にあります。座標軸で変換が必要な場合は、変換を変更できます
tmdavison

20

パッチを使用する必要があります。

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

AxesをFigureオブジェクト内にカプセル化する方法が気に入りました。Axesはプロットを実行し、Figureは高レベルのインターフェイスを実行します
Alex

19

サブプロットの必要はなく、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'))

7

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

画像データを変更したい場合(たとえば、画像上に長方形を描画する場合)は、PILのImageDrawOpenCVなどを使用できます。

長方形を描画するためのPILのImageDrawメソッドを次に示します

以下は、長方形を描画するためOpenCVのメソッドの 1つです。

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

ここに、あなたが知りたいと思ったことに対処する別の質問があります 。PILを使用して長方形とその中にテキストを描画します。

弊社のサイトを使用することにより、あなたは弊社のクッキーポリシーおよびプライバシーポリシーを読み、理解したものとみなされます。
Licensed under cc by-sa 3.0 with attribution required.