Pygameで画像処理４

続いて、２値変換を。

import pygame

_gSrc = None

pygame.init()
pygame.display.set_mode((150, 150), 0, 32)
pygame.display.set_caption("２値変換")
_gScr = pygame.display.get_surface()
tBuf = pygame.image.load("kenmo.jpg").convert()

threshold = 127 # 閾値
# ２値変換変換
w, h = tBuf.get_size()
for j in range(h):
	for i in range(w):
		colorA = tBuf.get_at((i, j))
		gray = (colorA[0] + colorA[1] + colorA[2]) / 3
		if(threshold > gray):
			colorA = (255, 255, 255, 255)
		else:
			colorA = (0, 0, 0, 255)
		tBuf.set_at((i, j), colorA)

while True:
	_gScr.blit(tBuf, (0, 0), tBuf.get_rect())
	pygame.display.update()
	pygame.time.wait(10)