m0_63697826的博客

原创 狗狗品种分类项目

在开头说过要处理这种CSV类型的文件，同时为了方便对图片进行加载与处理，所以需要编写了一个自定义的Dataset类来封装图像的加载、变换和标签的获取。# 自定义Dataset类，基于CSV文件self.file_paths = file_paths # 图像文件名列表self.labels = labels # 对应的标签列表self.root_dir = root_dir # 图像存储的根目录self.transform = transform # 图像转换# 获取图像的路径# 加载图像。