在学习python时遇到问题

Question

在学习python时出现了问题。

if __name__ == "__main__":
    img = cv2.imread("test_imgs/4.jpg")
    ocr_recognition = OCRRecognition()
    dbnet_det = DbNetInfer()
    certs = BusinessCerts(img, dbnet_det, ocr_recognition)
    for index, i in enumerate(certs.box_result):
        print(index, i)
    certs.ocr_pipline()

我现在想要批量读取图片而不是用一张，那么cv2.imread要修改成什么呢。（读取之后的数据类型不可以改变因为要交给模型处理）

我的目的是从指定的文件夹批量读取图片，并且读取出来的数据类型和cv2.imread要一致

Answer 1

os.listdir列出test_imgs目录下的图片，依次遍历用cv2.imgread读取处理什么的

import os

if __name__ == "__main__":
    images=os.listdir('test_imgs')
    for img in images:
        img = cv2.imread("test_imgs/"+img)
        ocr_recognition = OCRRecognition()
        dbnet_det = DbNetInfer()
        certs = BusinessCerts(img, dbnet_det, ocr_recognition)
        for index, i in enumerate(certs.box_result):
            print(index, i)
        certs.ocr_pipline()