Нарисуйте изображение, затем покажите и сохраните результат
Вот что я хочу сделать:
- Прочитайте изображение с диска (в моем случае PNG)
- Нарисуйте в нем какие-нибудь фигуры, например прямоугольники,...
- Отобразите полученное изображение в моем блокноте Jupyter, а также сохраните его обратно в файл.
В настоящее время я использую matplotlib для рисования. Проблема: этот процесс не сохраняет разрешение. Полученный png-файл, записанный в файл, масштабируется на некоторую случайную величину. Насколько я понимаю, это связано с тем, что matplotlib использует физические единицы измерения, и все зависит от размера экрана и т. д. Это проблема для меня, потому что я хочу иметь идеальное по пикселям рисование фигур и рисовать/сохранять изображение в точно таком же разрешении. . Я начинаю думать, что matplotlib может быть неподходящим инструментом для того, что я хочу сделать.
Любой совет? Ниже приведен код, который я сейчас использую.
import matplotlib.pyplot as plt
import matplotlib.image as mpimg
import matplotlib.patches as patches
# Read the image.
image_path = 'input_image.png'
output_image_path = 'output_image.png'
image = mpimg.imread(image_path)
# Create a figure.
fig, ax = plt.subplots(1)
ax.axis('off')
# Create a rectangle patch and add it.
rect = patches.Rectangle((100, 100), 20, 20, linewidth=1, edgecolor='r', facecolor='none')
ax.add_patch(rect)
ax.imshow(image)
# Save result.
plt.savefig(output_image_path, bbox_inches='tight', pad_inches=0)
# Input image: (1200x800)
# Output image: (496x330)
Ответы 1
В вашем коде нет ничего плохого. Причиной нежелательного вывода является неадекватная функциональность используемых вами библиотек Python. Поэтому я бы рекомендовал вам использовать библиотеку PIL, сохраняющую разрешение изображения. Я попытался переписать приведенный выше код, используя библиотеку PIL, и убедился, что в разрешении нет существенных потерь.
from PIL import Image, ImageDraw
import matplotlib.pyplot as plt
# Read the image.
image_path = 'sunset.jpeg' #Change this to your own image
output_image_path = 'sunset_output_image.jpeg'
image = Image.open(image_path)
# Draw shapes using Pillow's ImageDraw.
draw = ImageDraw.Draw(image)
draw.rectangle([100, 100, 320, 320], outline = "red") # Draw a rectangle with top-left corner at (100, 100) and bottom-right corner at (320, 320)
# Save the modified image using Pillow to preserve resolution.
image.save(output_image_path)
# Display the image in the Jupyter notebook.
plt.imshow(image)
plt.axis('off') # Hide axes
plt.show()
А вот результат, показывающий прямоугольник на реальном изображении без потери разрешения.
Возможно, вы захотите погуглить, как это сделать с помощью opencv, это неправильный способ сделать это. Что вы делаете, это строите график, а затем сохраняете его. Вы хотите использовать функции opencv, такие как
cv2.rectangle,cv2.circleи т. д.