programing tip

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

itbloger 2020. 8. 19. 07:48
반응형

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)

다음에 무엇을해야할지 모르겠습니다.


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

여기에 이미지 설명 입력


패치를 사용해야합니다.

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

서브 플롯이 필요하지 않으며 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의 방법 중 하나입니다 .

귀하의 질문은 Matplotlib에 대해 물었지만 아마도 이미지에 직사각형을 그리는 것에 대해 물었을 것입니다.

여기 당신이 알고 싶었던 것을 다루는 또 다른 질문이 있습니다 : PIL을 사용하여 직사각형과 텍스트를 그립니다.

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

반응형