pytorch源码 folder

最新推荐文章于 2025-07-17 18:42:50 发布

兢兢业业小码农

最新推荐文章于 2025-07-17 18:42:50 发布

阅读量2.2k

点赞数 1

CC 4.0 BY-SA版权

分类专栏： pytorch源码 caffe | pytorch 源码共读项目分享文章标签： pytorch源码 folder

本文链接：https://blog.youkuaiyun.com/qq_28660035/article/details/80483174

本文深入探讨了PyTorch中用于数据加载的ImageFolder模块，旨在帮助开发者更好地理解和利用这一工具，加速深度学习项目的开发。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

在使用pytorch构建数据库时，会使用到ImageFolder这个模块便于数据加载，了解其源码便于快速开发。

import torch.utils.data as data
#PIL: Python Image Library缩写，图像处理模块
#     Image,ImageFont,ImageDraw,ImageFilter
from PIL import Image    
import os
import os.path

# 图片扩展（图片格式）
IMG_EXTENSIONS = [
    '.jpg', '.JPG', '.jpeg', '.JPEG',
    '.png', '.PNG', '.ppm', '.PPM', '.bmp', '.BMP',
]

# 判断是不是图片文件
def is_image_file(filename):
    # 只要文件以IMG_EXTENSIONS结尾，就是图片
    # 注意any的使用
    return any(filename.endswith(extension) for extension in IMG_EXTENSIONS)

# 结果:classes:['0', '1', '2', '3', '4', '5', '6', '7', '8', '9']
# classes_to_idx:{'1': 1, '0': 0, '3': 3, '2': 2, '5': 5, '4': 4, '7': 7, '6': 6, '9': 9, '8': 8}
def find_classes(dir):
    '''
     返回dir下的类别名，classes:所有的类别，class_to_idx:将文件中str的类别名转化为int类别
     classes为目录下所有文件夹名字的集合
    '''
    # os.listdir：以列表的形式显示当前目录下的所有文件名和目录名，但不会区分文件和目录。
    # os.path.isdir：判定对象是否是目录，是则返回True，否则返回False
    # os.path.join：连接目录和文件名

    classes = [d for d i