pytorch
灰灰灰灰灰发
这个作者很懒,什么都没留下…
展开
-
Pytorch之数据增强1
Transforms on PIL Imagetorchvision.transforms# 亮度,对比,饱和,色调torchvision.transforms.ColorJitter(brightness=0, contrast=0, saturation=0, hue=0)torchvision.transforms.Grayscale(num_output_chann...原创 2019-04-25 15:50:53 · 847 阅读 · 0 评论 -
Pytorch之数据增强2
torchvision.transformstorchvision.transforms.ToPILImage(mode=None)torchvision.transforms.ToTensor()torchvision.transforms.Lambda(lambd)torchvision.transforms.LinearTransforma...原创 2019-04-25 15:57:31 · 682 阅读 · 0 评论 -
Pytorch之数据增强3
Transforms on PIL Imagetorchvision.transforms.functionaltorchvision.transforms.functional与torchvision.transforms的区别在于torchvision.transforms.functional可以自己设定格式,torchvision.transforms一般用于torchvis...原创 2019-04-25 15:32:58 · 1660 阅读 · 1 评论 -
pytorch中model.modules()和model.children()的区别
model.modules()和model.children()均为迭代器,model.modules()会遍历model中所有的子层,而model.children()仅会遍历当前层。使用:for key inmodel.modules(): print(key)#model.modules()类似于 [[1, 2], 3],其遍历结果为:[[1, 2], 3...原创 2019-06-10 22:40:37 · 16540 阅读 · 1 评论 -
Pytorch中torch.nn.functional.cross_entropy的ignore_index
ignore_index表示计算交叉熵时,自动忽略的标签值,example:import torchimport torch.nn.functional as Fpred = []pred.append([0.9, 0.1])pred.append([0.8, 0.2])pred = torch.Tensor(pred).view(-1, 2)label = torch.Lon...原创 2019-08-13 13:05:28 · 19047 阅读 · 7 评论