Как прочитать значение RGB данного пикселя в Python?
Вопрос
Если я открою изображение с помощью open("image.jpg")
, как я могу получить значения RGB пикселя, если у меня есть координаты пикселя?
Тогда как я могу сделать обратное?Начиная с пустого изображения, «написать» пиксель с определенным значением RGB?
Я бы предпочел, чтобы мне не приходилось загружать какие-либо дополнительные библиотеки.
Решение
Вероятно, лучше всего использовать Библиотека изображений Python для этого, боюсь, потребуется отдельная загрузка.
Самый простой способ сделать то, что вы хотите, — через метод load() для объекта Image который возвращает объект доступа к пикселю, которым вы можете манипулировать как массивом:
from PIL import Image
im = Image.open('dead_parrot.jpg') # Can be many different formats.
pix = im.load()
print im.size # Get the width and hight of the image for iterating over
print pix[x,y] # Get the RGBA Value of the a pixel of an image
pix[x,y] = value # Set the RGBA Value of the image (tuple)
im.save('alive_parrot.png') # Save the modified pixels as .png
Альтернативно, посмотрите ИзображениеРисовать что дает гораздо более богатый API для создания изображений.
Другие советы
PyPNG — облегченный декодер/кодировщик PNG
Хотя вопрос намекает на JPG, я надеюсь, что мой ответ будет полезен некоторым людям.
Вот как читать и записывать пиксели PNG, используя Модуль PyPNG:
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
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()
PyPNG — это отдельный модуль Python длиной менее 4000 строк, включая тесты и комментарии.
ПИЛ — это более полная библиотека изображений, но она также значительно тяжелее.
С использованием Подушка (который работает как с Python 3.X, так и с Python 2.7+), вы можете сделать следующее:
from PIL import Image
im = Image.open('image.jpg', 'r')
width, height = im.size
pixel_values = list(im.getdata())
Теперь у вас есть все значения пикселей.Если это RGB или другой режим, его можно прочитать с помощью im.mode
.Тогда вы можете получить пиксель (x, y)
к:
pixel_values[width*y+x]
Альтернативно вы можете использовать Numpy и изменить форму массива:
>>> pixel_values = numpy.array(pixel_values).reshape((width, height, 3))
>>> x, y = 0, 1
>>> pixel_values[x][y]
[ 18 18 12]
Полное и простое в использовании решение
def get_image(image_path):
"""Get a numpy array of an image so that one can access values[x][y]."""
image = Image.open(image_path, 'r')
width, height = image.size
pixel_values = list(image.getdata())
if image.mode == 'RGB':
channels = 3
elif image.mode == 'L':
channels = 1
else:
print("Unknown mode: %s" % image.mode)
return None
pixel_values = numpy.array(pixel_values).reshape((width, height, channels))
return pixel_values
Как сказал Дэйв Уэбб:
Вот мой рабочий код, печатающий цветы пикселей с изображения:
import os, sys import Image im = Image.open("image.jpg") x = 3 y = 4 pix = im.load() print pix[x,y]
photo = Image.open('IN.jpg') #your image
photo = photo.convert('RGB')
width = photo.size[0] #define W and H
height = photo.size[1]
for y in range(0, height): #each pixel has coordinates
row = ""
for x in range(0, width):
RGB = photo.getpixel((x,y))
R,G,B = RGB #now you can use the RGB value
На wiki.wxpython.org есть действительно хорошая статья под названием Работа с изображениями.В статье упоминается возможность использования wxWidgets (wxImage), PIL или PythonMagick.Лично я использовал PIL и wxWidgets, и оба упрощают манипулирование изображениями.
Вы можете использовать pygameмодуль Surfarray.Этот модуль имеет метод возврата массива 3D-пикселей под названием «pixels3d(surface)».Я показал использование ниже:
from pygame import surfarray, image, display
import pygame
import numpy #important to import
pygame.init()
image = image.load("myimagefile.jpg") #surface to render
resolution = (image.get_width(),image.get_height())
screen = display.set_mode(resolution) #create space for display
screen.blit(image, (0,0)) #superpose image on screen
display.flip()
surfarray.use_arraytype("numpy") #important!
screenpix = surfarray.pixels3d(image) #pixels in 3d array:
#[x][y][rgb]
for y in range(resolution[1]):
for x in range(resolution[0]):
for color in range(3):
screenpix[x][y][color] += 128
#reverting colors
screen.blit(surfarray.make_surface(screenpix), (0,0)) #superpose on screen
display.flip() #update display
while 1:
print finished
Надеюсь, был полезен.Последнее слово:экран заблокирован на время жизни screenpix.
Манипулирование изображениями — сложная тема, и будет лучше, если вы делать используйте библиотеку.Я могу порекомендовать gdмодуль который обеспечивает легкий доступ ко множеству различных форматов изображений из Python.
установите PIL с помощью команды «sudo apt-get install python-imaging» и запустите следующую программу.Он будет печатать значения RGB изображения.Если изображение большое, перенаправьте вывод в файл, используя «>», позже откройте файл, чтобы увидеть значения RGB.
import PIL
import Image
FILENAME='fn.gif' #image can be in gif jpeg or png format
im=Image.open(FILENAME).convert('RGB')
pix=im.load()
w=im.size[0]
h=im.size[1]
for i in range(w):
for j in range(h):
print pix[i,j]
Вы можете использовать модуль Tkinter, который является стандартным интерфейсом Python для набора инструментов Tk GUI, и вам не потребуется дополнительная загрузка.Видеть https://docs.python.org/2/library/tkinter.html.
(В Python 3 Tkinter переименован в tkinter)
Вот как можно установить значения RGB:
#from http://tkinter.unpythonic.net/wiki/PhotoImage
from Tkinter import *
root = Tk()
def pixel(image, pos, color):
"""Place pixel at pos=(x,y) on image, with color=(r,g,b)."""
r,g,b = color
x,y = pos
image.put("#%02x%02x%02x" % (r,g,b), (y, x))
photo = PhotoImage(width=32, height=32)
pixel(photo, (16,16), (255,0,0)) # One lone pixel in the middle...
label = Label(root, image=photo)
label.grid()
root.mainloop()
И получим RGB:
#from http://www.kosbie.net/cmu/spring-14/15-112/handouts/steganographyEncoder.py
def getRGB(image, x, y):
value = image.get(x, y)
return tuple(map(int, value.split(" ")))
import matplotlib.pyplot as plt
import matplotlib.image as mpimg
img=mpimg.imread('Cricket_ACT_official_logo.png')
imgplot = plt.imshow(img)
from PIL import Image
def rgb_of_pixel(img_path, x, y):
im = Image.open(img_path).convert('RGB')
r, g, b = im.getpixel((x, y))
a = (r, g, b)
return a
Если вы хотите иметь три цифры в виде цветового кода RGB, следующий код должен сделать именно это.
i = Image.open(path)
pixels = i.load() # this is not a list, nor is it list()'able
width, height = i.size
all_pixels = []
for x in range(width):
for y in range(height):
cpixel = pixels[x, y]
all_pixels.append(cpixel)
Это может сработать для вас.