Bagaimana untuk mengekstrak warna daripada imej menggunakan Python

Cara menggunakan Python untuk mengekstrak warna daripada gambar

Pengenalan:
Gambar adalah bahagian yang sangat diperlukan dalam kehidupan kita, dan dalam bidang penglihatan komputer, pengekstrakan warna daripada gambar adalah tugas yang sangat penting. Artikel ini akan memperkenalkan cara menggunakan bahasa pengaturcaraan Python untuk melaksanakan pengekstrakan warna daripada imej, dan melampirkan contoh kod untuk rujukan pembaca.

1. Import perpustakaan yang diperlukan
   Pertama, kita perlu mengimport perpustakaan PIL dan perpustakaan numpy. Pustaka PIL ialah singkatan dari Python Imaging Library, yang menyediakan fungsi pemprosesan imej yang kaya. Dan perpustakaan numpy adalah perpustakaan untuk pengkomputeran saintifik, kita boleh menggunakannya untuk memproses data imej.

from PIL import Image
import numpy as np

2. Buka fail imej
   Gunakan fungsi Image.open() perpustakaan PIL untuk membuka fail imej dan menukarnya kepada imej RGB. Imej RGB terdiri daripada tiga saluran warna: merah, hijau dan biru, dan julat nilai setiap saluran ialah 0-255. Image.open()函数打开图片文件，并将其转换为一张RGB图像。RGB图像由红、绿、蓝三个颜色通道组成，每个通道的取值范围是0-255。

image = Image.open('image.jpg').convert('RGB')

3. 将图片转换为numpy数组
   通过调用PIL库中ImageToArray()

image_array = np.array(image)

Tukar imej kepada tatasusunan numpy
Tukar imej kepada tatasusunan numpy dengan memanggil fungsi ImageToArray() dalam perpustakaan PIL. Tatasusunan numpy ialah objek tatasusunan berbilang dimensi yang boleh memproses dan menganalisis imej dengan mudah.
4. 
   

   red_channel = image_array[:,:,0]
   green_channel = image_array[:,:,1]
   blue_channel = image_array[:,:,2]

Ekstrak warna gambar
Kita boleh mengekstrak maklumat warna gambar melalui operasi indeks tatasusunan numpy. Katakan kita ingin mengekstrak maklumat warna saluran merah, hijau dan biru dalam gambar, kita boleh melakukannya dengan cara berikut:
5. 
   

   unique_colors, counts = np.unique(image_array.reshape(-1, 3), axis=0, return_counts=True)

Maklumat warna statistik
Dengan mengira saluran warna yang diekstrak dalam langkah sebelumnya, kita boleh mendapatkan Bilangan piksel warna yang berbeza dalam gambar. Ini membantu kami memahami keseluruhan ton dan taburan warna imej.
6. 
   

   import matplotlib.pyplot as plt
   
   colors = unique_colors / 255.0
   plt.pie(counts, colors=colors)
   plt.show()

Memvisualisasikan maklumat warna

Akhir sekali, kita boleh menggunakan perpustakaan matplotlib untuk menggambarkan bilangan piksel warna yang berbeza dalam gambar. Matplotlib ialah perpustakaan visualisasi data yang boleh kami gunakan untuk melukis carta seperti histogram dan carta pai.

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

# 打开图片文件
image = Image.open('image.jpg').convert('RGB')

# 将图片转换为numpy数组
image_array = np.array(image)

# 提取图片颜色
red_channel = image_array[:,:,0]
green_channel = image_array[:,:,1]
blue_channel = image_array[:,:,2]

# 统计颜色信息
unique_colors, counts = np.unique(image_array.reshape(-1, 3), axis=0, return_counts=True)

# 可视化颜色信息
colors = unique_colors / 255.0
plt.pie(counts, colors=colors)
plt.show()


Contoh kod: rrreee🎜Ringkasan: 🎜Artikel ini memperkenalkan cara menggunakan Python untuk mengekstrak warna daripada imej dan menyediakan contoh kod yang sepadan. Dengan melakukan pengekstrakan warna pada imej, kita boleh memperoleh pemahaman yang lebih mendalam tentang maklumat warna imej dan meletakkan asas untuk pemprosesan dan analisis imej seterusnya. Semoga artikel ini dapat memberi manfaat kepada pembaca. 🎜

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