第二週、圖像預處理transforms

一、transforms運行機制

1.1 計算機視覺工具包torchvision

  • torchvision.transforms : 常用的圖像預處理方法
  • torchvision.datasets : 常用數據集的dataset實現,MNIST,CIFAR-10,ImageNet等
  • torchvision.model : 常用的模型預訓練,AlexNet,VGG, ResNet,GoogLeNet等

1.2 常用的圖像預處理方法torchvision.transforms

  • 數據中心化
  • 數據標準化
  • 縮放
  • 裁剪
  • 旋轉
  • 翻轉
  • 填充
  • 噪聲添加
  • 灰度變換
  • 線性變換
  • 仿射變換
  • 亮度、飽和度及對比度變換

1.3 流程圖

在流程圖中加入transforms,在getitem中調用transforms,我們在getitem中讀取一張圖片,然後對這一張圖片進行一系列預處理,然後返回我們的圖片以及標籤。
在這裏插入圖片描述

二、數據標準化transforms.normalize

transforms. Normalize
功能:逐channel的對圖像進行標準化(將數據的均值變爲0,標準差變爲1)
output = (input -mean) / std

  • mean:各通道的均值
  • std:各通道的標準差
  • inplace:是否原地操作
    在這裏插入圖片描述
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章