我tmd 烦死了，没有数据集？不好意思，我要白嫖！

from time import sleepfrom aip import AipFaceimport cv2import base64import numpy as npimport osimport tracebackimport jsonimport tqdm##cv2转base64def cv2_base64(image):    base64_str = cv2.imencode('.jpg', image)[1].tobytes()    base64_str = base64.b64encode(base64_str)    return base64_str##base64转cv2def base64_cv2(base64_str):    imgString = base64.b64decode(base64_str)    nparr = np.fromstring(imgString, np.uint8)    image = cv2.imdecode(nparr, cv2.IMREAD_COLOR)    return image

""" 你的 APPID AK SK """ APP_ID = '*****'API_KEY = '*****'SECRET_KEY = '*****'
client = AipFace(APP_ID, API_KEY, SECRET_KEY)imageType = "BASE64"""" 如果有可选参数 """options = {}options["face_field"] = "age,beauty"options["max_face_num"] = 2options["face_type"] = "LIVE"options["liveness_control"] = "LOW"
error_time = 0def predict(img_path):    """    使用百度API 调用接口 返回人脸相关数据（年龄，颜值，矩形框位置等等）    :param img_path: 图片路径    :return:    """    result = {"error_code": -1}    try:        # opencv 方式读取 读取出来为numpy        img_cv = cv2.imread(img_path)        img_cv = A.SmallestMaxSize(256)(image=img_cv)["image"]        # numpy -> base64 注意需要在最后加个 decode        image = cv2_base64(img_cv).decode('utf-8')
        for i in range(3):            """ 带参数调用人脸检测 """            result = client.detect(image, imageType, options)
            error_code = result["error_code"]            if error_code == 0 or error_code == 222202:                return result            sleep(1.0)            print(f"the {i+1} try error!")            print(img_path)            print(result)    except Exception as e:        print(traceback.format_exc())    return result
def write_label_json(label_json_path, result):    """    将json写入到指定的文件下    :param label_json_path:写入的文件名称    :param result: 字典    :return:    """    with open(label_json_path, "w") as f:        json.dump(result,f)
def write_log(log_path, msg):    with open(log_path, "a+") as f:        f.write(msg)

def make_label_json(root_path, img_path_name):    """    label制作    :param root_path: 数据集存在的根目录    :param img_path_name: 图片存在的目录名称    :return:    """    img_root_path = os.path.join(root_path, img_path_name)    img_name_ls = os.listdir(img_root_path)    img_path_ls = [os.path.join(img_root_path, img_name) for img_name in img_name_ls]
    label_root_path = os.path.join(root_path, "label_json")    if not os.path.exists(label_root_path):        os.mkdir(label_root_path)    log_path = os.path.join(root_path, "error_img.txt")    write_log(log_path, "-----------------------------------\n")

    error_json_dir_path = os.path.join(root_path, "error_json")    if not os.path.exists(error_json_dir_path):        os.mkdir(error_json_dir_path)    success_counts = 0    for i, img_path in tqdm.tqdm(list(enumerate(img_path_ls))):        result = predict(img_path)        img_name = img_name_ls[i]        name, ext = os.path.splitext(img_name)        label_name = name + ".json"        label_json_path = os.path.join(label_root_path, label_name)        if result["error_code"] != 0:            write_log(log_path, img_path + "\n")            write_label_json(os.path.join(error_json_dir_path, label_name), result)        else:            success_counts += 1            write_label_json(label_json_path, result)        sleep(0.5)        if i % 100 == 0:            print("成功检测：", success_counts)    print("成功检测：", success_counts)

root_path = "./"img_path_name = "test"make_label_json(root_path, img_path_name)

# 1、no module named 'aip'pip install baidu_aip# 2、如果需要在vscode 中方便的查看 json文件， 可以考虑# 安装json-tools使用 ctrl + alt + m，切换显示风格