Data Augmentationを行いたい

Question

解決したいこと

Data Augmentationを行いたいです。明るさを変えたり、Affine変換を行いたいと思っています。
Opencvを用いてみたのですが、回転などしかできず困っています。

該当するソースコード

import time
import numpy as np
import pandas as pd
import torch
import torchvision
from torch import nn,optim
import torch.nn.functional as F
from torch.utils.data import Dataset,DataLoader,TensorDataset

from sklearn.model_selection import train_test_split
import matplotlib.pyplot as plt
%matplotlib inline

#kaggleのfashion-mnistのcsvファイルをtrain.csvとしています

train =pd.read_csv('train.csv')

y = train['label']#正解ラベルの取り出し
x = train.drop('label', axis=1)#正解ラベルの消去

x=x.values
y=y.values

#画像チェック
x=x.reshape(-1,1,28,28)
plt.imshow(x[3][0], cmap = 'gray', vmin = 0, vmax = 255, interpolation = 'none')
plt.show()
#この後にData Augmentationをしたい

自分で試したこと

OpenCVを使うのだろうと思われましたが、回転や反転しかできませんでした。affine変換などを教えていただきたいです。よろしくお願いします。

Answer 1

Python, OpenCVで幾何変換（アフィン変換・射影変換など）

PyTorch使うのなら、こちらも
Pytorch – torchvision で使える Transform まとめ

Comments

[toast-uz]

若干補足すると、データ拡張は(1)学習前にデータそのものを増やしておく方法、(2)学習中の各エポック内でデータをランダム性を入れて変換することで複数エポック合計として拡張効果を出す方法、の2つがあります。主流なのは(2)ですね。jbpb0様の示した方法の前者は(1)、後者は(2)で使いますが、(2)のTransformを「自作」するのに前者が使えます。

[jbpb0]

toast-uzさん

> Transformを「自作」するのに前者が使えます。

なるほど
勉強になります

[tomo193]

回答ありがとうございます。
opencvはエラーが解決しなかったのですが、pytorchのtransformのほうでdata augmentationができました。とりあえずpytorchのtransformを使っていこうと思います。