機械学習で使う画像の水増しを行いたい

Question

前提・実現したいこと
機械学習で使う画像を左右反転して水増ししたい。

現在画像の識別アプリを作成しており、
そこで使う画像を水増ししたいです。

具体的にはフォルダ(saito_out)の中に画像が複数入っており、
その中の画像全てを左右反転し、
別のフォルダ(saito_mizumashi)に保存するコードを書きたいのですが、
何故かデスクトップ上に３枚のみ、左右反転した画像が保存されてしまいます。

エラーメッセージ
cv2.error: OpenCV(4.2.0) /Users/travis/build/skvark/opencv-python/opencv/modules/imgcodecs/src/loadsave.cpp:715: error: (-215:Assertion failed) !_img.empty() in function 'imwrite

該当のソースコード
cv2.imwrite('saito1.jpg', xAxis)

ここに言語名を入力
python

ソースコード

import cv2
import matplotlib.pyplot as plt
import numpy as np
import sys, os
from PIL import Image
 
#入力ファイルのパスを指定
in_jpg = "./saito_out/"
out_jpg = "./saito_mizumashi/"
 
#リストで結果を返す関数
def get_file(dir_path):
    filenames = os.listdir(dir_path)
    return filenames
 
pic = get_file(in_jpg)
 
for i in pic:
    # 画像の読み込み 
    image_gs = cv2.imread(in_jpg + i)
 
    
    xAxis = cv2.flip(image_gs, 0)
    yAxis = cv2.flip(image_gs, 1)
    xyAxis = cv2.flip(image_gs, -1)

    cv2.imwrite('saito1.jpg', xAxis)
    cv2.imwrite('saito2.jpg', yAxis)
    cv2.imwrite('saito3.jpg', xyAxis)

試したこと
フォルダもコードを書いたファイルも全てdesktop上に置いています。

補足情報（FW/ツールのバージョンなど）
mac python3.7

Answer 1

error: (-215:Assertion failed) !_img.empty() in function 'imwrite というエラー

os.listdir() はディレクトリ内のすべてのファイルを列挙する関数なので、画像ファイル以外も混じっている可能性はないでしょうか。(例: .DS_STORE とか)
読み込んだ結果 image_gs は None になっていないことを確認してください。

何故かデスクトップ上に３枚のみ、左右反転した画像が保存されてしまいます。

出力ディレクトリのパスを頭につけてないからだと思います。

- cv2.imwrite('saito1.jpg', xAxis)
- cv2.imwrite('saito2.jpg', yAxis)
- cv2.imwrite('saito3.jpg', xyAxis)
+ cv2.imwrite(out_jpg + 'saito1.jpg', xAxis)
+ cv2.imwrite(out_jpg + 'saito2.jpg', yAxis)
+ cv2.imwrite(out_jpg + 'saito3.jpg', xyAxis)

サンプルコード

import cv2
from pathlib import Path


def _get_img_paths(img_dir):
    """指定したディレクトリ内の画像ファイルのパス一覧を取得する。
    """
    img_extensions = [".jpg", ".jpeg", ".png", ".bmp"]
    img_paths = [p for p in img_dir.iterdir() if p.suffix in img_extensions]

    return img_paths


input_dir = Path("saito_out")  # 入力ディレクトリ
output_dir = Path("saito_mizumashi")  # 出力ディレクトリ
output_dir.mkdir(exist_ok=True)

for path in _get_img_paths(input_dir):
    img = cv2.imread(str(path))

    img1 = cv2.flip(img, 0)  # 左右反転
    img2 = cv2.flip(img, 1)  # 上下反転
    img3 = cv2.flip(img, -1)  # 上下左右反転

    # ファイル名は <元のファイル名>_flip1<拡張子> で保存する。
    output_path = output_dir / f"{path.stem}_flip1{path.suffix}"
    cv2.imwrite(str(output_path), img1)

    output_path = output_dir / f"{path.stem}_flip2{path.suffix}"
    cv2.imwrite(str(output_path), img2)

    output_path = output_dir / f"{path.stem}_flip3{path.suffix}"
    cv2.imwrite(str(output_path), img3)

Comments

[0000]

回答、またサンプルコードの添付ありがとうございます。
サンプルコードの方では無事出来ました‼︎

あと追記で一点教えて頂きたいのですが、
読み込んだ結果 image_gs は None になっていないことを確認してください。
>>これはどういった意味でしょうか？
どうしても理解出来ず、回答頂けたら幸いです。
よろしくお願いします。

[tiitoi]

cv2.imread() は読み込みに失敗した場合は None を返します。
そのような場合、cv2.imread() では処理は止まりませんが、その後続の処理でエラーとなります。
今回エラーが発生しているのは imwrite() のところですが、その原因は imread() でそもそも読み込みに失敗していたのではないかと思ったので聞きました。

image_gs = cv2.imread(in_jpg + i) のあとに
assert image_gs is not None
と入れて関数の返り値をチェックするようにすれば、読み込みに失敗した場合、すぐにエラーとなるので、気付けると思います。

[0000]

そういう事だったんですね‼︎
分かりやすく、理解出来ました。
ありがとうございました。