Проблема количественного определения количества пикселей определенного цвета в Python с использованием PIL - PullRequest
0 голосов
/ 13 июня 2019

Я хотел бы напечатать на экране количество красных и зеленых пикселей во всех изображениях данной папки (код прилагается ниже).

Однако, похоже, он повторяется снова и снова, что приводит к следующему выводу: красный = 675, зеленый = 1036, повторяется более 200 раз.

Где я ошибся? (П.С. Я совсем новичок в кодировании)

import os 
inputdir = r'C:\Users\XXXX\Desktop\Processed_Frames\LMT_T3_2mice_nothing - Copy\2dig'

from PIL import Image


for file in os.listdir(inputdir):
    im = Image.open('000013.png')
red = 0
green = 0

for pixel in im.getdata():
    if pixel == (255, 0, 0, 255): # if your image is RGB (if RGBA, (0, 0, 0, 255) or so
        red += 1
    if pixel == (000, 128, 000, 255): # if your image is RGB (if RGBA, (0, 0, 0, 255) or so
        green += 1
    print('red=' + str(red)+', green='+str(green))
end

1 Ответ

0 голосов
/ 13 июня 2019

Отступы имеют значение в Python! Команда печати появляется после каждого пикселя, а не после каждого изображения. Вам также не нужен конец для цикла for в python. Этот код напечатает общее количество красных и зеленых пикселей на изображение:

import os 
inputdir = r'C:\Users\XXXX\Desktop\Processed_Frames\LMT_T3_2mice_nothing - Copy\2dig'

from PIL import Image


for file in os.listdir(inputdir):
    im = Image.open('000013.png')
    red = 0
    green = 0

    for pixel in im.getdata():
        if pixel == (255, 0, 0, 255): # if your image is RGB (if RGBA, (0, 0, 0, 255) or so
            red += 1
        if pixel == (000, 128, 000, 255): # if your image is RGB (if RGBA, (0, 0, 0, 255) or so
            green += 1
    print('red=' + str(red)+', green='+str(green))
...