PIL.Image convert to numpy array

perphyyoung發表於2019-02-16

當使用PIL.Image讀取影像時,如果直接使用numpy.array()轉換會出現錯誤:

lst = list()
for file_name in os.listdir(dir_image):
    image = PIL.Image.open(file_name)
    lst.append(image)
arr = numpy.array(lst)

此時,上述最後一行在執行時會出現錯誤:

TypeError: int() argument must be a string, a bytes-like object or a number, not `Image`

解決辦法如下:

lst = list()
for file_name in os.listdir(dir_image):
    image = PIL.Image.open(file_name)
    lst.append(np.array(image))
arr = numpy.array(lst)

即,在list中的元素都已轉化為numpy.array,而非直接的Image物件。

相關文章