import os
from cv2 import cv2


def split_image(src_path, rownum, colnum, file):
    img = cv2.imread(src_path)
    # cv2.imwrite(path, img)
    size = img.shape[0:2]
    w = size[1]
    h = size[0]
    # print(file, w, h)
    # 每行的高度和每列的寬度
    row_height = h // rownum
    col_width = w // rownum
    num = 0
    for i in range(rownum):
        for j in range(colnum):
        	# 保存切割好的圖片的路徑，記得要填上后綴，以及名字要處理一下，可以是
        	# src_path.split('.')[0] + '_' + str((i+1)*(j+1)) + '.jpg'
            save_path = ''
            row_start = j * col_width
            row_end = (j+1) * col_width
            col_start = i * row_height
            col_end = (i+1) * row_height
            # print(row_start, row_end, col_start, col_end)
            # cv2圖片： [高， 寬]
            child_img = img[col_start:col_end, row_start:row_end]
            cv2.imwrite(save_path, child_img)


if __name__ == '__main__':
    # 可以遍歷文件夾
    # file_path = r'我是路徑（文件夾路徑）'
    # for file in file_names:
    # src_path 具體圖片路徑，包含后綴
    src_path = ''
    row = 4
    col = 4
    split_image(src_path, row, col, file.split('.')[0])