image_helpers.py 文件源码

def read_3channel_images(image_filename, num_images, file_regex):
    images = []

    for i in range(1, num_images + 1):
        imageid = file_regex % i
        filename = image_filename + imageid + ".png"

        if os.path.isfile(filename):
            print('Loading ' + filename)
            img = mpimg.imread(filename)
            tmp = np.array(img)
            if len(tmp.shape) == 3:
                img = img[:, :, :3]

            images.append(img)
        else:
            print('File ' + filename + ' does not exist')

    return np.array(images)