随笔分类 -  深度学习

摘要:训练部分配置 训练集目录(可以是从标注平台标注后生成的标签与图片): |-- car/ -- images/ -- train # 训练集的图片 -- val # 验证集的图片,只要有图片就可以 -- lables/ # 特征标签 -- train # txt特征文件 -- val # txt特征文 阅读全文
posted @ 2024-05-27 15:22 PythonNew_Mr.Wang 阅读(59) 评论(0) 推荐(0) 编辑
摘要:下载数据集 import torch import torchvision import torchvision.transforms as transforms # 定义数据预处理的转换 transform = transforms.Compose( [transforms.ToTensor(), 阅读全文
posted @ 2023-11-02 14:31 PythonNew_Mr.Wang 阅读(58) 评论(0) 推荐(0) 编辑
摘要:基本运算 import torch # 创建一个自定义的张量 t = torch.tensor([1.0, 2.0, 3.0]) # tensor([1., 2., 3.]) # 求平均值 t.mean() # tensor(2.) # 创建一个指定行列的张量 x = torch.empty(3,5 阅读全文
posted @ 2023-11-02 14:22 PythonNew_Mr.Wang 阅读(23) 评论(0) 推荐(0) 编辑

点击右上角即可分享
微信分享提示