画像の水増しを行いたい

Question

現在、Deep Learningのためのデータセットを作成するにあたり、画像の水増しを行おうと考えています。以下のURLを参考にしました。
参考URL

また、作成したコードが以下のものとないます。

import os
import cv2
import glob
from scipy import ndimage
"""
Faceディレクトリから画像を読み込んで回転、ぼかし、閾値処理をしてFaceEditedディレクトリに保存する.
"""
for a in range(0,141):
    print("{}の写真を増やします。".format(a))
    in_dir = "./cut_Steve_Jobs/"
    out_dir = "./cut_Steve_Jobs/"
    os.makedirs(out_dir, exist_ok=True)
    in_jpg=glob.glob(in_dir)
    img_file_name_list=os.listdir("./cut_Steve_Jobs/")
    for i in range(len(in_jpg)):
        #print(str(in_jpg[i]))
        img = cv2.imread(str(in_jpg[i]))
        # 回転
        for ang in [-10,0,10]:
            img_rot = ndimage.rotate(img,ang)
            img_rot = cv2.resize(img_rot,ImgSize)
            fileName=os.path.join(out_dir,str(i)+"_"+str(ang)+".jpg")
            cv2.imwrite(str(fileName),img_rot)
            # 閾値
            img_thr = cv2.threshold(img_rot, 100, 255, cv2.THRESH_TOZERO)[1]
            fileName=os.path.join(out_dir,str(i)+"_"+str(ang)+"thr.jpg")
            cv2.imwrite(str(fileName),img_thr)
            # ぼかし
            img_filter = cv2.GaussianBlur(img_rot, (5, 5), 0)
            fileName=os.path.join(out_dir,str(i)+"_"+str(ang)+"filter.jpg")
            cv2.imwrite(str(fileName),img_filter)

print("画像の水増しに大成功しました！")

実行したところ、以下のようなエラーが出てしまいます。
RuntimeError: invalid rotation plane specified
無効な回転面とはなんなのでしょうか？
テストデータの水増し作業をしたことがなかったため解決策がわかりません。教えていただけると助かります。よろしくお願いします。

Answer 1

scipyのGithubを見てみると、以下のような状況で質問のエラーが発生するようです。

#axesの中身が0以下、もしくはaxesの要素が入力の次元数を超えている場合
if axes[0] < 0 or axes[1] < 0 or axes[0] >= ndim or axes[1] >= ndim:
    raise ValueError('invalid rotation plane specified')

参考：https://github.com/scipy/scipy/blob/master/scipy/ndimage/interpolation.py(引用部分は703~704行目)

axesはこちら側で設定しなければ自動的にaxes=(0,1)となり、質問者さんのコードにおいてaxesを新たに設定している様子は見られません。

となると、残るはndim側、すなわち入力(img)の次元数に問題があるのだと思います。

print(img.ndim)などを用いて次元数が1以下になっていないかどうか確認してみてはいかがでしょうか(画像データならば次元数は2になるはず)。

ご参考になれば幸いです。