2021年12月30日
pytorch基本使用
摘要： 自定义一个数据集 from torch.utils.data import Dataset import os import cv2 # 定义一个类，继承Dataset class MyData(Dataset): def __init__(self, root_dir, label_dir): s 阅读全文
posted @ 2021-12-30 17:53 Gazikel 阅读(71) 评论(0) 推荐(0)