摘要: 加载数据集dataloader from torch.utils.data import DataLoader form 自己写的dataset import Dataset train_set = Dataset(train=True) val_set = Dataset(train=False) 阅读全文
posted @ 2022-04-14 11:44 CAMILIA 阅读(742) 评论(0) 推荐(0)
摘要: 模型测试 import cv2 from torchvision import transforms, datasets, models from torch.utils.data import DataLoader import torch import numpy as np import os 阅读全文
posted @ 2022-04-14 11:26 CAMILIA 阅读(240) 评论(0) 推荐(0)