- 博客(10)
- 收藏
- 关注
转载 Pytorch常用Tensor操作
Tensor基本操作1.numpy向量转tensorimport numpy as npa=np.array([2,2,2,2,3])b=torch.from_numpy(a)print(a)#[2 2 2 2 3]print(b)#tensor([2, 2, 2, 2, 3], dtype=torch.int32)2.列表转tensora=torch.tensor([2,2])b=torch.FloatTensor([2,2.])#不常用c=torch.tensor([[1,2],[
2021-03-12 23:05:02
686
原创 GTSRB数据集测试集label数据
链接:https://pan.baidu.com/s/1cIWmP6WzKaEFZfcfngheLQ提取码:apca
2020-07-06 16:26:00
1223
4
原创 从1制作数据集制作DataLoader
代码部分采用jupyter notebook编写首先建立了cat 和dog两个文件夹,从网上随便一类找了一张图片放进去了代码主要如下主要思路是建立一个csv文件,每一类图片通过简单的数据增强(本文选了旋转)将猫狗图片分别增加到十张,然后将猫的图片赋予标签0,狗的图片赋予标签1,写入csv文件中,然后制作自定义的DataLoader,主要是写好init ,getitem以及len三个类,其实就是将image和label的单个列表逐一读取,后续再利用dataloader实现迭代的数据循环,最后测试了 一下
2020-06-28 15:22:38
679
原创 Pytorch处理GTRSB数据集
最近使用了一下GTRSB数据集,主要涉及数据集的转换以及自定义制作DataLoader首先上一下GTRSB数据集的地址http://benchmark.ini.rub.de/?section=gtsrb&subsection=dataset#Annotationformat 这里我们需要的训练数据是Images and annotations (GTSRB_Final_Training_Images.zip)测试数据是 - Images and annotations (without
2020-06-28 09:31:32
2928
3
转载 python os模块 读写、创建 文件
python os模块 对文件夹、文件(文件操作函数)os.getcwd() #返回当前工作路径os.path.isabs() #判断是否是绝对路径os.path.isfile() #检验给出的路径是否是一个文件os.path.isdir() #检验给出的路径是否是一个目录os.sep #文件的路径分隔符 如:在windows上是 '\' 而在Linux上是 '/'os.walk() #文件遍历 如:def file_name(file_dir): for ro
2020-06-24 20:28:56
537
转载 常用预训练pth链接
Resnet:model_urls = {‘resnet18’: ‘https://download.pytorch.org/models/resnet18-5c106cde.pth’,‘resnet34’: ‘https://download.pytorch.org/models/resnet34-333f7ec4.pth’,‘resnet50’: ‘https://download.pytorch.org/models/resnet50-19c8e357.pth’,‘resnet101’: ‘
2020-06-24 00:35:34
484
原创 celeba数据集资源
网上很多失效链接,分享给大家这个可以用的链接:https://pan.baidu.com/s/12PPBxoZwm_4jmVNzYATdqg提取码:5pb9
2020-06-24 00:19:03
842
4
原创 Pytorch教程-2 Pytorch训练分类器
Pytorch训练分类器Pytorch针对视觉图像创建了一个叫做torchvision的包,其中包含了针对Imagenet、CIFAR10、MNIST等常用数据集的数据加载器(data loaders),还有一些对图像数据转换的操作,例如torchvision.datasets和torch.utils.data.DataLoader训练一个分类器常用流程:通过torchvision加载MNIST里面的训练和测试数据集,并对数据进行标准化定义卷积神经网络定义损失函数利用训练数据训练网络利用测试
2020-06-18 20:07:51
253
1
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人