現在、Deep Learningのためのデータセットを作成するにあたり、画像の水増しを行おうと考えています。以下のURLを参考にしました。
参考URL
また、作成したコードが以下のものとないます。
python
1import os 2import cv2 3import glob 4from scipy import ndimage 5""" 6Faceディレクトリから画像を読み込んで回転、ぼかし、閾値処理をしてFaceEditedディレクトリに保存する. 7""" 8for a in range(0,141): 9 print("{}の写真を増やします。".format(a)) 10 in_dir = "./cut_Steve_Jobs/" 11 out_dir = "./cut_Steve_Jobs/" 12 os.makedirs(out_dir, exist_ok=True) 13 in_jpg=glob.glob(in_dir) 14 img_file_name_list=os.listdir("./cut_Steve_Jobs/") 15 for i in range(len(in_jpg)): 16 #print(str(in_jpg[i])) 17 img = cv2.imread(str(in_jpg[i])) 18 # 回転 19 for ang in [-10,0,10]: 20 img_rot = ndimage.rotate(img,ang) 21 img_rot = cv2.resize(img_rot,ImgSize) 22 fileName=os.path.join(out_dir,str(i)+"_"+str(ang)+".jpg") 23 cv2.imwrite(str(fileName),img_rot) 24 # 閾値 25 img_thr = cv2.threshold(img_rot, 100, 255, cv2.THRESH_TOZERO)[1] 26 fileName=os.path.join(out_dir,str(i)+"_"+str(ang)+"thr.jpg") 27 cv2.imwrite(str(fileName),img_thr) 28 # ぼかし 29 img_filter = cv2.GaussianBlur(img_rot, (5, 5), 0) 30 fileName=os.path.join(out_dir,str(i)+"_"+str(ang)+"filter.jpg") 31 cv2.imwrite(str(fileName),img_filter) 32 33print("画像の水増しに大成功しました!") 34
実行したところ、以下のようなエラーが出てしまいます。
RuntimeError: invalid rotation plane specified
無効な回転面とはなんなのでしょうか?
テストデータの水増し作業をしたことがなかったため解決策がわかりません。教えていただけると助かります。よろしくお願いします。
回答1件
あなたの回答
tips
プレビュー
バッドをするには、ログインかつ
こちらの条件を満たす必要があります。