Numpy로이 코드의 속도를 높이려면 어떻게해야합니까?

Question

현재 나는 흰색에 모든 검은 색이 아닌 픽셀을 변환하려면 다음 코드를 사용하고 있습니다 :

def convert(self, img): 
    for i in range(img.shape[0]): 
     for j in range(img.shape[1]): 
      if img.item(i, j) != 0: 
       img.itemset((i, j), 255) 
    return img 

어떻게 내가 속도를 높일 수 있습니다?

Answer 1

0이 아닌 255으로 변경해야하는 모든 요소를 ​​어떻게 PIL 사용에 대한

a[a != 0] = 255 

Answer 2

와 같은 기능을합니다

def convert (self,image): 
    return image.convert('1') 

테스트 코드 :

from PIL import Image 
import matplotlib.pyplot as plt 

def convert (image): 
    return image.convert('1') 

img = Image.open('./test.png') 
plt.figure(); plt.imshow(img) 
BW = convert(img) 
plt.figure(); plt.imshow(BW) 
plt.show() 

을 결과 :

그리고 BTW, 경우에 당신은 PIL 이미지 객체의 NumPy와 배열을 필요로 쉽게 사용하여 얻을 수 있습니다 :

matrix_of_img = numpy.asarray(img.convert('L'))