《電子技術應用》
您所在的位置:首頁 > 人工智能 > 業界動態 > 基于PyTorch的圖像數據增強技術

基于PyTorch的圖像數據增強技術

2022-11-29
來源:機械視覺沙龍
關鍵詞: PyTorch 圖像數據

  機器學習或深度學習模型的訓練的目標是成為“通用”模型。這就需要模型沒有過度擬合訓練數據集,或者換句話說,我們的模型對看不見的數據有很好的了解。數據增強也是避免過度擬合的眾多方法之一。

  擴展用于訓練模型的數據量的過程稱為數據增強。通過訓練具有多種數據類型的模型,我們可以獲得更“泛化”的模型。“多種數據類型”是什么意思呢?本片文章只討論“圖像”數據增強技術,只詳細地介紹各種圖片數據增強策略。我們還將使用 PyTorch 動手實踐并實現圖像數據或計算機視覺中主要使用的數據增強技術。

66.JPG

  因為介紹的是數據增強技術。所以只使用一張圖片就可以了,我們先看看可視話的代碼

  import PIL.Image as Image

  import torch

  from torchvision import transforms

  import matplotlib.pyplot as plt

  import numpy as np

  import warnings

  def imshow(img\_path, transform):

  “”“

  FuncTIon to show data augmentaTIon

  Param img\_path: path of the image

  Param transform: data augmentaTIon technique to apply

  ”“”

  img = Image.open(img\_path)

  fig, ax = plt.subplots(1, 2, figsize=(15, 4))

  ax[0].set\_TItle(f'Original image {img.size}‘)

  ax[0].imshow(img)

  img = transform(img)

  ax[1].set\_title(f'Transformed image {img.size}’)

  ax[1].imshow(img)

  Resize/Rescale

  此函數用于將圖像的高度和寬度調整為我們想要的特定大小。下面的代碼演示了我們想要將圖像從其原始大小調整為 224 x 224。

  path = './kitten.jpeg'

  transform = transforms.Resize((224, 224))

  imshow(path, transform)

 67.JPG

  Cropping

  該技術將要選擇的圖像的一部分應用于新圖像。例如,使用 CenterCrop 來返回一個中心裁剪的圖像。

  transform = transforms.CenterCrop((224, 224))

  imshow(path, transform)

  68.JPG

  RandomResizedCrop

  這種方法同時結合了裁剪和調整大小。

  transform = transforms.RandomResizedCrop((100, 300))

  imshow(path, transform)

 69.JPG

  Flipping

  水平或垂直翻轉圖像,下面代碼將嘗試應用水平翻轉到我們的圖像。

  transform = transforms.RandomHorizontalFlip()

  imshow(path, transform)

70.JPG

  Padding

  填充包括在圖像的所有邊緣上按指定的數量填充。我們將每條邊填充50像素。

  transform = transforms.Pad((50,50,50,50))

  imshow(path, transform)

71.JPG

  Rotation

  對圖像隨機施加旋轉角度。我們將這個角設為15度。

  transform = transforms.RandomRotation(15)

  imshow(path, transform)

 72.JPG

  Random Affine

  這種技術是一種保持中心不變的變換。這種技術有一些參數:

  degrees:旋轉角度

  translate:水平和垂直轉換

  scale:縮放參數

  share:圖片裁剪參數

  fillcolor:圖像外部填充的顏色

  transform = transforms.RandomAffine(1, translate=(0.5, 0.5), scale=(1, 1), shear=(1,1), fillcolor=(256,256,256))

  imshow(path, transform)

 73.JPG

  Gaussian Blur

  圖像將使用高斯模糊進行模糊處理。

  transform = transforms.GaussianBlur(7, 3)

  imshow(path, transform)

 74.JPG

  Grayscale

  將彩色圖像轉換為灰度。

  transform = transforms.Grayscale(num\_output\_channels=3)

  imshow(path, transform)

75.JPG

顏色增強,也稱為顏色抖動,是通過改變圖像的像素值來修改圖像的顏色屬性的過程。下面的方法都是顏色相關的操作。

  Brightness

  改變圖像的亮度當與原始圖像對比時,生成的圖像變暗或變亮。

  transform = transforms.ColorJitter(brightness=2)

  imshow(path, transform)

76.JPG

  Contrast

  圖像最暗和最亮部分之間的區別程度被稱為對比度。圖像的對比度也可以作為增強進行調整。

  transform = transforms.ColorJitter(contrast=2)

  imshow(path, transform)

77.JPG

  Saturation

  圖片中顏色的分離被定義為飽和度。

  transform = transforms.ColorJitter(saturation=20)

  imshow(path, transform)

 78.JPG

  Hue

  色調被定義為圖片中顏色的深淺。

  transform = transforms.ColorJitter(hue=2)

  imshow(path, transform)

  79.JPG

  總結

  圖像本身的變化將有助于模型對未見數據的泛化,從而不會對數據進行過擬合。



更多信息可以來這里獲取==>>電子技術應用-AET<< 

mmexport1621241704608.jpg

本站內容除特別聲明的原創文章之外,轉載內容只為傳遞更多信息,并不代表本網站贊同其觀點。轉載的所有的文章、圖片、音/視頻文件等資料的版權歸版權所有權人所有。本站采用的非本站原創文章及圖片等內容無法一一聯系確認版權者。如涉及作品內容、版權和其它問題,請及時通過電子郵件或電話通知我們,以便迅速采取適當措施,避免給雙方造成不必要的經濟損失。聯系電話:010-82306118;郵箱:aet@chinaaet.com。
主站蜘蛛池模板: 欧美在线网址 | 日韩欧美国产精品第一页不卡 | 欧美成人高清 | 99在线视频网站 | 国产精品日韩欧美在线 | 欧美午夜激情影院 | 欧美一级手机免费观看片 | 国产精品v一区二区三区 | 黄色影视网站 | 人人看人人看人做人人模 | 免费欧美一级片 | 亚洲444444在线观看 | 久久精品视频5 | 在线观看黄色片 | 免费av片在线观看网站 | 亚洲成在人色婷婷 | 亚洲欧美精品一区 | 天天做天天躁天天躁 | 丝袜美腿综合 | free欧美性杂交hd | 成人网18免费 | 日本一道本高清免费 | 在线不卡国产 | 伊人888| 秋霞午夜鲁丝片午夜精品久 | 五月天婷婷伊人 | 久草新免费 | 黄色视屏在线免费播放 | 国产又湿又黄又硬又刺激 | 亚洲片在线 | 欧美freehdvideos性 | 国产不卡一区二区视频免费 | 香港美女一级毛片 视频 | 亚洲欧美日韩中文在线 | 国产成人精品日本亚洲语音1 | 婷婷亚洲综合 | 在线播放亚洲精品 | 国产成人深夜福利在线观看 | 生活片毛片 | 黄色影院免费 | 国产 日韩 欧美 亚洲 |