04 2022 档案

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 Eep2026 阅读(70) 评论(0) 推荐(0)