- 博客(3)
- 收藏
- 关注
原创 使用torchvision加载数据集mnist出错出错的解决方案
使用torchvision加载数据集mnist出错出错的解决方案错误提示为:RuntimeError: Dataset not found. You can use download=True to download it加载数据集代码为:trainset_mnist = torchvision.datasets.MNIST(root='d:/dataset/MNIST', train=True, download=False, transform=transform_train)错误的原因是’d:/
2020-12-18 20:07:19 1661 2
原创 2020-11-11
torch.tensor属性:grad 张量梯度grad_fn 梯度函数data 张量数据is_leaf 是否叶子节点devices 张量所在的设备(CPU/GPU)shape 张量的形状(64,3,244,244)dtype data的数据类型requires_grad 是否需要梯度
2020-11-11 20:08:57 49
原创 跑代码记录(backgrounds_challenge)
这里写自定义目录标题跑代码记录(backgrounds_challenge)数据准备model选择预测输出跑代码记录(backgrounds_challenge)数据准备利用mask合成验证集图片。设置mask所有值为1(表示保留)或者0(表示去除),这样与图片矩阵相乘即可获得想要的图片内容。# tensor的操作参数就是其shape,inp为背景图片,im为前景图片,tile()按指定shape铺开(复制)矩阵,图片类型为'uint8'。fg_mask = np.tile(fg_mask[:,
2020-10-30 22:08:22 340
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人