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)

Я не знаю, як діяти далі.

python image matplotlib

— KAI ZHAO
джерело

251

Ви можете додати Rectangleпластир до matplotlib 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()

— tmdavison
джерело

Дякую за вашу відповідь! Це працює, але здається, що прямокутник намальований на осі, а не сама картина. Якщо я спробую зберегти зображення у файл, прямокутник не буде збережено. Чи існує спосіб, щоб прямокутник замінював значення пікселів на зображенні? Знову дякую!

— Янфенг Лю

Не звертай уваги. Я знайшов це посилання і, здається, працює

— :)

Якщо ви все ще заповнюєте прямокутник, передайте fill=FalseпрапорRectangle

— Івану Талалаєву

7

Це дивно. У документації на patches.Rectangleте, що перші два числа є The bottom and left rectangle coordinates. Я бачу тут, що перші два числа (50100) відповідають ТОП та лівій координаті прямокутника. Я збентежений.

— Моніка Хеднек

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

— Безтурботність
джерело

Мені сподобалось, як ви інкапсульовані осі всередині фігурного об’єкта: осі роблять графік, фігура робить інтерфейс високого рівня

— 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 ImageDraw , OpenCV або щось подібне.

Ось метод ImageDraw PIL для малювання прямокутника .

Ось один із методів OpenCV для малювання прямокутника .

Ваше запитання задавало питання про Matplotlib, але, напевно, варто було просто запитати про нанесення прямокутника на зображення.

Ось ще одне питання, яке стосується того, що я думаю, що ви хотіли знати: Намалюйте прямокутник і текст у ньому за допомогою PIL

— користувач3731622
джерело