Transform(二)

常见的transforms里的类:

一、Normalize(归一化):

1、归一化是干什么的:

对图像数据进行预处理,以确保输入模型的数据在相同的尺度上

2、为什么需要归一化:

为了消除奇异值,将数据缩放到一个小范围,这样梯度下降得更快,更容易求解

3、transforms.Normalize(mean, std)

将图像的每个通道(RGB,共3个通道)按特定的均值和标准差进行归一化

meanstd 的值是通常在 ImageNet 数据集上计算得出)

python 复制代码
#Normalize
mean = [0.485, 0.456, 0.406]
std = [0.229, 0.224, 0.225]
normalize=transforms.Normalize(mean,std) #创建归一化工具的实例化对象
img_norm=normalize(img_tensor)
print(img_norm[0][0][0]) #img_norm[0][0][0]获取该单张图像的第一个通道的第一行的第一个像素值
print(img_norm.shape) #CHW
writer.add_image("Normalize_images",img_norm,4,dataformats="CHW")

二、Resize(改变像素的大小)

注意:这里改变的大小是指改变图片的像素值,调整清晰度

resize=transforms.Resize(size) 中的 size 可以是一个元组 (height, width) 或者一个整数

(如果是整数,图像的较短边会被调整到这个值,较长边会保持比例)

python 复制代码
#Resizes
resize=transforms.Resize((700,50)) #注意size参数是()括起来的一对值
img_resize=resize(img_tensor) #输入:PIL Image or Tensor 对应输出的也是PIL Image or Tensor
writer.add_image("Resize_img",img_resize,1)

三、Compose(按顺序应用多个变换)

参数是list of ``Transform`` objects

python 复制代码
trans_resize=transforms.Resize(512)
trans_compose=transforms.Compose([trans_resize,trans_tensor])
img_resize_2=trans_compose(img_PIL)
writer.add_image("Compose",img_resize_2,1)
print(type(img_resize_2))

四、RandomCrop(随意裁剪)

python 复制代码
# RandomCrop
trans_random=transforms.RandomCrop((200,300))
trans_compose_2=transforms.Compose([trans_random,trans_tensor])
for i in range(10):
    img_crop=trans_compose_2(img_PIL)
    writer.add_image("RandomCrop1",img_crop,i)
  • size 参数可以是一个元组 (height, width),表示裁剪的目标大小
  • 每次应用时,裁剪的位置都是随机的,增加了数据的多样性

Resize 是统一调整大小,而 RandomCrop 是随机选择部分区域)

补充:

python 复制代码
from torch.utils.tensorboard import  SummaryWriter
from torchvision import transforms
from PIL import Image

Tips:

多看官方文档

关注输入和输出类型

关注方法需要什么参数

不知道返回值的时候,可以print()或print(type())或debug

相关推荐
用户51914958484511 分钟前
对抗性工程实践:利用AI自动化构建GitHub仓库的虚假提交历史
人工智能·aigc
riveting38 分钟前
重塑工业设备制造格局:明远智睿 T113-i 的破局之道
人工智能·物联网·制造·t113·明远智睿
zzywxc7871 小时前
详细探讨AI在金融、医疗、教育和制造业四大领域的具体落地案例,并通过代码、流程图、Prompt示例和图表等方式展示这些应用的实际效果。
开发语言·javascript·人工智能·深度学习·金融·prompt·流程图
算家计算1 小时前
32K上下文开源语音理解、40分钟深度交互——Voxtral-Small-24B-2507本地部署教程
人工智能·开源·aigc
聚客AI1 小时前
📝工程级开源:PyTorch手搓LLaMA4-MoE全栈指南
人工智能·llm·掘金·日新计划
TechubNews1 小时前
加密资产投资的六种策略:稳定币合规后的 Web3 投资和 RWA
人工智能·web3
机器之心1 小时前
7年了,OpenAI官方给出五代GPT对比,网友却怀念起「狂野」初代
人工智能·openai
后端小肥肠2 小时前
Coze+ComfyUI 实战:视频制作成本降10 倍,高质量成片这么做
人工智能·aigc·coze
爱分享的飘哥2 小时前
第七十章:告别“手写循环”噩梦!Trainer结构搭建:PyTorch Lightning让你“一键炼丹”!
人工智能·pytorch·分布式训练·lightning·accelerate·训练框架·trainer
阿里云大数据AI技术2 小时前
PAIFuser:面向图像视频的训练推理加速框架
人工智能·机器学习