app update

.gitignore

@@ -0,0 +1,54 @@
+# 图片格式
+*.jpg
+*.jpeg
+*.png
+*.svg
+*.gif
+
+# 视频格式
+*.mp4
+*.avi
+.ipynb_checkpoints
+*/__pycache__
+
+# 日志格式
+*.log
+*.datas
+*.txt
+
+# 生成文件
+*.pdf
+*.xlsx
+*.csv
+
+# 参数文件
+*.yaml
+*.json
+
+# 压缩文件格式
+*.zip
+*.tar
+*.tar.gz
+*.rar
+
+# 字体格式
+*.ttc
+*.ttf
+*.otf
+
+# 模型文件
+*.pt
+*.db
+
+test.py
+test*.py
+
+/flagged
+/run
+!requirements.txt
+!cls_name/*
+!model_config/*
+!img_example/*
+!packages.txt
+
+app copy.py

README.md

@@ -1,8 +1,8 @@
 ---
 title: Gradio YOLOv5 Det V5
-emoji: 💻
+emoji: 🚀
 colorFrom: red
-colorTo: pink
+colorTo: red
 sdk: gradio
 sdk_version: 3.1.1
 app_file: app.py

__init__.py

@@ -0,0 +1,2 @@
+__author__ = "曾逸夫（Zeng Yifu）"
+__email__ = "[email protected]"

app.py

@@ -0,0 +1,834 @@
+# Gradio YOLOv5 Det v0.5
+# author: Zeng Yifu（曾逸夫）
+# creation time: 2022-07-22
+# email: [email protected]
+# project homepage: https://gitee.com/CV_Lab/gradio_yolov5_det
+
+import os
+import argparse
+import csv
+import sys
+
+csv.field_size_limit(sys.maxsize)
+
+import gc
+import json
+from collections import Counter
+from pathlib import Path
+
+import cv2
+import gradio as gr
+import matplotlib.pyplot as plt
+import numpy as np
+import pandas as pd
+import plotly.express as px
+from matplotlib import font_manager
+from matplotlib.ticker import MaxNLocator
+
+ROOT_PATH = sys.path[0]  # 项目根目录
+
+# --------------------- 字体库 ---------------------
+SimSun_path = f"{ROOT_PATH}/fonts/SimSun.ttf"  # 宋体文件路径
+TimesNesRoman_path = f"{ROOT_PATH}/fonts/TimesNewRoman.ttf"  # 新罗马字体文件路径
+# 宋体
+SimSun = font_manager.FontProperties(fname=SimSun_path, size=12)
+# 新罗马字体
+TimesNesRoman = font_manager.FontProperties(fname=TimesNesRoman_path, size=12)
+
+import torch
+import yaml
+from PIL import Image, ImageDraw, ImageFont
+
+from util.fonts_opt import is_fonts
+from util.pdf_opt import pdf_generate
+
+ROOT_PATH = sys.path[0]  # 根目录
+
+# yolov5路径
+yolov5_path = "ultralytics/yolov5"
+
+# 本地模型路径
+local_model_path = f"{ROOT_PATH}/models"
+
+# Gradio YOLOv5 Det版本
+GYD_VERSION = "Gradio YOLOv5 Det v0.5"
+
+# 模型名称临时变量
+model_name_tmp = ""
+
+# 设备临时变量
+device_tmp = ""
+
+# 文件后缀
+suffix_list = [".csv", ".yaml"]
+
+# 字体大小
+FONTSIZE = 25
+
+# 目标尺寸
+obj_style = ["小目标", "中目标", "大目标"]
+
+
+def parse_args(known=False):
+    parser = argparse.ArgumentParser(description="Gradio YOLOv5 Det v0.5")
+    parser.add_argument("--source", "-src", default="upload", type=str, help="image input source")
+    parser.add_argument("--source_video", "-src_v", default="upload", type=str, help="video input source")
+    parser.add_argument("--img_tool", "-it", default="editor", type=str, help="input image tool")
+    parser.add_argument("--model_name", "-mn", default="yolov5s", type=str, help="model name")
+    parser.add_argument(
+        "--model_cfg",
+        "-mc",
+        default="./model_config/model_name_p5_p6_all.yaml",
+        type=str,
+        help="model config",
+    )
+    parser.add_argument(
+        "--cls_name",
+        "-cls",
+        default="./cls_name/cls_name_zh.yaml",
+        type=str,
+        help="cls name",
+    )
+    parser.add_argument(
+        "--nms_conf",
+        "-conf",
+        default=0.5,
+        type=float,
+        help="model NMS confidence threshold",
+    )
+    parser.add_argument("--nms_iou", "-iou", default=0.45, type=float, help="model NMS IoU threshold")
+    parser.add_argument(
+        "--device",
+        "-dev",
+        default="cuda:0",
+        type=str,
+        help="cuda or cpu",
+    )
+    parser.add_argument("--inference_size", "-isz", default=640, type=int, help="model inference size")
+    parser.add_argument("--max_detnum", "-mdn", default=50, type=float, help="model max det num")
+    parser.add_argument("--slider_step", "-ss", default=0.05, type=float, help="slider step")
+    parser.add_argument(
+        "--is_login",
+        "-isl",
+        action="store_true",
+        default=False,
+        help="is login",
+    )
+    parser.add_argument('--usr_pwd',
+                        "-up",
+                        nargs='+',
+                        type=str,
+                        default=["admin", "admin"],
+                        help="user & password for login")
+    parser.add_argument(
+        "--is_share",
+        "-is",
+        action="store_true",
+        default=False,
+        help="is login",
+    )
+
+    args = parser.parse_known_args()[0] if known else parser.parse_args()
+    return args
+
+
+# yaml文件解析
+def yaml_parse(file_path):
+    return yaml.safe_load(open(file_path, encoding="utf-8").read())
+
+
+# yaml csv 文件解析
+def yaml_csv(file_path, file_tag):
+    file_suffix = Path(file_path).suffix
+    if file_suffix == suffix_list[0]:
+        # 模型名称
+        file_names = [i[0] for i in list(csv.reader(open(file_path)))]  # csv版
+    elif file_suffix == suffix_list[1]:
+        # 模型名称
+        file_names = yaml_parse(file_path).get(file_tag)  # yaml版
+    else:
+        print(f"{file_path}格式不正确！程序退出！")
+        sys.exit()
+
+    return file_names
+
+
+#  模型加载
+def model_loading(model_name, device, opt=[]):
+
+    # 加载本地模型
+    try:
+        torch.hub._validate_not_a_forked_repo = lambda a, b, c: True
+        model = torch.hub.load(
+            yolov5_path,
+            "custom",
+            path=f"{local_model_path}/{model_name}",
+            device=device,
+            force_reload=[True if "refresh_yolov5" in opt else False][0],
+            _verbose=True,
+        )
+    except Exception as e:
+        print("模型加载失败！")
+        print(e)
+        return False
+    else:
+        print(f"🚀 欢��使用{GYD_VERSION}，{model_name}加载成功！")
+
+    return model
+
+
+# 检测信息
+def export_json(results, img_size):
+
+    return [[{
+        "ID": i,
+        "CLASS": int(result[i][5]),
+        "CLASS_NAME": model_cls_name_cp[int(result[i][5])],
+        "BOUNDING_BOX": {
+            "XMIN": round(result[i][:4].tolist()[0], 6),
+            "YMIN": round(result[i][:4].tolist()[1], 6),
+            "XMAX": round(result[i][:4].tolist()[2], 6),
+            "YMAX": round(result[i][:4].tolist()[3], 6),},
+        "CONF": round(float(result[i][4]), 2),
+        "FPS": round(1000 / float(results.t[1]), 2),
+        "IMG_WIDTH": img_size[0],
+        "IMG_HEIGHT": img_size[1],} for i in range(len(result))] for result in results.xyxyn]
+
+
+# 标签和边界框颜色设置
+def color_set(cls_num):
+    color_list = []
+    for i in range(cls_num):
+        color = tuple(np.random.choice(range(256), size=3))
+        # color = ["#"+''.join([random.choice('0123456789ABCDEF') for j in range(6)])]
+        color_list.append(color)
+
+    return color_list
+
+
+# 检测绘制
+def pil_draw(img, score_l, bbox_l, cls_l, cls_index_l, textFont, color_list, opt):
+
+    img_pil = ImageDraw.Draw(img)
+    id = 0
+
+    for score, (xmin, ymin, xmax, ymax), label, cls_index in zip(score_l, bbox_l, cls_l, cls_index_l):
+
+        img_pil.rectangle([xmin, ymin, xmax, ymax], fill=None, outline=color_list[cls_index], width=2)  # 边界框
+        countdown_msg = f"{id}-{label} {score:.2f}"
+        text_w, text_h = textFont.getsize(countdown_msg)  # 标签尺寸
+        if "label" in opt:
+            # 标签背景
+            img_pil.rectangle(
+                (xmin, ymin, xmin + text_w, ymin + text_h),
+                fill=color_list[cls_index],
+                outline=color_list[cls_index],
+            )
+
+            # 标签
+            img_pil.multiline_text(
+                (xmin, ymin),
+                countdown_msg,
+                fill=(255, 255, 255),
+                font=textFont,
+                align="center",
+            )
+
+        id += 1
+
+    return img
+
+
+# YOLOv5图片检测函数
+def yolo_det_img(img, device, model_name, infer_size, conf, iou, max_num, model_cls, opt):
+
+    global model, model_name_tmp, device_tmp
+
+    if img is None or img == "":
+        # 判断是否有图片存在
+        print("图片不存在！")
+        return None, None, None, None, None, None, None
+
+    det_img = img.copy()
+    # 目标尺寸个数
+    s_obj, m_obj, l_obj = 0, 0, 0
+
+    area_obj_all = []  # 目标面积
+
+    score_det_stat = []  # 置信度统计
+    bbox_det_stat = []  # 边界框统计
+    cls_det_stat = []  # 类别数量统计
+    cls_index_det_stat = []  # 类别索引统计
+
+    pdf_csv_xlsx = []  # 文件生成列表
+
+    if model_name_tmp != model_name:
+        # 模型判断，避免反复加载
+        model_name_tmp = model_name
+        print(f"正在加载模型{model_name_tmp}......")
+        model = model_loading(model_name_tmp, device, opt)
+    elif device_tmp != device:
+        # 设备判断，避免反复加载
+        device_tmp = device
+        print(f"正在加载模型{model_name_tmp}......")
+        model = model_loading(model_name_tmp, device, opt)
+    else:
+        print(f"正在加载模型{model_name_tmp}......")
+        model = model_loading(model_name_tmp, device, opt)
+
+    # ----------- 模型调参 -----------
+    model.conf = conf  # NMS 置信度阈值
+    model.iou = iou  # NMS IoU阈值
+    model.max_det = int(max_num)  # 最大检测框数
+    model.classes = model_cls  # 模型类别
+
+    color_list = color_set(len(model_cls_name_cp))  # 设置颜色
+
+    img_size = img.size  # 帧尺寸
+
+    results = model(img, size=infer_size)  # 检测
+    # 判断检测对象是否为空
+    # 参考：https://gitee.com/CV_Lab/face-labeling/blob/master/face_labeling.py
+    is_results_null = results.xyxyn[0].shape == torch.Size([0, 6])
+
+    if not is_results_null:
+
+        # ---------------- 目标裁剪 ----------------
+        crops = results.crop(save=False)
+        img_crops = []
+        for i in range(len(crops)):
+            img_crops.append(crops[i]["im"][..., ::-1])
+
+        # 数据表
+        dataframe = results.pandas().xyxy[0].round(2)
+
+        report = "./Det_Report.pdf"
+        det_csv = "./Det_Report.csv"
+        det_excel = "./Det_Report.xlsx"
+
+        if "csv" in opt:
+            dataframe.to_csv(det_csv, index=False)
+            pdf_csv_xlsx.append(det_csv)
+        else:
+            det_csv = None
+
+        if "excel" in opt:
+            dataframe.to_excel(det_excel, sheet_name='sheet1', index=False)
+            pdf_csv_xlsx.append(det_excel)
+        else:
+            det_excel = None
+
+        # ---------------- 加载字体 ----------------
+        yaml_index = cls_name.index(".yaml")
+        cls_name_lang = cls_name[yaml_index - 2:yaml_index]
+
+        if cls_name_lang == "zh":
+            # 中文
+            textFont = ImageFont.truetype(str(f"{ROOT_PATH}/fonts/SimSun.ttf"), size=FONTSIZE)
+        elif cls_name_lang in ["en", "ru", "es", "ar"]:
+            # 英文、俄语、西班牙语、阿拉伯语
+            textFont = ImageFont.truetype(str(f"{ROOT_PATH}/fonts/TimesNewRoman.ttf"), size=FONTSIZE)
+        elif cls_name_lang == "ko":
+            # 韩语
+            textFont = ImageFont.truetype(str(f"{ROOT_PATH}/fonts/malgun.ttf"), size=FONTSIZE)
+
+        for result in results.xyxyn:
+            for i in range(len(result)):
+                # id = int(i)  # 实例ID
+                obj_cls_index = int(result[i][5])  # 类别索引
+                cls_index_det_stat.append(obj_cls_index)
+
+                obj_cls = model_cls_name_cp[obj_cls_index]  # 类别
+                cls_det_stat.append(obj_cls)
+
+                # ------------ 边框坐标 ------------
+                x0 = float(result[i][:4].tolist()[0])
+                y0 = float(result[i][:4].tolist()[1])
+                x1 = float(result[i][:4].tolist()[2])
+                y1 = float(result[i][:4].tolist()[3])
+
+                # ------------ 边框实际坐标 ------------
+                x0 = int(img_size[0] * x0)
+                y0 = int(img_size[1] * y0)
+                x1 = int(img_size[0] * x1)
+                y1 = int(img_size[1] * y1)
+                bbox_det_stat.append((x0, y0, x1, y1))
+
+                conf = float(result[i][4])  # 置信度
+                score_det_stat.append(conf)
+
+                # fps = f"{(1000 / float(results.t[1])):.2f}"  # FPS
+
+                # ---------- 加入目标尺寸 ----------
+                w_obj = x1 - x0
+                h_obj = y1 - y0
+                area_obj = w_obj * h_obj
+                area_obj_all.append(area_obj)
+
+        det_img = pil_draw(img, score_det_stat, bbox_det_stat, cls_det_stat, cls_index_det_stat, textFont, color_list,
+                           opt)
+        # ------------ JSON生成 ------------
+        det_json = export_json(results, img.size)[0]  # 检测信息
+        det_json_format = json.dumps(det_json, sort_keys=False, indent=4, separators=(",", ":"),
+                                     ensure_ascii=False)  # JSON格式化
+        if "json" not in opt:
+            det_json = None
+
+        # -------------- PDF生成 --------------
+        if "pdf" in opt:
+            pdf_generate(f"{det_json_format}", report, GYD_VERSION)
+            pdf_csv_xlsx.append(report)
+        else:
+            report = None
+
+        # -------------- 目标尺寸计算 --------------
+        for i in range(len(area_obj_all)):
+            if (0 < area_obj_all[i] <= 32 ** 2):
+                s_obj = s_obj + 1
+            elif (32 ** 2 < area_obj_all[i] <= 96 ** 2):
+                m_obj = m_obj + 1
+            elif (area_obj_all[i] > 96 ** 2):
+                l_obj = l_obj + 1
+
+        sml_obj_total = s_obj + m_obj + l_obj
+        objSize_dict = {}
+        objSize_dict = {obj_style[i]: [s_obj, m_obj, l_obj][i] / sml_obj_total for i in range(3)}
+
+        # ------------ 类别统计 ------------
+        clsRatio_dict = {}
+        clsDet_dict = Counter(cls_det_stat)
+        clsDet_dict_sum = sum(clsDet_dict.values())
+        for k, v in clsDet_dict.items():
+            clsRatio_dict[k] = v / clsDet_dict_sum
+
+        return det_img, img_crops, objSize_dict, clsRatio_dict, dataframe, det_json, pdf_csv_xlsx
+    else:
+        print("图片目标不存在！")
+        return None, None, None, None, None, None, None
+
+
+# YOLOv5视频检测函数
+def yolo_det_video(video, device, model_name, infer_size, conf, iou, max_num, model_cls, opt, draw_style):
+
+    global model, model_name_tmp, device_tmp
+
+    if video is None or video == "":
+        # 判断是否有图片存在
+        print("视频不存在！")
+        return None, None, None
+
+    # 目标尺寸个数
+    s_obj, m_obj, l_obj = 0, 0, 0
+
+    area_obj_all = []  # 目标面积
+    s_list, m_list, l_list = [], [], []
+
+    score_det_stat = []  # 置信度统计
+    bbox_det_stat = []  # 边界框统计
+    cls_det_stat = []  # 类别数量统计
+    cls_index_det_stat = []  # 类别索引统计
+
+    fps_list = []
+
+    frame_count = 0  # 帧数
+    fps = 0  # FPS
+
+    os.system("""
+        if [ -e './output.mp4' ]; then
+        rm ./output.mp4
+        fi
+        """)
+
+    if model_name_tmp != model_name:
+        # 模型判断，避免反复加载
+        model_name_tmp = model_name
+        print(f"正在加载模型{model_name_tmp}......")
+        model = model_loading(model_name_tmp, device, opt)
+    elif device_tmp != device:
+        # 设备判断，避免反复加载
+        device_tmp = device
+        print(f"正在加载模型{model_name_tmp}......")
+        model = model_loading(model_name_tmp, device, opt)
+    else:
+        print(f"正在加载模型{model_name_tmp}......")
+        model = model_loading(model_name_tmp, device, opt)
+
+    # ----------- 模型调参 -----------
+    model.conf = conf  # NMS 置信度阈值
+    model.iou = iou  # NMS IOU阈值
+    model.max_det = int(max_num)  # 最大检测框数
+    model.classes = model_cls  # 模型类别
+
+    color_list = color_set(len(model_cls_name_cp))  # 设置颜色
+
+    # ---------------- 加载字体 ----------------
+    yaml_index = cls_name.index(".yaml")
+    cls_name_lang = cls_name[yaml_index - 2:yaml_index]
+
+    if cls_name_lang == "zh":
+        # 中文
+        textFont = ImageFont.truetype(str(f"{ROOT_PATH}/fonts/SimSun.ttf"), size=FONTSIZE)
+    elif cls_name_lang in ["en", "ru", "es", "ar"]:
+        # 英文、俄语、西班牙语、阿拉伯语
+        textFont = ImageFont.truetype(str(f"{ROOT_PATH}/fonts/TimesNewRoman.ttf"), size=FONTSIZE)
+    elif cls_name_lang == "ko":
+        # 韩语
+        textFont = ImageFont.truetype(str(f"{ROOT_PATH}/fonts/malgun.ttf"), size=FONTSIZE)
+
+    # video->frame
+    gc.collect()
+    output_video_path = "./output.avi"
+    cap = cv2.VideoCapture(video)
+    fourcc = cv2.VideoWriter_fourcc(*"I420")  # 编码器
+
+    out = cv2.VideoWriter(output_video_path, fourcc, 30.0, (int(cap.get(3)), int(cap.get(4))))
+    if cap.isOpened():
+        while cap.isOpened():
+            ret, frame = cap.read()
+            # 判断空帧
+            if not ret:
+                break
+
+            frame_count += 1  # 帧数自增
+
+            results = model(frame, size=infer_size)  # 检测
+
+            h, w, _ = frame.shape  # 帧尺寸
+            img_size = (w, h)  # 帧尺寸
+
+            for result in results.xyxyn:
+                for i in range(len(result)):
+                    # id = int(i)  # 实例ID
+                    obj_cls_index = int(result[i][5])  # 类别索引
+                    cls_index_det_stat.append(obj_cls_index)
+
+                    obj_cls = model_cls_name_cp[obj_cls_index]  # 类别
+                    cls_det_stat.append(obj_cls)
+
+                    # ------------边框坐标------------
+                    x0 = float(result[i][:4].tolist()[0])
+                    y0 = float(result[i][:4].tolist()[1])
+                    x1 = float(result[i][:4].tolist()[2])
+                    y1 = float(result[i][:4].tolist()[3])
+
+                    # ------------边框实际坐标------------
+                    x0 = int(img_size[0] * x0)
+                    y0 = int(img_size[1] * y0)
+                    x1 = int(img_size[0] * x1)
+                    y1 = int(img_size[1] * y1)
+                    bbox_det_stat.append((x0, y0, x1, y1))
+
+                    conf = float(result[i][4])  # 置信度
+                    score_det_stat.append(conf)
+
+                    fps = f"{(1000 / float(results.t[1])):.2f}"  # FPS
+
+                    # ---------- 加入目标尺寸 ----------
+                    w_obj = x1 - x0
+                    h_obj = y1 - y0
+                    area_obj = w_obj * h_obj
+                    area_obj_all.append(area_obj)
+
+            # 判断检测对象是否为空
+            # 参考：https://gitee.com/CV_Lab/face-labeling/blob/master/face_labeling.py
+            is_results_null = results.xyxyn[0].shape == torch.Size([0, 6])
+            if not is_results_null:
+                fps_list.append(float(fps))
+            else:
+                fps_list.append(0.0)
+
+            frame = Image.fromarray(cv2.cvtColor(frame, cv2.COLOR_BGR2RGB))
+            frame = pil_draw(frame, score_det_stat, bbox_det_stat, cls_det_stat, cls_index_det_stat, textFont,
+                             color_list, opt)
+            frame = cv2.cvtColor(np.asarray(frame), cv2.COLOR_RGB2BGR)
+
+            # frame->video
+            out.write(frame)
+
+            # ----- 清空统计列表 -----
+            score_det_stat = []
+            bbox_det_stat = []
+            cls_det_stat = []
+            cls_index_det_stat = []
+
+            # -------------- 目标尺寸计算 --------------
+            for i in range(len(area_obj_all)):
+                if (0 < area_obj_all[i] <= 32 ** 2):
+                    s_obj = s_obj + 1
+                elif (32 ** 2 < area_obj_all[i] <= 96 ** 2):
+                    m_obj = m_obj + 1
+                elif (area_obj_all[i] > 96 ** 2):
+                    l_obj = l_obj + 1
+
+            s_list.append(s_obj)
+            m_list.append(m_obj)
+            l_list.append(l_obj)
+
+            # 目标尺寸个数
+            s_obj, m_obj, l_obj = 0, 0, 0
+            # 目标面积
+            area_obj_all = []
+
+        out.release()
+        cap.release()
+        # cv2.destroyAllWindows()
+
+        df_objSize = pd.DataFrame({"fID": list(range(frame_count))})
+        df_objSize[obj_style[0]] = tuple(s_list)
+        df_objSize[obj_style[1]] = tuple(m_list)
+        df_objSize[obj_style[2]] = tuple(l_list)
+        print(df_objSize)
+
+        if draw_style == "Plotly":
+            # -------------------- 帧数-目标尺寸数图 --------------------
+            fig_objSize = px.scatter(df_objSize, x="fID", y=obj_style)  # 散点图
+            # fig_objSize = px.line(df_objSize, x="fID", y=obj_style, markers=True)  # 折线图
+            fig_objSize.update_layout(title="帧数-目标尺寸数", xaxis_title="帧数", yaxis_title="目标尺寸数")
+
+            # -------------------- 帧数-FPS图 --------------------
+            fig_fps = px.scatter(df_objSize, x="fID", y=fps_list)
+            # fig_fps = px.line(df_objSize, x="fID", y=fps_list, markers=True)
+            fig_fps.update_layout(title="帧数-FPS", xaxis_title="帧数", yaxis_title="FPS")
+
+        elif draw_style == "Matplotlib":
+            # -------------------- 帧数-目标尺寸数图 --------------------
+            fig_objSize = plt.figure()
+
+            # -------------------- 散点图 --------------------
+            plt.scatter(df_objSize['fID'], df_objSize[obj_style[0]])
+            plt.scatter(df_objSize['fID'], df_objSize[obj_style[1]])
+            plt.scatter(df_objSize['fID'], df_objSize[obj_style[2]])
+            # plt.plot(df_objSize['fID'], df_objSize[obj_style])  # 折线图
+            plt.title("帧数-目标尺寸数图", fontsize=12, fontproperties=SimSun)
+            plt.xlabel("帧数", fontsize=12, fontproperties=SimSun)
+            plt.ylabel("目标尺寸数", fontsize=12, fontproperties=SimSun)
+            plt.legend(obj_style, prop=SimSun, fontsize=12, loc="best")
+
+            # -------------------- 帧数-FPS图 --------------------
+            fig_fps = plt.figure()
+
+            plt.scatter(df_objSize['fID'], fps_list)
+            # plt.plot(df_objSize['fID'], df_objSize[obj_style])  # 折线图
+            plt.title("帧数-FPS", fontsize=12, fontproperties=SimSun)
+            plt.xlabel("帧数", fontsize=12, fontproperties=SimSun)
+            plt.ylabel("FPS", fontsize=12, fontproperties=SimSun)
+
+        return output_video_path, fig_objSize, fig_fps
+
+    else:
+        print("视频加载失败！")
+        return None, None, None
+
+
+def main(args):
+    gr.close_all()
+
+    global model, model_cls_name_cp, cls_name
+
+    source = args.source
+    source_video = args.source_video
+    img_tool = args.img_tool
+    nms_conf = args.nms_conf
+    nms_iou = args.nms_iou
+    model_name = args.model_name
+    model_cfg = args.model_cfg
+    cls_name = args.cls_name
+    device = args.device
+    inference_size = args.inference_size
+    max_detnum = args.max_detnum
+    slider_step = args.slider_step
+    is_login = args.is_login
+    usr_pwd = args.usr_pwd
+    is_share = args.is_share
+
+    is_fonts(f"{ROOT_PATH}/fonts")  # 检查字体文件
+
+    # 模型加载
+    model = model_loading(model_name, device)
+
+    model_names = yaml_csv(model_cfg, "model_names")  # 模型名称
+    model_cls_name = yaml_csv(cls_name, "model_cls_name")  # 类别名称
+
+    model_cls_name_cp = model_cls_name.copy()  # 类别名称
+
+    # ------------------- 图片模式输入组件 -------------------
+    inputs_img = gr.Image(image_mode="RGB", source=source, tool=img_tool, type="pil", label="原始图片")
+    inputs_device01 = gr.Radio(choices=["cuda:0", "cpu"], value=device, label="设备")
+    inputs_model01 = gr.Dropdown(choices=model_names, value=model_name, type="value", label="模型")
+    inputs_size01 = gr.Slider(384, 1536, step=128, value=inference_size, label="推理尺寸")
+    input_conf01 = gr.Slider(0, 1, step=slider_step, value=nms_conf, label="置信度阈值")
+    inputs_iou01 = gr.Slider(0, 1, step=slider_step, value=nms_iou, label="IoU 阈值")
+    inputs_maxnum01 = gr.Number(value=max_detnum, label="最大检测数")
+    inputs_clsName01 = gr.CheckboxGroup(choices=model_cls_name, value=model_cls_name, type="index", label="类别")
+    inputs_opt01 = gr.CheckboxGroup(choices=["refresh_yolov5", "label", "pdf", "json", "csv", "excel"],
+                                    value=["label", "pdf"],
+                                    type="value",
+                                    label="操作")
+
+    # ------------------- 视频模式输入组件 -------------------
+    inputs_video = gr.Video(format="mp4", source=source_video, mirror_webcam=False, label="原始视频")  # webcam
+    inputs_device02 = gr.Radio(choices=["cuda:0", "cpu"], value=device, label="设备")
+    inputs_model02 = gr.Dropdown(choices=model_names, value=model_name, type="value", label="模型")
+    inputs_size02 = gr.Slider(384, 1536, step=128, value=inference_size, label="推理尺寸")
+    input_conf02 = gr.Slider(0, 1, step=slider_step, value=nms_conf, label="置信度阈值")
+    inputs_iou02 = gr.Slider(0, 1, step=slider_step, value=nms_iou, label="IoU 阈值")
+    inputs_maxnum02 = gr.Number(value=max_detnum, label="最大检测数")
+    inputs_clsName02 = gr.CheckboxGroup(choices=model_cls_name, value=model_cls_name, type="index", label="类别")
+    inputs_opt02 = gr.CheckboxGroup(choices=["refresh_yolov5", "label"], value=["label"], type="value", label="操作")
+    inputs_draw02 = gr.Radio(choices=["Matplotlib", "Plotly"], value="Matplotlib", label="绘图")
+
+    # ------------------- 图片模式输入参数 -------------------
+    inputs_img_list = [
+        inputs_img,  # 输入图片
+        inputs_device01,  # 设备
+        inputs_model01,  # 模型
+        inputs_size01,  # 推理尺寸
+        input_conf01,  # 置信度阈值
+        inputs_iou01,  # IoU阈值
+        inputs_maxnum01,  # 最大检测数
+        inputs_clsName01,  # 类别
+        inputs_opt01,  # 检测操作
+    ]
+
+    # ------------------- 视频模式输入参数 -------------------
+    inputs_video_list = [
+        inputs_video,  # 输入图片
+        inputs_device02,  # 设备
+        inputs_model02,  # 模型
+        inputs_size02,  # 推理尺寸
+        input_conf02,  # 置信度阈值
+        inputs_iou02,  # IoU阈值
+        inputs_maxnum02,  # 最大检测数
+        inputs_clsName02,  # 类别
+        inputs_opt02,  # 检测操作
+        inputs_draw02,  # 绘图操作
+    ]
+
+    # ------------------- 图片模式输出组件 -------------------
+    outputs_img = gr.Image(type="pil", label="检测图片")
+    outputs_df = gr.Dataframe(max_rows=5, overflow_row_behaviour="paginate", type="pandas", label="检测信息列表")
+    outputs_crops = gr.Gallery(label="目标裁剪")
+    outputs_objSize = gr.Label(label="目标尺寸占比统计")
+    outputs_clsSize = gr.Label(label="类别检测占比统计")
+    outputs_json = gr.JSON(label="检测信息")
+    outputs_pdf = gr.File(label="检测报告")
+
+    # ------------------- 视频模式输出组件 -------------------
+    outputs_video = gr.Video(format='mp4', label="检测视频")
+    outputs_frame_objSize_plot = gr.Plot(label="帧数-目标尺寸数")
+    outputs_frame_fps_plot = gr.Plot(label="帧数-FPS")
+
+    # ------------------- 图片模式输出参数 -------------------
+    outputs_img_list = [
+        outputs_img, outputs_crops, outputs_objSize, outputs_clsSize, outputs_df, outputs_json, outputs_pdf]
+
+    # ------------------- 视频模式输出参数 -------------------
+    outputs_video_list = [outputs_video, outputs_frame_objSize_plot, outputs_frame_fps_plot]
+
+    # 标题
+    title = "Gradio YOLOv5 Det v0.5"
+
+    # 描述
+    description = "<div align='center'>可自定义目标检测模型、安装简单、使用方便</div>"
+    # article="https://gitee.com/CV_Lab/gradio_yolov5_det"
+
+    # 示例图片
+    examples = [
+        [
+            "./img_examples/bus.jpg",
+            "cpu",
+            "yolov5s",
+            640,
+            0.6,
+            0.5,
+            10,
+            ["人", "公交车"],
+            ["label", "pdf"],],
+        [
+            "./img_examples/giraffe.jpg",
+            "cuda:0",
+            "yolov5l",
+            320,
+            0.5,
+            0.45,
+            12,
+            ["长颈鹿"],
+            ["label", "pdf"],],
+        [
+            "./img_examples/zidane.jpg",
+            "cuda:0",
+            "yolov5m",
+            640,
+            0.6,
+            0.5,
+            15,
+            ["人", "领带"],
+            ["pdf", "json"],],
+        [
+            "./img_examples/Millenial-at-work.jpg",
+            "cuda:0",
+            "yolov5s6",
+            1280,
+            0.5,
+            0.5,
+            20,
+            ["人", "椅子", "杯子", "笔记本电脑"],
+            ["label", "pdf", "csv", "excel"],],]
+
+    # 接口
+    gyd_img = gr.Interface(
+        fn=yolo_det_img,
+        inputs=inputs_img_list,
+        outputs=outputs_img_list,
+        title=title,
+        description=description,
+        # article=article,
+        examples=examples,
+        # theme="seafoam",
+        # live=True, # 实时变更输出
+        flagging_dir="run",  # 输出目录
+        # allow_flagging="manual",
+        # flagging_options=["good", "generally", "bad"],
+    )
+
+    gyd_video = gr.Interface(
+        fn=yolo_det_video,
+        inputs=inputs_video_list,
+        outputs=outputs_video_list,
+        title=title,
+        description=description,
+        # article=article,
+        # examples=examples,
+        # theme="seafoam",
+        # live=True, # 实时变更输出
+        flagging_dir="run",  # 输出目录
+        allow_flagging="never",
+        # flagging_options=["good", "generally", "bad"],
+    )
+
+    gyd = gr.TabbedInterface(interface_list=[gyd_img, gyd_video], tab_names=["图片模式", "视频模式"])
+
+    if not is_login:
+        gyd.launch(
+            inbrowser=True,  # 自动打开默认浏览器
+            show_tips=True,  # 自动显示gradio最新功能
+            share=is_share,  # 项目共享，其他设备可以访问
+            favicon_path="./icon/logo.ico",  # 网页图标
+            show_error=True,  # 在浏览器控制台中显示错误信息
+            quiet=True,  # 禁止大多数打印语句
+        )
+    else:
+        gyd.launch(
+            inbrowser=True,  # 自动打开默认浏览器
+            show_tips=True,  # 自动显示gradio最新功能
+            auth=usr_pwd,  # 登录界面
+            share=is_share,  # 项目共享，其他设备可以访问
+            favicon_path="./icon/logo.ico",  # 网页图标
+            show_error=True,  # 在浏览器控制台中显示错误信息
+            quiet=True,  # 禁止大多数打印语句
+        )
+
+
+if __name__ == "__main__":
+    args = parse_args()
+    main(args)

cls_name/cls_name.csv

@@ -0,0 +1,80 @@
+人
+自行车
+汽车
+摩托车
+飞机
+公交车
+火车
+卡车
+船
+红绿灯
+消防栓
+停止标志
+停车收费表
+长凳
+鸟
+猫
+狗
+马
+羊
+牛
+象
+熊
+斑马
+长颈鹿
+背包
+雨伞
+手提包
+领带
+手提箱
+飞盘
+滑雪板
+单板滑雪
+运动球
+风筝
+棒球棒
+棒球手套
+滑板
+冲浪板
+网球拍
+瓶子
+红酒杯
+杯子
+叉子
+刀
+勺
+碗
+香蕉
+苹果
+三明治
+橙子
+西兰花
+胡萝卜
+热狗
+比萨
+甜甜圈
+蛋糕
+椅子
+长椅
+盆栽
+床
+餐桌
+马桶
+电视
+笔记本电脑
+鼠标
+遥控器
+键盘
+手机
+微波炉
+烤箱
+烤面包机
+洗碗槽
+冰箱
+书
+时钟
+花瓶
+剪刀
+泰迪熊
+吹风机
+牙刷

cls_name/cls_name.yaml

@@ -0,0 +1,7 @@
+model_cls_name: ['人', '自行车', '汽车', '摩托车', '飞机', '公交车', '火车', '卡车', '船', '红绿灯', '消防栓', '停止标志',
+                '停车收费表', '长凳', '鸟', '猫', '狗', '马', '羊', '牛', '象', '熊', '斑马', '长颈鹿', '背包', '雨伞', '手提包', '领带',
+                '手提箱', '飞盘', '滑雪板', '单板滑雪', '运动球', '风筝', '棒球棒', '棒球手套', '滑板', '冲浪板', '网球拍', '瓶子', '红酒杯',
+                '杯子', '叉子', '刀', '勺', '碗', '香蕉', '苹果', '三明治', '橙子', '西兰花', '胡萝卜', '热狗', '比萨', '甜甜圈', '蛋糕',
+                '椅子', '长椅', '盆栽', '床', '餐桌', '马桶', '电视', '笔记本电脑', '鼠标', '遥控器', '键盘', '手机', '微波炉', '烤箱',
+                '烤面包机', '洗碗槽', '冰箱', '书', '时钟', '花瓶', '剪刀', '泰迪熊', '吹风机', '牙刷'
+            ]

cls_name/cls_name_ar.yaml

@@ -0,0 +1,9 @@
+model_cls_name: [" الناس " , " الدراجات " , " السيارات " , " الدراجات النارية " , " الطائرات " , " الحافلات " , " القطارات " , " الشاحنات " , " السفن " , " إشارات المرور " ,
+" صنبور " , " علامة " , " موقف سيارات " , " الجدول " , " مقعد " , " الطيور " , " القط " , " الكلب " , " الحصان " , " الأغنام " , " الثور " , " الفيل " ,
+" الدب " , " حمار وحشي " , " الزرافة " , " حقيبة " , " مظلة " , " حقيبة يد " , " ربطة عنق " , " حقيبة " , " الفريسبي " , " الزلاجات " , " الزلاجات " ,
+" الكرة الرياضية " , " طائرة ورقية " , " مضرب بيسبول " , " قفازات البيسبول " , " لوح التزلج " , " ركوب الأمواج " , " مضرب تنس " , " زجاجة " ,
+" كأس " , " كأس " , " شوكة " , " سكين " , " ملعقة " , " وعاء " , " الموز " , " التفاح " , " ساندويتش " , " البرتقال " , " القرنبيط " ,
+" الجزر " , " الكلاب الساخنة " , " البيتزا " , " دونات " , " كعكة " , " كرسي " , " أريكة " , " بوعاء " , " السرير " , " طاولة الطعام " , " المرحاض " ,
+التلفزيون , الكمبيوتر المحمول , الفأرة , وحدة تحكم عن بعد , لوحة المفاتيح , الهاتف المحمول , فرن الميكروويف , محمصة خبز كهربائية , بالوعة , ثلاجة ,
+" كتاب " , " ساعة " , " زهرية " , " مقص " , " دمية دب " , " مجفف الشعر " , " فرشاة الأسنان "
+]

cls_name/cls_name_en.yaml

@@ -0,0 +1,9 @@
+model_cls_name: ['person', 'bicycle', 'car', 'motorcycle', 'airplane', 'bus', 'train', 'truck', 'boat', 'traffic light',
+    'fire hydrant', 'stop sign', 'parking meter', 'bench', 'bird', 'cat', 'dog', 'horse', 'sheep', 'cow', 'elephant',
+    'bear', 'zebra', 'giraffe', 'backpack', 'umbrella', 'handbag', 'tie', 'suitcase', 'frisbee', 'skis', 'snowboard',
+    'sports ball', 'kite', 'baseball bat', 'baseball glove', 'skateboard', 'surfboard', 'tennis racket', 'bottle',
+    'wine glass', 'cup', 'fork', 'knife', 'spoon', 'bowl', 'banana', 'apple', 'sandwich', 'orange', 'broccoli',
+    'carrot', 'hot dog', 'pizza', 'donut', 'cake', 'chair', 'couch', 'potted plant', 'bed', 'dining table', 'toilet',
+    'tv', 'laptop', 'mouse', 'remote', 'keyboard', 'cell phone', 'microwave', 'oven', 'toaster', 'sink', 'refrigerator',
+    'book', 'clock', 'vase', 'scissors', 'teddy bear', 'hair drier', 'toothbrush'
+]

cls_name/cls_name_es.yaml

@@ -0,0 +1,9 @@
+model_cls_name: ['persona', 'bicicleta', 'coche', 'motocicleta', 'avión', 'autobús', 'tren', 'camión', 'barco', 'semáforo',
+     'boca de incendios', 'señal de alto', 'parquímetro', 'banco', 'pájaro', 'gato', 'perro', 'caballo', 'oveja', 'vaca', 'elefante',
+     'oso', 'cebra', 'jirafa', 'mochila', 'paraguas', 'bolso', 'corbata', 'maleta', 'frisbee', 'esquís', 'snowboard',
+     'pelota deportiva', 'cometa', 'bate de béisbol', 'guante de béisbol', 'monopatín', 'tabla de surf', 'raqueta de tenis', 'botella',
+     'copa de vino', 'taza', 'tenedor', 'cuchillo', 'cuchara', 'tazón', 'plátano', 'manzana', 'sándwich', 'naranja', 'brócoli',
+     'zanahoria', 'perrito caliente', 'pizza', 'rosquilla', 'pastel', 'silla', 'sofá', 'planta en maceta', 'cama', 'mesa de comedor', 'inodoro',
+     'tv', 'laptop', 'ratón', 'control remoto', 'teclado', 'celular', 'microondas', 'horno', 'tostadora', 'fregadero', 'nevera',
+     'libro', 'reloj', 'jarrón', 'tijeras', 'oso de peluche', 'secador de pelo', 'cepillo de dientes'
+]

cls_name/cls_name_ko.yaml

@@ -0,0 +1,9 @@
+model_cls_name: ['사람', '자전거', '자동차', '오토바이', '비행기', '버스', '기차', '트럭', '보트', '신호등',
+     '소화전', '정지 신호', '주차 미터기', '벤치', '새', '고양이', '개', '말', '양', '소', '코끼리',
+     '곰', '얼룩말', '기린', '배낭', '우산', '핸드백', '타이', '여행가방', '프리스비', '스키', '스노우보드',
+     '스포츠 공', '연', '야구 방망이', '야구 글러브', '스케이트보드', '서프보드', '테니스 라켓', '병',
+     '와인잔', '컵', '포크', '나이프', '숟가락', '그릇', '바나나', '사과', '샌드위치', '오렌지', '브로콜리',
+     '당근', '핫도그', '피자', '도넛', '케이크', '의자', '소파', '화분', '침대', '식탁', '화장실',
+     'tv', '노트북', '마우스', '리모컨', '키보드', '휴대전화', '전자레인지', '오븐', '토스터', '싱크대', '냉장고',
+     '책', '시계', '꽃병', '가위', '테디베어', '드라이기', '칫솔'
+]

cls_name/cls_name_ru.yaml

@@ -0,0 +1,9 @@
+model_cls_name: ['человек', 'велосипед', 'автомобиль', 'мотоцикл', 'самолет', 'автобус', 'поезд', 'грузовик', 'лодка', 'светофор',
+     'пожарный гидрант', 'стоп', 'паркомат', 'скамейка', 'птица', 'кошка', 'собака', 'лошадь', 'овца', 'корова', 'слон',
+     'медведь', 'зебра', 'жираф', 'рюкзак', 'зонт', 'сумочка', 'галстук', 'чемодан', 'фрисби', 'лыжи', 'сноуборд',
+     'спортивный мяч', 'воздушный змей', 'бейсбольная бита', 'бейсбольная перчатка', 'скейтборд', 'доска для серфинга', 'теннисная ракетка', 'бутылка',
+     'бокал', 'чашка', 'вилка', 'нож', 'ложка', 'миска', 'банан', 'яблоко', 'бутерброд', 'апельсин', 'брокколи',
+     'морковь', 'хот-дог', 'пицца', 'пончик', 'торт', 'стул', 'диван', 'растение в горшке', 'кровать', 'обеденный стол', 'туалет',
+     'телевизор', 'ноутбук', 'мышь', 'пульт', 'клавиатура', 'мобильный телефон', 'микроволновая печь', 'духовка', 'тостер', 'раковина', 'холодильник',
+     'книга', 'часы', 'ваза', 'ножницы', 'плюшевый мишка', 'фен', 'зубная щетка'
+]

cls_name/cls_name_zh.yaml

@@ -0,0 +1,7 @@
+model_cls_name: ['人', '自行车', '汽车', '摩托车', '飞机', '公交车', '火车', '卡车', '船', '红绿灯', '消防栓', '停止标志',
+                '停车收费表', '长凳', '鸟', '猫', '狗', '马', '羊', '牛', '象', '熊', '斑马', '长颈鹿', '背包', '雨伞', '手提包', '领带',
+                '手提箱', '飞盘', '滑雪板', '单板滑雪', '运动球', '风筝', '棒球棒', '棒球手套', '滑板', '冲浪板', '网球拍', '瓶子', '红酒杯',
+                '杯子', '叉子', '刀', '勺', '碗', '香蕉', '苹果', '三明治', '橙子', '西兰花', '胡萝卜', '热狗', '比萨', '甜甜圈', '蛋糕',
+                '椅子', '长椅', '盆栽', '床', '餐桌', '马桶', '电视', '笔记本电脑', '鼠标', '遥控器', '键盘', '手机', '微波炉', '烤箱',
+                '烤面包机', '洗碗槽', '冰箱', '书', '时钟', '花瓶', '剪刀', '泰迪熊', '吹风机', '牙刷'
+            ]

model_config/model_name_p5_all.csv

@@ -0,0 +1,5 @@
+yolov5n
+yolov5s
+yolov5m
+yolov5l
+yolov5x

model_config/model_name_p5_all.yaml

@@ -0,0 +1 @@
+model_names: ["yolov5n", "yolov5s", "yolov5m", "yolov5l", "yolov5x"]

model_config/model_name_p5_n.csv

@@ -0,0 +1 @@
+yolov5n

model_config/model_name_p5_n.yaml

@@ -0,0 +1 @@
+model_names: ["yolov5n"]

model_config/model_name_p5_p6_all.yaml

@@ -0,0 +1 @@
+model_names: ["yolov5n", "yolov5s", "yolov5m", "yolov5l", "yolov5x", "yolov5n6", "yolov5s6", "yolov5m6", "yolov5l6", "yolov5x6"]

model_config/model_name_p6_all.csv

@@ -0,0 +1,5 @@
+yolov5n6
+yolov5s6
+yolov5m6
+yolov5l6
+yolov5x6

model_config/model_name_p6_all.yaml

@@ -0,0 +1 @@
+model_names: ["yolov5n6", "yolov5s6", "yolov5m6", "yolov5l6", "yolov5x6"]

model_download/yolov5_model_p5_all.sh

@@ -0,0 +1,8 @@
+cd ./yolov5
+
+# 下载YOLOv5模型
+wget -c -t 0 https://github.com/ultralytics/yolov5/releases/download/v6.1/yolov5n.pt
+wget -c -t 0 https://github.com/ultralytics/yolov5/releases/download/v6.1/yolov5s.pt
+wget -c -t 0 https://github.com/ultralytics/yolov5/releases/download/v6.1/yolov5m.pt
+wget -c -t 0 https://github.com/ultralytics/yolov5/releases/download/v6.1/yolov5l.pt
+wget -c -t 0 https://github.com/ultralytics/yolov5/releases/download/v6.1/yolov5x.pt

model_download/yolov5_model_p5_n.sh

@@ -0,0 +1,4 @@
+cd ./yolov5
+
+# 下载YOLOv5模型
+wget -c -t 0 https://github.com/ultralytics/yolov5/releases/download/v6.1/yolov5n.pt

model_download/yolov5_model_p6_all.sh

@@ -0,0 +1,8 @@
+cd ./yolov5
+
+# 下载YOLOv5模型
+wget -c -t 0 https://github.com/ultralytics/yolov5/releases/download/v6.1/yolov5n6.pt
+wget -c -t 0 https://github.com/ultralytics/yolov5/releases/download/v6.1/yolov5s6.pt
+wget -c -t 0 https://github.com/ultralytics/yolov5/releases/download/v6.1/yolov5m6.pt
+wget -c -t 0 https://github.com/ultralytics/yolov5/releases/download/v6.1/yolov5l6.pt
+wget -c -t 0 https://github.com/ultralytics/yolov5/releases/download/v6.1/yolov5x6.pt

requirements.txt

@@ -0,0 +1,45 @@
+# Base ----------------------------------------
+matplotlib>=3.2.2
+numpy>=1.22.3
+opencv-python-headless>=4.5.5.64
+Pillow>=7.1.2
+PyYAML>=5.3.1
+requests>=2.23.0
+scipy>=1.4.1  # Google Colab version
+torch>=1.7.0
+torchvision>=0.8.1
+tqdm>=4.41.0
+
+# Gradio YOLOv5 Det ----------------------------------------
+gradio>=3.0.3
+wget>=3.2
+rich>=12.2.0
+fpdf>=1.7.2
+plotly>=5.7.0
+bokeh>=2.4.2
+openpyxl>=3.0.10
+
+# Logging -------------------------------------
+tensorboard>=2.4.1
+# wandb
+
+# Plotting ------------------------------------
+pandas>=1.1.4
+seaborn>=0.11.0
+
+# Export --------------------------------------
+# coremltools>=4.1  # CoreML export
+# onnx>=1.9.0  # ONNX export
+# onnx-simplifier>=0.3.6  # ONNX simplifier
+# scikit-learn==0.19.2  # CoreML quantization
+# tensorflow>=2.4.1  # TFLite export
+# tensorflowjs>=3.9.0  # TF.js export
+# openvino-dev  # OpenVINO export
+
+# Extras --------------------------------------
+ipython  # interactive notebook
+psutil  # system utilization
+thop  # FLOPs computation
+# albumentations>=1.0.3
+# pycocotools>=2.0  # COCO mAP
+# roboflow

util/fonts_opt.py

@@ -0,0 +1,69 @@
+# font management
+# author: Zeng Yifu（曾逸夫）
+# creation time: 2022-05-01
+# email: [email protected]
+# project homepage: https://gitee.com/CV_Lab/gradio_yolov5_det
+
+import os
+import sys
+from pathlib import Path
+
+import wget
+from rich.console import Console
+
+ROOT_PATH = sys.path[0] # Project root directory
+
+# Chinese, English, Russian, Spanish, Arabic, Korean
+fonts_list = ["SimSun.ttf", "TimesNewRoman.ttf", "malgun.ttf"] # font list
+fonts_suffix = ["ttc", "ttf", "otf"] # font suffix
+
+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()
+
+
+# create font library
+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] # font name
+            Path(f"{ROOT_PATH}/fonts").mkdir(parents=True, exist_ok=True) # Create a directory
+
+            file_path = f"{ROOT_PATH}/fonts/{font_name}" # font path
+
+            try:
+                # Download font file
+                wget.download(v, file_path)
+            except Exception as e:
+                print("Path error! Program ended!")
+                print(e)
+                sys.exit()
+            else:
+                print()
+                console.print(f"{font_name} [bold green]font file download complete![/bold green] has been saved to: {file_path}")
+
+
+# Determine the font file
+def is_fonts(fonts_dir):
+    if os.path.isdir(fonts_dir):
+        # if the font library exists
+        f_list = os.listdir(fonts_dir) # local font library
+
+        font_diff = list(set(fonts_list).difference(set(f_list)))
+
+        if font_diff != []:
+            # font does not exist
+            console.print("[bold red] font does not exist, loading...[/bold red]")
+            add_fronts(font_diff) # Create a font library
+        else:
+            console.print(f"{fonts_list}[bold green]font already exists![/bold green]")
+    else:
+        # The font library does not exist, create a font library
+        console.print("[bold red]font library does not exist, creating...[/bold red]")
+        add_fronts(fonts_list) # Create a font library

util/pdf_opt.py

@@ -0,0 +1,78 @@
+# PDF management
+# author: Zeng Yifu
+# creation time: 2022-05-05
+
+from fpdf import FPDF
+
+
+# PDF generation class
+class PDF(FPDF):
+    # Reference: https://pyfpdf.readthedocs.io/en/latest/Tutorial/index.html
+    def header(self):
+        # Set Chinese font
+        self.add_font("SimSun", "", "./fonts/SimSun.ttf", uni=True)
+        self.set_font("SimSun", "", 16)
+        # Calculate width of title and position
+        w = self.get_string_width(title) + 6
+        self.set_x((210 - w) / 2)
+        # Colors of frame, background and text
+        self.set_draw_color(255, 255, 255)
+        self.set_fill_color(255, 255, 255)
+        self.set_text_color(0, 0, 0)
+        # Thickness of frame (1 mm)
+        # self.set_line_width(1)
+        # Title
+        self.cell(w, 9, title, 1, 1, "C", 1)
+        # Line break
+        self.ln(10)
+
+    def footer(self):
+        # Position at 1.5 cm from bottom
+        self.set_y(-15)
+        # Set Chinese font
+        self.add_font("SimSun", "", "./fonts/SimSun.ttf", uni=True)
+        self.set_font("SimSun", "", 12)
+        # Text color in gray
+        self.set_text_color(128)
+        # Page number
+        self.cell(0, 10, "Page " + str(self.page_no()), 0, 0, "C")
+
+    def chapter_title(self, num, label):
+        # Set Chinese font
+        self.add_font("SimSun", "", "./fonts/SimSun.ttf", uni=True)
+        self.set_font("SimSun", "", 12)
+        # Background color
+        self.set_fill_color(200, 220, 255)
+        # Title
+        # self.cell(0, 6, 'Chapter %d : %s' % (num, label), 0, 1, 'L', 1)
+        self.cell(0, 6, "Detection Result:", 0, 1, "L", 1)
+        # Line break
+        self.ln(4)
+
+    def chapter_body(self, name):
+
+        # Set Chinese font
+        self.add_font("SimSun", "", "./fonts/SimSun.ttf", uni=True)
+        self.set_font("SimSun", "", 12)
+        # Output justified text
+        self.multi_cell(0, 5, name)
+        # Line break
+        self.ln()
+        self.cell(0, 5, "--------------------------------------")
+
+    def print_chapter(self, num, title, name):
+        self.add_page()
+        self.chapter_title(num, title)
+        self.chapter_body(name)
+
+
+# pdf generation function
+def pdf_generate(input_file, output_file, title_):
+    global title
+
+    title = title_
+    pdf = PDF()
+    pdf.set_title(title)
+    pdf.set_author("Zeng Yifu")
+    pdf.print_chapter(1, "A RUNAWAY REEF", input_file)
+    pdf.output(output_file)