Duplicate from Zengyf-CVer/FaceRecognition

Co-authored-by: ZengYifu <Zengyf-CVer@users.noreply.huggingface.co>

.gitattributes

@@ -0,0 +1,31 @@
+*.7z filter=lfs diff=lfs merge=lfs -text
+*.arrow filter=lfs diff=lfs merge=lfs -text
+*.bin filter=lfs diff=lfs merge=lfs -text
+*.bz2 filter=lfs diff=lfs merge=lfs -text
+*.ftz filter=lfs diff=lfs merge=lfs -text
+*.gz filter=lfs diff=lfs merge=lfs -text
+*.h5 filter=lfs diff=lfs merge=lfs -text
+*.joblib filter=lfs diff=lfs merge=lfs -text
+*.lfs.* filter=lfs diff=lfs merge=lfs -text
+*.model filter=lfs diff=lfs merge=lfs -text
+*.msgpack filter=lfs diff=lfs merge=lfs -text
+*.npy filter=lfs diff=lfs merge=lfs -text
+*.npz filter=lfs diff=lfs merge=lfs -text
+*.onnx filter=lfs diff=lfs merge=lfs -text
+*.ot filter=lfs diff=lfs merge=lfs -text
+*.parquet filter=lfs diff=lfs merge=lfs -text
+*.pickle filter=lfs diff=lfs merge=lfs -text
+*.pkl filter=lfs diff=lfs merge=lfs -text
+*.pb filter=lfs diff=lfs merge=lfs -text
+*.pt filter=lfs diff=lfs merge=lfs -text
+*.pth filter=lfs diff=lfs merge=lfs -text
+*.rar filter=lfs diff=lfs merge=lfs -text
+saved_model/**/* filter=lfs diff=lfs merge=lfs -text
+*.tar.* filter=lfs diff=lfs merge=lfs -text
+*.tflite filter=lfs diff=lfs merge=lfs -text
+*.tgz filter=lfs diff=lfs merge=lfs -text
+*.wasm filter=lfs diff=lfs merge=lfs -text
+*.xz filter=lfs diff=lfs merge=lfs -text
+*.zip filter=lfs diff=lfs merge=lfs -text
+*.zstandard filter=lfs diff=lfs merge=lfs -text
+*tfevents* filter=lfs diff=lfs merge=lfs -text

.gitignore

@@ -0,0 +1,65 @@
+# Gradio YOLOv5 Det
+# 创建人：曾逸夫
+# 项目地址：https://gitee.com/CV_Lab/gradio_yolov5_det
+
+# 图片格式
+*.jpg
+*.jpeg
+*.png
+*.svg
+*.gif
+
+# 视频格式
+*.mp4
+*.avi
+.ipynb_checkpoints
+/__pycache__
+*/__pycache__
+
+# 日志格式
+*.log
+*.data
+*.txt
+
+# 生成文件
+*.pdf
+*.xlsx
+*.csv
+
+# 参数文件
+*.yaml
+*.json
+
+# 压缩文件格式
+*.zip
+*.tar
+*.tar.gz
+*.rar
+
+# 字体格式
+*.ttc
+*.ttf
+*.otf
+*.pkl
+
+# 模型文件
+*.pt
+*.db
+
+/flagged
+/run
+!requirements.txt
+!cls_name/*
+!model_config/*
+!img_examples/*
+!img_examples/known/*
+!img_examples/unknown/*
+
+!requirements.txt
+!.pre-commit-config.yaml
+
+
+test.py
+test*.py
+
+model_download.py

README.md

@@ -0,0 +1,14 @@
+---
+title: FaceRecognition
+emoji: 🚀
+colorFrom: red
+colorTo: red
+sdk: gradio
+sdk_version: 3.1.1
+app_file: app.py
+pinned: false
+license: gpl-3.0
+duplicated_from: Zengyf-CVer/FaceRecognition
+---
+
+Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference

app.py

@@ -0,0 +1,188 @@
+# Face Recognition Hub
+# author: Zeng Yifu（曾逸夫）
+# creation time: 2022-07-28
+# email: zyfiy1314@163.com
+# project homepage: https://gitee.com/CV_Lab/face-recognition-hub
+
+import os
+import sys
+from pathlib import Path
+
+import face_recognition
+import gradio as gr
+from PIL import Image, ImageDraw, ImageFont
+
+from util.fonts_opt import is_fonts
+
+ROOT_PATH = sys.path[0]  # 项目根目录
+
+IMG_PATH_Test = "./img_examples/unknown"
+
+FONTSIZE = 15
+
+OCR_TR_DESCRIPTION = '''# Face Recognition
+<div id="content_align">https://github.com/ageitgey/face_recognition demo</div>'''
+
+def str_intercept(img_path):
+    img_path_ = img_path[::-1]
+    point_index = 0  # 记录反转后第一个点的位置
+    slash_index = 0  # 记录反转后第一个斜杠的位置
+
+    flag_pi = 0
+    flag_si = 0
+
+    for i in range(len(img_path_)):
+        if (img_path_[i] == "." and flag_pi == 0):
+            point_index = i
+            flag_pi = 1
+
+        if (img_path_[i] == "/" and flag_si == 0):
+            slash_index = i
+            flag_si = 1
+
+    point_index = len(img_path) - 1 - point_index
+    slash_index = len(img_path) - 1 - slash_index
+
+    return point_index, slash_index
+
+
+# 人脸录入
+def face_entry(img_path, name_text):
+    if img_path == "" or name_text == "" or img_path is None or name_text is None:
+        return None, None, None
+
+    point_index, slash_index = str_intercept(img_path)
+    img_renamePath = f"{img_path[:slash_index+1]}{name_text}{img_path[point_index:]}"
+    os.rename(img_path, img_renamePath)
+    img_ = Image.open(img_renamePath)
+    print(img_renamePath)
+
+    return img_, img_renamePath, name_text
+
+
+# 设置示例
+def set_example_image(example: list):
+    return gr.Image.update(value=example[0])
+
+
+def face_recognition_(img_srcPath, img_tagPath, img_personName):
+    if img_tagPath == "" or img_tagPath is None:
+        return None
+
+    image_of_person = face_recognition.load_image_file(img_srcPath)
+    person_face_encoding = face_recognition.face_encodings(image_of_person)[0]
+
+    known_face_encodings = [
+        person_face_encoding,]
+
+    known_face_names = [
+        img_personName,]
+
+    test_image = face_recognition.load_image_file(img_tagPath)
+
+    face_locations = face_recognition.face_locations(test_image)
+    face_encodings = face_recognition.face_encodings(test_image, face_locations)
+
+    pil_image = Image.fromarray(test_image)
+    img_pil = ImageDraw.Draw(pil_image)
+    textFont = ImageFont.truetype(str(f"{ROOT_PATH}/fonts/SimSun.ttf"), size=FONTSIZE)
+    #     ymin, xmax, ymax, xmin
+    for (top, right, bottom, left), face_encoding in zip(face_locations, face_encodings):
+        matches = face_recognition.compare_faces(known_face_encodings, face_encoding)
+
+        name = "Unknown Person"
+
+        if True in matches:
+            first_matches_index = matches.index(True)
+            name = known_face_names[first_matches_index]
+
+        img_pil.rectangle([left, top, right, bottom], fill=None, outline=(255, 228, 181), width=2)  # 边界框
+        text_w, text_h = textFont.getsize(name)  # 标签尺寸
+        # 标签背景
+        img_pil.rectangle(
+            (left, top, left + text_w, top + text_h),
+            fill=(255, 255, 255),
+            outline=(255, 255, 255),
+        )
+
+        # 标签
+        img_pil.multiline_text(
+            (left, top),
+            name,
+            fill=(0, 0, 0),
+            font=textFont,
+            align="center",
+        )
+
+    del img_pil
+    return pil_image
+
+
+def main():
+    is_fonts(f"{ROOT_PATH}/fonts")  # 检查字体文件
+
+    with gr.Blocks(css='style.css') as demo:
+        gr.Markdown(OCR_TR_DESCRIPTION)
+
+        # -------------- 人脸识别 录入 --------------
+        with gr.Row():
+            gr.Markdown("### Step 01: Face Entry")
+        with gr.Row():
+            with gr.Column():
+                with gr.Row():
+                    input_img = gr.Image(image_mode="RGB", source="upload", type="filepath", label="face entry")
+                with gr.Row():
+                    input_name = gr.Textbox(label="Name")
+                with gr.Row():
+                    btn = gr.Button(value="Entry")
+
+            with gr.Column():
+                with gr.Row():
+                    output_ = gr.Image(image_mode="RGB", source="upload", type="pil", label="entry image")
+                    input_srcImg = gr.Variable(value="")
+                    input_srcName = gr.Variable(value="")
+        with gr.Row():
+            example_list = [["./img_examples/known/ChengLong.jpg", "成龙"],
+                            ["./img_examples/known/VinDiesel.jpg", "VinDiesel"],
+                            ["./img_examples/known/JasonStatham.jpg", "JasonStatham"],
+                            ["./img_examples/known/ZhenZidan.jpg", "甄子丹"]]
+            gr.Examples(example_list,
+                        [input_img, input_name],
+                        output_,
+                        set_example_image,
+                        cache_examples=False)
+
+
+        # -------------- 人脸识别 测试 --------------
+        with gr.Row():
+                gr.Markdown("### Step 02: Face Test")
+        with gr.Row():
+            with gr.Column():
+                with gr.Row():
+                    input_img_test = gr.Image(image_mode="RGB", source="upload", type="filepath", label="test image")
+                with gr.Row():
+                    btn_test = gr.Button(value="Test")
+                with gr.Row():
+                    paths = sorted(Path(IMG_PATH_Test).rglob('*.jpg'))
+                    example_images_test = gr.Dataset(components=[input_img],
+                                                     samples=[[path.as_posix()] for path in paths])
+
+            with gr.Column():
+                with gr.Row():
+                    output_test = gr.Image(image_mode="RGB", source="upload", type="pil", label="identify image")
+
+        btn.click(fn=face_entry, inputs=[input_img, input_name], outputs=[output_, input_srcImg, input_srcName])
+
+        btn_test.click(fn=face_recognition_,
+                       inputs=[input_srcImg, input_img_test, input_srcName],
+                       outputs=[output_test])
+        example_images_test.click(fn=set_example_image, inputs=[
+            example_images_test,], outputs=[
+                input_img_test,])
+
+        return demo
+
+
+if __name__ == "__main__":
+    demo = main()
+    demo.launch(inbrowser=True)

requirements.txt

@@ -0,0 +1,3 @@
+wget
+rich
+face_recognition

style.css

@@ -0,0 +1,7 @@
+h1 {
+  text-align: center;
+}
+
+#content_align {
+  text-align: center;
+}

util/fonts_opt.py

@@ -0,0 +1,67 @@
+# 字体管理
+# 创建人：曾逸夫
+# 创建时间：2022-05-01
+
+import os
+import sys
+from pathlib import Path
+
+import wget
+from rich.console import Console
+
+ROOT_PATH = sys.path[0]  # 项目根目录
+
+# 中文、英文、俄语、西班牙语、阿拉伯语、韩语
+fonts_list = ["SimSun.ttf", "TimesNewRoman.ttf", "malgun.ttf"]  # 字体列表
+fonts_suffix = ["ttc", "ttf", "otf"]  # 字体后缀
+
+data_url_dict = {
+    "SimSun.ttf": "https://gitee.com/CV_Lab/gradio_yolov5_det/attach_files/1053539/download/SimSun.ttf",
+    "TimesNewRoman.ttf": "https://gitee.com/CV_Lab/gradio_yolov5_det/attach_files/1053537/download/TimesNewRoman.ttf",
+    "malgun.ttf": "https://gitee.com/CV_Lab/gradio_yolov5_det/attach_files/1053538/download/malgun.ttf",}
+
+console = Console()
+
+
+# 创建字体库
+def add_fronts(font_diff):
+
+    global font_name
+
+    for k, v in data_url_dict.items():
+        if k in font_diff:
+            font_name = v.split("/")[-1]  # 字体名称
+            Path(f"{ROOT_PATH}/fonts").mkdir(parents=True, exist_ok=True)  # 创建目录
+
+            file_path = f"{ROOT_PATH}/fonts/{font_name}"  # 字体路径
+
+            try:
+                # 下载字体文件
+                wget.download(v, file_path)
+            except Exception as e:
+                print("路径错误！程序结束！")
+                print(e)
+                sys.exit()
+            else:
+                print()
+                console.print(f"{font_name} [bold green]字体文件下载完成！[/bold green] 已保存至：{file_path}")
+
+
+# 判断字体文件
+def is_fonts(fonts_dir):
+    if os.path.isdir(fonts_dir):
+        # 如果字体库存在
+        f_list = os.listdir(fonts_dir)  # 本地字体库
+
+        font_diff = list(set(fonts_list).difference(set(f_list)))
+
+        if font_diff != []:
+            # 字体不存在
+            console.print("[bold red]字体不存在，正在加载。。。[/bold red]")
+            add_fronts(font_diff)  # 创建字体库
+        else:
+            console.print(f"{fonts_list}[bold green]字体已存在！[/bold green]")
+    else:
+        # 字体库不存在，创建字体库
+        console.print("[bold red]字体库不存在，正在创建。。。[/bold red]")
+        add_fronts(fonts_list)  # 创建字体库