随笔分类 - 深度学习
摘要:训练部分配置 训练集目录(可以是从标注平台标注后生成的标签与图片): |-- car/ -- images/ -- train # 训练集的图片 -- val # 验证集的图片,只要有图片就可以 -- lables/ # 特征标签 -- train # txt特征文件 -- val # txt特征文
阅读全文
摘要:下载数据集 import torch import torchvision import torchvision.transforms as transforms # 定义数据预处理的转换 transform = transforms.Compose( [transforms.ToTensor(),
阅读全文
摘要:基本运算 import torch # 创建一个自定义的张量 t = torch.tensor([1.0, 2.0, 3.0]) # tensor([1., 2., 3.]) # 求平均值 t.mean() # tensor(2.) # 创建一个指定行列的张量 x = torch.empty(3,5
阅读全文