gradio中如何从文件夹遍历图片并显示
时间: 2023-08-02 20:06:56 浏览: 678
在 Gradio 中遍历文件夹中的图片并显示可以使用 Python 中的 PIL 库和 os 库来实现。下面是一个示例代码:
```python
import os
from PIL import Image
import gradio as gr
def image_folder_viewer(folder_path):
images = []
for filename in os.listdir(folder_path):
if filename.endswith(".jpg") or filename.endswith(".png"):
image = Image.open(os.path.join(folder_path, filename))
images.append(image)
return images
iface = gr.Interface(fn=image_folder_viewer,
inputs="folder",
outputs="image",
title="Image Folder Viewer")
iface.launch()
```
在上面的代码中,`image_folder_viewer()` 函数接受一个文件夹路径作为输入,并使用 os 库的 `listdir()` 方法遍历文件夹中的所有文件。然后,使用 PIL 库中的 `Image.open()` 方法打开所有以 `.jpg` 或 `.png` 结尾的图片,并将它们存储在一个列表中。最后,该函数返回包含所有图片的列表。
Gradio 接口使用 `folder` 作为输入类型和 `image` 作为输出类型。当用户在界面中选择一个文件夹时,`image_folder_viewer()` 函数会被调用并显示该文件夹中所有的图片。
阅读全文