迭代Python中所有子文件夹和OCR图片(Iterate all subfolders and O

我有一个具有多个子文件夹和图像的文件夹，我想用百度OCR提取图像中的文本文件中的每个子文件夹，并写入一个Excel文件（需要分裂内容）文件的子文件夹名称命名的每个子文件夹：

folder
        \ sub1\file0.jpg
        \ sub1\file1.jpg
        \ sub1\file2.png
        .
        .
        .
        \ sub2\xxx.png
        \ sub2\yyy.jpg
        \ sub2\zzz.png
        .
        .
        .

预期成绩：

folder
        \ sub1\file0.jpg
        \ sub1\file1.jpg
        \ sub1\file2.png
        \ sub1\sub1.xlsx
        .
        .
        .
        \ sub2\xxx.png
        \ sub2\yyy.jpg
        \ sub2\zzz.png
        \ sub2\sub2.xlsx
        .
        .
        .

以下是我已经试过，但我不知道如何实现的全过程。 请分享你的见解和想法。 谢谢。

第一步：遍历所有子文件夹和图像文件：

import os
dir_name = "D:/folder"     
for root, dirs, files in os.walk(dir_name, topdown=False):
    for file in files:
        print(file)
        print(root)
        print(dirs)

步骤2：OCR一个图像

from aip import AipOcr

APP_ID = '<APP_ID>'
API_KEY = '<APP_KEY>'
SECRET_KEY = '<APP_SECRET>'

aipOcr = AipOcr(APP_ID, API_KEY, SECRET_KEY)

filePath = "test.jpg"

def get_file_content(filePath):
    with open(filePath, 'rb') as fp:
        return fp.read()

options = {
    'detect_direction': 'true',
    'language_type': 'CHN_ENG',
    'recognize_granularity': 'big',
    'vertexes_location': 'true',
    #'probability': 'true',
    #'detect_language': 'true'
}

result = aipOcr.basicAccurate(get_file_content(filePath), options)

print(result)

df = DataFrame(result)

writer = ExcelWriter('test.xlsx')
df.to_excel(writer, index = False)
writer.save()

第三步：写每个子文件夹Excel文件（感谢@Florian 1H）

使用Python中的子文件夹的名字为每个子文件夹中的空文件

from os import listdir
from os.path import isfile, join

mypath = "D:/"

def write_files(path):
    folders = [f for f in listdir(path) if not isfile(join(path, f))]
    if len(folders) == 0:
        #Writing the actual File
        open(path+"/"+path.split("/")[-1]+".xlsx", "w+")
    else:
        for folder in folders:
            write_files(path+"/"+folder)

write_files(mypath)

文章来源: Iterate all subfolders and OCR images in Python