如何在DeepFace库的find方法中传入预训练模型?
DeepFace.find方法预训练模型的正确传入方式
首先,model_name参数必须传入DeepFace官方支持的预训练模型名称,可选值包括:VGG-Face, Facenet, Facenet512, OpenFace, DeepFace, DeepID, Dlib, ArcFace, SFace,注意名称大小写要严格匹配。
另外,你当前代码存在一个关键误区:DeepFace.find的第二个参数是人脸数据库的文件夹路径,不是单张图片路径,该文件夹需存放所有用于比对的人脸图片。
修正后的代码示例:
# 指定合法的预训练模型 model_name = "Facenet" # 调用find方法,db_path为存放比对人脸的文件夹路径 output2 = DeepFace.find(img_path=img2, db_path="./your_face_db_folder", model_name=model_name)
额外注意:第一次使用指定模型时,DeepFace会自动下载对应的预训练权重,请确保网络环境可正常访问下载资源。
内容的提问来源于stack exchange,提问作者Aniket Pawar
相关产品推荐
相关产品推荐

