- 博客(2)
- 收藏
- 关注
RSS订阅原创 代码系列：pytorch——make_grid，ReflectionPad2d，BatchNorm2d
1. make_gridtorchvision.utils.make_grid()作用是将若干幅图像拼成一幅图像，如下所示：torchvision.utils.make_grid(tensor, nrow=8, padding=2, normalize=False, range=None, scale_each=False)tensor是4D张量，比如(B, H, W, C)的图像；n...
2019-10-16 20:24:41
906
原创 代码系列：pytorch——torchvision.datasets.ImageFolder
参考博客：https://blog.csdn.net/TH_NUM/article/details/80877435用于从文件夹中读取数据，源码链接：https://github.com/pytorch/vision/blob/master/torchvision/datasets/folder.pyImageFolder是DatasetFolder的子类，有以下属性：Attribu...
2019-10-16 14:31:40
1619
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人