app update

.gitignore

@@ -0,0 +1,66 @@
+# 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
+
+
+app.py
+test.py
+test*.py
+
+model_download.py

README.md

@@ -1,8 +1,8 @@
 ---
 title: FaceRecognition
-emoji: 🦀
-colorFrom: purple
-colorTo: gray
+emoji: 🚀
+colorFrom: red
+colorTo: red
 sdk: gradio
 sdk_version: 3.1.1
 app_file: app.py

requirements.txt

@@ -0,0 +1,3 @@
+face_recognition
+gradio
+streamlit

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)  # 创建字体库