Raspbian'daki PyPNG ile png piksellerini seçici olarak değiştirme

Question

Raspberry pi'nin kameramla yakaladığım png'nin her bir pikselinden geçmeye çalışıyorum ve belirli bir r, g veya b değerinin üstünde veya altında olan pikselleri seçici olarak değiştiriyorum . Oldukça verimsiz bir algoritma olduğunu biliyorum, sadece python betiklerinin asılmasına çalışıyorum. Kodumu @ Constantin'ın kodundan şu sorudan aldım: How can I read the RGB value of a given pixel in Python? Onun kodu aşağıdadır.

ne olur

#!/usr/bin/python 

import png, array 

reader = png.Reader(filename='test.png') 
w, h, pixels, metadata = reader.read_flat() 
pixel_byte_width = 4 if metadata['alpha'] else 3 

for x in range(w): 
    for y in range(h): 
     point_index = x+(y-1)*w#This is the bit that I said I'd fix above. 
     r = pixels[point_index * pixel_byte_width + 0] 
     g = pixels[point_index * pixel_byte_width + 1] 
     b = pixels[point_index * pixel_byte_width + 2] 
     pixel = pixels[point_index * pixel_byte_width : 
      (point_index+1) * pixel_byte_width] 
     #Above we have all the info about each byte, and below is our devious plan 
     new_pixel = (0, 0, 0, 0) if metadata['alpha'] else (0, 0, 0) 
     #if g > 175: 
     pixel = array.array('B', new_pixel) 

output = open('test_edited.png', 'wb') 
writer = png.Writer(w, h, **metadata) 
writer.write_array(output, pixels) 
output.close() 

pi bir iki dakika düşünür, sonra ben de aynı olan yeni bir png açabiliriz:

import png, array 

point = (2, 10) # coordinates of pixel to be painted red 

reader = png.Reader(filename='image.png') 
w, h, pixels, metadata = reader.read_flat() 
pixel_byte_width = 4 if metadata['alpha'] else 3 
#The line below is, I think wrong. I'll point out the what I did in my code below 
pixel_position = point[0] + point[1] * w 
new_pixel_value = (255, 0, 0, 0) if metadata['alpha'] else (255, 0, 0) 
pixels[ 
    pixel_position * pixel_byte_width : 
    (pixel_position + 1) * pixel_byte_width] = array.array('B', new_pixel_value) 

output = open('image-with-red-dot.png', 'wb') 
writer = png.Writer(w, h, **metadata) 
writer.write_array(output, pixels) 
output.close() 

Ve bu değiştirdim. betiğimde ne eksik, yoksa Python'dan daha iyi bir platform var mı?

Çok teşekkürler!

Answer 1

Yeni kitaplıklara açık iseniz, Python Görüntüleme Kitaplığı'nın (PIL) devamı olan pillow'u öneririm. Png ile sınırlı değil, çok daha basit. Documentation is here.

from PIL import Image, ImageDraw 

img = Image.open('image.png') 
img = img.convert("RGB") # Make sure we are in 8-bit RGB 
draw = ImageDraw.Draw(img) 

for y in range(img.height): 
    for x in range(img.width): 
     # getpixel returns a tuple with (R, G, B) 
     if img.getpixel((x, y))[1] > 175: # If too green 
      draw.point((x,y), '#000000') # Color syntax is CSS-like 

img.save('test_edited.png', 'PNG') 
# You can also use img.show() in a graphical environment