2021年8月21日
Pytorch 中 Dataset 类的使用
摘要： 导入包 import torch import os from torch.utils.data import Dataset from PIL import Image 创建 Dataset 继承类 class MyData(Dataset): def __init__(self, root_di 阅读全文
posted @ 2021-08-21 18:26 popozyl 阅读(353) 评论(0) 推荐(1) 编辑