[openCV/Python]HDU-ITMO本科生实践基地选拔测试基础测试一V2

 

import cv2 as cv
import os
import numpy as np


def getFileList(dir, Filelist, ext=None):  # 遍历文件函数
    """
    获取文件夹及其子文件夹中文件列表
    输入 dir：文件夹根目录
    输入 ext: 扩展名
    返回： 文件路径列表
    """
    newDir = dir
    if os.path.isfile(dir):
        if ext is None:
            Filelist.append(dir)
        else:
            if ext in dir[-3:]:
                Filelist.append(dir)

    elif os.path.isdir(dir):
        for s in os.listdir(dir):
            newDir = os.path.join(dir, s)
            getFileList(newDir, Filelist, ext)

    return Filelist


def pro(img):
    yellow = {}  # 字典保存黄色HSV阈值范围
    yellow[0] = [15, 50, 100]  # 黄低
    yellow[1] = [34, 255, 255]  # 黄高
    img_hsv = cv.cvtColor(img, cv.COLOR_BGR2HSV)

    lower_bound = np.array(yellow[0])
    upper