python - opencv中的dnn模块使用 简单的物体识别

% 用于读取图像文件的py文件

import os

# 可供筛选的后缀名 以“元组”的方式，不然后续会查找出错
images_type = (".jpg", ".jpeg", ".png", ".bmp", ".tif", ".tiff")

def list_images(base_path, contains=None):
    return list_file(base_path, validExt=images_type, contain=contains)

def list_file(base_path, validExt=None, contain=None):
    # 遍历base_path：rootDir 为根路径 dirNames为子文件夹名 fileNames为文件名
    for (rootDir, dirNames, fileNames) in os.walk(base_path):
        for file in fileNames:
            if contain is not None and file.find(contain) == -1:
                continue
            ext = file[file.rfind("."):].lower()
            if validExt is None or ext.endswith(validExt):
                imagePath = os.path.join(rootDir, file)
                yield imagePath  # 配合list使用

% main 文件
import utils_paths
import