Bagaimana untuk mewarnakan imej segmen menggunakan Python

Cara menggunakan Python untuk mewarnakan imej segmen

Apabila kita memproses imej, kadangkala kita perlu mengasingkan bahagian warna yang berbeza pada imej untuk pemprosesan atau analisis yang berasingan. Ini boleh dicapai dengan menggunakan beberapa perpustakaan pemprosesan imej bahasa pengaturcaraan Python. Artikel ini akan memperkenalkan cara menggunakan Python untuk mewarnakan imej segmen dengan cara yang mudah, dengan contoh kod dilampirkan.

Langkah 1: Pasang perpustakaan yang diperlukan

Mula-mula, kita perlu memasang Bantal perpustakaan pemprosesan imej Python. Jalankan arahan berikut dalam terminal atau command prompt untuk memasang perpustakaan Bantal:

pip install pillow

Langkah 2: Import perpustakaan yang diperlukan

Dalam kod Python, kita perlu mengimport perpustakaan Bantal bersama beberapa perpustakaan lain yang diperlukan seperti yang ditunjukkan di bawah:

from PIL import Image
import numpy as np
import matplotlib.pyplot as plt

Langkah 3: Muatkan imej

Seterusnya, kita perlu memuatkan imej untuk disegmen warna. Gunakan fungsi Image.open() daripada perpustakaan Bantal untuk memuatkan fail imej dan menukarnya kepada tatasusunan NumPy untuk pemprosesan selanjutnya. Kod sampel adalah seperti berikut:

image_path = "image.jpg"  # 图像文件的路径
image = Image.open(image_path)
image_array = np.array(image)

Langkah 4: Pembahagian warna

Setelah kami memuatkan imej dan menukarnya kepada tatasusunan NumPy, kami boleh menggunakan fungsi perpustakaan NumPy untuk mewarnakan segmen imej. Kod sampel berikut akan membahagikan imej berdasarkan nilai RGB warna:

red_mask = (image_array[:, :, 0] > 100)  # 红色通道大于100的像素点为True，其余为False
green_mask = (image_array[:, :, 1] < 50)  # 绿色通道小于50的像素点为True，其余为False
blue_mask = (image_array[:, :, 2] < 75)  # 蓝色通道小于75的像素点为True，其余为False

# 创建一个与图像大小相同的全黑图像
segmented_image = np.zeros_like(image_array)

# 使用颜色掩码将分割后的像素点赋值给新图像
segmented_image[red_mask] = image_array[red_mask]
segmented_image[green_mask] = image_array[green_mask]
segmented_image[blue_mask] = image_array[blue_mask]

Langkah 5: Paparkan imej tersegmen

Akhir sekali, kita boleh menggunakan perpustakaan Matplotlib untuk memaparkan imej tersegmen. Kod sampel berikut memaparkan imej belah pada skrin:

plt.imshow(segmented_image)
plt.axis("off")  # 关闭坐标轴
plt.show()

Selepas melengkapkan langkah di atas, kita boleh menjalankan kod dan melihat imej pecahan warna. Bergantung pada keperluan anda, anda boleh menyesuaikan peraturan pembahagian warna anda berdasarkan nilai saluran warna yang berbeza bagi imej.

Kod lengkap adalah seperti berikut:

from PIL import Image
import numpy as np
import matplotlib.pyplot as plt

image_path = "image.jpg"  # 图像文件的路径
image = Image.open(image_path)
image_array = np.array(image)

red_mask = (image_array[:, :, 0] > 100)  # 红色通道大于100的像素点为True，其余为False
green_mask = (image_array[:, :, 1] < 50)  # 绿色通道小于50的像素点为True，其余为False
blue_mask = (image_array[:, :, 2] < 75)  # 蓝色通道小于75的像素点为True，其余为False

segmented_image = np.zeros_like(image_array)

segmented_image[red_mask] = image_array[red_mask]
segmented_image[green_mask] = image_array[green_mask]
segmented_image[blue_mask] = image_array[blue_mask]

plt.imshow(segmented_image)
plt.axis("off")  # 关闭坐标轴
plt.show()

Melalui langkah di atas, kita boleh menggunakan Python untuk mewarnakan imej segmen dengan mudah. Berdasarkan keperluan khusus dan ciri imej, anda boleh menyesuaikan peraturan pembahagian warna dan pemprosesan dan analisis imej seterusnya.

Atas ialah kandungan terperinci Bagaimana untuk mewarnakan imej segmen menggunakan Python. Untuk maklumat lanjut, sila ikut artikel berkaitan lain di laman web China PHP!