ColabとGoogleドライブで画像を水増しするよ😇

全然わかってないですが、

どうも機械学習は反転させたり...ほか何かしたりして素材を水増しするらしいです。これはいい情報です。どうやって数万枚も集めるんだよって思っていました。とはいえそこらも組み込まれたりするらしいですけど。でもとりあえず反転してみようと思いました。

Googleドライブをマウントする

pip install Pillow

import os #ファイルやフォルダ操作
from PIL import Image #PillowからImageをインポート
import cv2
import glob

from google.colab import drive #Googleドライブとの連携
from google.colab import files
drive.mount("/content/drive")

Googleドライブで画像のあるフォルダを指定して

flip_dir_name = "/content/drive/MyDrive/ここから/ディレクトリを/指定する/日本語OK" #保存する先のフォルダ

files = glob.glob(flip_dir_name+"/*")#反転を反転すると戻って、増幅するので注意

ループさせながらopencvで反転したものを出力します

for n in range(len(files)):
  img = cv2.imread(files[n])
  img_horizontalflip = cv2.flip(img, 1)
  print(files[n])
  print(flip_dir_name + "/" + os.path.split(files[n])[1][:-4] + "_flip_lr.jpg")

  if os.path.split(files[n])[1].endswith('.jpg'):
    cv2.imwrite(flip_dir_name + "/" + os.path.split(files[n])[1][:-4] + "_flip_lr.jpg", img_horizontalflip)
  elif os.path.split(files[n])[1].endswith('.jpeg'):
    cv2.imwrite(flip_dir_name + "/" + os.path.split(files[n])[1][:-5] + "_flip_lr.jpg", img_horizontalflip)

拡張子の判定とかは良しなにお願いします。たまたま「jpg, jpeg」だったのでif分で仕分けています。printはなくてもいいです。無事反転したものが出ました。これで単純に素材が2倍です。