2024年6月17日
Pytorch数据加载与使用
摘要： 前言 在训练的时候通常使用Dataset来处理数据集。 Dataset的作用 提供一个方式获取数据内容和标签（label）。 实战 from torch.utils.data import Dataset from PIL import Image import os class get_data( 阅读全文
posted @ 2024-06-17 23:05 云岛夜川川 阅读(35) 评论(0) 推荐(0)