随笔分类 - PyTorch
摘要：加载数据集dataloader from torch.utils.data import DataLoader form 自己写的dataset import Dataset train_set = Dataset(train=True) val_set = Dataset(train=False)
阅读全文
摘要：模型测试 import cv2 from torchvision import transforms, datasets, models from torch.utils.data import DataLoader import torch import numpy as np import os
阅读全文
摘要：数据转换 1.numpy->tensor data2 = torch.tensor(data1) 2.tensor->numpy 非训练数据（训练结束后的tensor）的转换： data2 = np.array(data1) 如果是训练过程中需要转换，则： data2 = data1.cpu().d
阅读全文
摘要：RuntimeError: Input type (torch.cuda.HalfTensor) and weight type (torch.cuda.FloatTensor) should be the same 模型输入的数据类型要与模型参数的数据类型一致。 torch.cuda.HalfTe
阅读全文
摘要：PyTorch框架中常用torchvision模块来辅助计算机视觉算法的搭建，transforms用于图像的预处理。 from torchvision import transforms 预处理操作集合：Compose rans = transforms.Compose([ transforms.T
阅读全文
