matplotlib: 이미지에 직사각형을 그리는 방법

matplotlib: 이미지에 직사각형을 그리는 방법

다음과 같이 이미지에 직사각형을 그리는 방법:

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)

어떻게 해야 할지 모르겠어요.

matplotlib 축에 패치를 추가할 수 있습니다.

예를 들어(여기서 튜토리얼의 이미지를 사용):

import matplotlib.pyplot as plt
import matplotlib.patches as patches
from PIL import Image

im = Image.open('stinkbug.png')

# Create figure and axes
fig, ax = plt.subplots()

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

하위 플롯이 필요하지 않으며 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'))

패치를 사용해야 합니다.

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

내가 알기로는 matplotlib는 플롯 라이브러리이다.

이미지 데이터를 변경하려면(예: 이미지에 직사각형 그리기) PIL의 ImageDraw, OpenCV 등을 사용할 수 있습니다.

직사각형을 그리는 PIL의 ImageDraw 메서드는 다음과 같습니다.

다음은 직사각형을 그리는 OpenCV의 방법 중 하나입니다.

Matplotlib에 대해 질문했지만 이미지에 직사각형을 그리는 것에 대해 질문했어야 합니다.

알고 싶은 것에 대처하는 또 다른 질문이 있습니다: PIL을 사용하여 직사각형과 텍스트를 그 안에 그립니다.

순서가 지정된 점의 좌표 세트가 있는 경우 함수를 사용하여 직선 패치를 사용하지 않고 직접 표시할 수도 있습니다.다음으로 @tmdavison이 제안한 예를 나타냅니다.

import matplotlib.pyplot as plt
import matplotlib.patches as patches
from PIL import Image

im = Image.open('/content/stinkbug.png')

# Create figure and axes
fig, ax = plt.subplots()

# Display the image
ax.imshow(im)

# Coordinates of rectangle vertices
# in clockwise order
xs = [50, 90, 90, 50, 50]
ys = [100, 100, 130, 130, 100]
ax.plot(xs, ys, color="red")

plt.show()

언급URL : https://stackoverflow.com/questions/37435369/matplotlib-how-to-draw-a-rectangle-on-image