您现在的位置是:首页 >科技 > 2025-03-27 22:06:51 来源:

📚✨Python小技巧:如何遍历文件夹中的所有`.jpg`文件?🤔💻

导读 在深度学习中,数据预处理是关键环节之一,而PyTorch提供了强大的工具来帮助我们加载和管理数据。假设你需要处理某个文件夹下的所有图片,...

在深度学习中,数据预处理是关键环节之一,而PyTorch提供了强大的工具来帮助我们加载和管理数据。假设你需要处理某个文件夹下的所有图片,那么首先得找到这些`.jpg`文件吧!这里教你用Python轻松搞定!

首先,使用`os`模块可以快速遍历目录。代码如下:

```python

import os

folder_path = './images'

for file_name in os.listdir(folder_path):

if file_name.endswith('.jpg'):

print(file_name)

```

👏搞定啦!这样就能列出文件夹里所有的`.jpg`文件啦!

接着,如果你正在用PyTorch进行图像分类任务,可以结合`torchvision.datasets.ImageFolder`模块进一步加载数据:

```python

from torchvision import datasets, transforms

data_transforms = transforms.Compose([

transforms.Resize((224, 224)),

transforms.ToTensor()

])

dataset = datasets.ImageFolder(root='./images', transform=data_transforms)

```

😎是不是超简单?无论是Python还是PyTorch,都为开发者提供了极佳的便利性!💪🎉