如何用pytorch写出读取一组图片的第一张图片和最后一张图片
时间: 2023-07-19 07:08:59 浏览: 43
可以使用PyTorch中的`torchvision`模块来读取图片,并使用Python的`os`模块来获取图片路径。以下是读取一组图片的第一张图片和最后一张图片的示例代码:
```python
import torchvision.transforms as transforms
from torchvision.datasets import ImageFolder
import os
from PIL import Image
# 设置图片路径
data_dir = "./images"
# 构建数据集
dataset = ImageFolder(
root=data_dir,
transform=transforms.Compose([
transforms.Resize((256, 256)),
transforms.CenterCrop((224, 224)),
transforms.ToTensor()
])
)
# 获取第一张图片的路径
first_img_path = os.path.join(data_dir, dataset.imgs[0][0])
# 获取最后一张图片的路径
last_img_path = os.path.join(data_dir, dataset.imgs[-1][0])
# 打印第一张图片和最后一张图片的尺寸
first_img = Image.open(first_img_path)
last_img = Image.open(last_img_path)
print("First image size:", first_img.size)
print("Last image size:", last_img.size)
```
在上面的代码中,我们首先使用`ImageFolder`类构建了一个数据集对象,然后使用`os`模块获取了第一张图片和最后一张图片的路径。最后,我们使用`PIL`模块打开这两张图片,并打印它们的尺寸。