フォルダを一括読み込みで、画像分割をしたい

Question

フォルダに入っている画像を指定して、中にある画像を一括で分割できるようにしたいです。
下記の内容を実行しても、エラーメッセージも結果も出ない状況です。
解決方法がありましたら教えてください。

import os
# 画像分割

import os
import cv2

from PIL import Image
import sys
from datetime import datetime

def main():
    data_dir_path = u"./test/wood"
    file_list = os.listdir(r'./test/wood')

    for file_name in file_list:
        root, ext = os.path.splitext(file_name)
        if ext == u'.png' or u'.jpeg' or u'.jpg':
            abs_name = data_dir_path + '/' + file_name
            image = cv.imread(abs_name)
            
            h, w, c = image.shape
            v_split = 5  # 垂直方向の分割数
            h_split = 5 # 水平方向の分割数
            
            _img = image[:h // v_split * v_split, :w // h_split * h_split]
            print('{} -> {}'.format(image.shape, _image.shape)) 
            
            out_imgs = []
            for h_img in np.vsplit(_img, v_split):  # 垂直方向に分割する。
                for v_img in np.hsplit(h_img, h_split):  # 水平方向に分割する。
                    out_imgs.append(v_img)
#                     out_imgs = np.array(out_imgs)
#             print(out_imgs.shape)
            
            os.makedirs('./test/wood/output', exist_ok=True)
            for i, img in enumerate(out_imgs):
                cv2.imwrite(os.path.join('./test/wood/output', '{}.jpg'.format(i)), img)

Answer 1

            out_imgs = []
            for h_img in np.vsplit(_img, v_split):  # 垂直方向に分割する。
                for v_img in np.hsplit(h_img, h_split):  # 水平方向に分割する。
                    out_imgs.append(v_img)
                    out_imgs = np.array(out_imgs)
            print(out_imgs.shape)

は変ですね。2行をコメントにして

            out_imgs = []
            for h_img in np.vsplit(_img, v_split):  # 垂直方向に分割する。
                for v_img in np.hsplit(h_img, h_split):  # 水平方向に分割する。
                    out_imgs.append(v_img)
#                    out_imgs = np.array(out_imgs)
#            print(out_imgs.shape)

に変更すると動くのではないでしょうか。

Comments

[ykykykk69]

該当の部分をコメントアウトしましたが、動きませんでした。
他に思い当たる部分があれば、教えていただきたいです。