Pythonで水増し処理をしたいが、1枚しか出力されない

Question

import os 
import cv2

def make_image(input_img):
  #画像のサイズ
  img_size=input_img.shape
  filter_one=np.ones((3,3))

  #回転用
  mat1=cv2.getRotationMatrix2D(tuple(np.array(input_img.shape[:2])/2),23,1)
  mat2=cv2.getRotationMatrix2D(tuple(np.array(input_img.shape[:2])/2),144,0.8)

  #水増しメソッドに使う関数です
  fake_method_array=np.array([
                              lambda image:cv2.warpAffine(image,mat1,image.shape[:2]),
                              lambda image:cv2.warpAffine(image,mat2,image.shape[:2]),
                              lambda image:cv2.threshold(image,100,255,cv2.THRESH_TOZERO)[1],
                              lambda image:cv2.GaussianBlur(image,(5,5),0),
                              lambda image:cv2.resize(cv2.resize(image,(img_size[1]//5,img_size[0]//5)),(img_size[1],img_size[0])),
                              lambda image:cv2.erode(image,filter_one),
                              lambda image:cv2.flip(image,1),
  ])
  
  #画像変換処理を行います。
  images=[]

  for method in fake_method_array:
    faked_img=method(input_img)
    images.append(faked_img)

    return images

#画像を読み込む
target_img=cv2.imread("IMG_1565.JPG")

#画像を水増しする
fake_images=make_image(target_img)

#画像を保存するフォルダを作成します。
if not os.path.exists("fake_images"):
  os.mkdir("fake_images")

  for number,img in enumerate(fake_images):
    #まず保存先のディレクトリ"fake_images/"を指定して番号を付けて保存します。

   cv2.imwrite("fake_images/"+ str(number) + ".jpg",img)

水増ししても、データの量が増えない理由がわかりません。

Answer 1

    return images

のインデントが間違っているので、for文の最初のループでreturnしています。

  return images

に変更しましょう。

Comments

[k.nirasawa]

水増しすることができました。ありがとうございました。
しかし、問題解決前もエラーとは表示されず、1つファイルができただけと表示されていたのですが、この場合はエラーにはならないのでしょうか？