深度学习(图像数据增强)

 针对于四种天气进行的图片处理(数据增加)

#你需要的头文件
import torch
import torch.nn as nn
import torch.nn.functional as F
import torch.optim as optim
import numpy as np
import matplotlib.pyplot as plt
%matplotlib inline

import torchvision
from torchvision import transforms
import os
torchvision.transforms.RandomCrop()#随机位置裁剪
torchvision.transforms.RandomHorizontalFlip(p=1)#随机水平翻转
torchvision.transforms.RandomVerticalFlip(p=1)#随机上下翻转
torchvision.transforms.RandomRotation#随机旋转
torchvision.transforms.ColorJitter(brightness=1)#明暗程度
torchvision.transforms.ColorJitter(contrast=1)#对比度
torchvision.transforms.ColorJitter(saturation=0.5)#饱和度
torchvision.transforms.ColorJitter(hue=0.5)#颜色
torchvision.transforms.RandomGrayscale(p=0.5)#灰度化都是为了更加适应模型的变化

        因为需要让机器更加准确的识别图像,比如旁晚可以选择把晴天的亮度调暗,随机裁剪一个位置,旋转,改变对比度,颜色等条件让机器更好的识别出图像的内容。 

        这个数据集用于主要测试。随机抽样的数据集不用特意加强,会在一定程度上提高学习的准确率。

#测试数据加强数据特征
train_transform = transforms.Compose([
    transforms.Resize(224),
    transforms.RandomResizedCrop(192, scale=(0.6,1.0), ratio=(0.8,1.0)),#随机裁剪
    transforms.RandomHorizontalFlip(),#左右翻转
    transforms.RandomRotation(0.2),#随机翻转
    torchvision.transforms.ColorJitter(brightness=0.5, contrast=0, saturation=0, hue=0),#都是超参数,所以需要自己指定和尝试
    torchvision.transforms.ColorJitter(brightness=0, contrast=0.5, saturation=0, hue=0),
    transforms.ToTensor(),#转化成tensor的格式
    transforms.Normalize(mean=[.5, .5, .5], std=[.5, .5, .5])
])

(本内容借鉴于日月光华)

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

钟一淼

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值