2022年4月8日
PyTorch1加载数据初认识+Dataset
摘要： 1 from torch.utils.data import Dataset 2 from PIL import Image 3 4 import cv2 5 import os 6 7 class MyData(Dataset): 8 def __init__(self, root_dir, la 阅读全文
posted @ 2022-04-08 18:38 Xxiaoyu 阅读(35) 评论(0) 推荐(0) 编辑