python - 键入注释返回 PIL 图像数组的函数的正确方法

Question

我有一个加载目录中所有图像的函数，get_images()。如何正确键入注释函数以表示它返回图像数组？

以下错误:

import os
from PIL import Image
from typing import List

def get_images(folder: str)->List[Image]:
    images = []
    for path in os.listdir(folder):
        if path.endswith(".jpg"):
            img_path = "/".join([folder, path])
            print(img_path)
            img = Image.open(img_path)
            images.append(img)
    return images

TypeError: Parameters to generic types must be types. Got <module 'PIL.Image' from '/Library/Frameworks/Python.framework/Versions/3.6/lib/python3.6/site-packa.

Answer 1

通过使用List[Image]，您正在调用模块列表Image，您想要的是对象Image.Image的类型所以它会是:

def get_images(folder: str)->List[Image.Image]: