深度学习数据的制作

最新推荐文章于 2023-03-17 10:08:01 发布

朕；振；震

最新推荐文章于 2023-03-17 10:08:01 发布

阅读量894

点赞数

文章标签：深度学习计算机视觉人工智能

本文链接：https://blog.csdn.net/qq_55369285/article/details/126698881

版权

data_transforms = {'train':transforms.Compose([transform.RandomRotation(45),#随机旋转，-45到45度之间随机选
                    transforms.CenterCrop(224),#从中心开始裁剪
                   transforms.RandomHorizontalFlip(p=0.5),#随机垂直翻转 
                   transforms.ColorJitter(brightness=0.2,contrast=0.1),#参数1为亮度
                   transforms.RandomGrayscale(p=0.025),#概率转换成灰度率
                   transforms.ToTensor(),
                   transforms.Normalize([0.485, 0.456, 0.406], [0.229, 0.224, 0.225])#均值，标准差])，
'valid':transforms.Compose([transforms.Resize(256), 
                    transforms.CenterCrop(224),transforms.ToTensor(),
                   transforms.Normalize([0.485, 0.456, 0.406], [0.229, 0.224, 0.225])],
}