v03 update

app.py

@@ -1,8 +1,8 @@
 # Gradio YOLOv5 Det v0.3
-# author：Zeng Yifu（曾逸夫）
-# creation time：2022-05-09
-# email：zyfiy1314@163.com
-# Project homepage：https://gitee.com/CV_Lab/gradio_yolov5_det
+# author: Zeng Yifu（曾逸夫）
+# creation time: 2022-05-09
+# email: zyfiy1314@163.com
+# project homepage: https://gitee.com/CV_Lab/gradio_yolov5_det
 
 import os
 
@@ -74,7 +74,7 @@ def parse_args(known=False):
     parser.add_argument(
         "--device",
         "-dev",
-        default="cuda:0",
+        default="cpu",
         type=str,
         help="cuda or cpu",
     )

util/fonts_opt.py

@@ -1,6 +1,8 @@
-# 字体管理
-# 创建人：曾逸夫
-# 创建时间：2022-05-01
+# font management
+# author: Zeng Yifu（曾逸夫）
+# creation time: 2022-05-01
+# email: zyfiy1314@163.com
+# project homepage: https://gitee.com/CV_Lab/gradio_yolov5_det
 
 import os
 import sys
@@ -9,11 +11,11 @@ from pathlib import Path
 import wget
 from rich.console import Console
 
-ROOT_PATH = sys.path[0]  # 项目根目录
+ROOT_PATH = sys.path[0] # Project root directory
 
-# 中文、英文、俄语、西班牙语、阿拉伯语、韩语
-fonts_list = ["SimSun.ttf", "TimesNewRoman.ttf", "malgun.ttf"]  # 字体列表
-fonts_suffix = ["ttc", "ttf", "otf"]  # 字体后缀
+# 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",
@@ -23,45 +25,45 @@ data_url_dict = {
 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]  # 字体名称
-            Path(f"{ROOT_PATH}/fonts").mkdir(parents=True, exist_ok=True)  # 创建目录
+            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}"  # 字体路径
+            file_path = f"{ROOT_PATH}/fonts/{font_name}" # font path
 
             try:
-                # 下载字体文件
+                # Download font file
                 wget.download(v, file_path)
             except Exception as e:
-                print("路径错误！程序结束！")
+                print("Path error! Program ended!")
                 print(e)
                 sys.exit()
             else:
                 print()
-                console.print(f"{font_name} [bold green]字体文件下载完成！[/bold green] 已保存至：{file_path}")
+                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):
-        # 如果字体库存在
-        f_list = os.listdir(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 != []:
-            # 字体不存在
-            console.print("[bold red]字体不存在，正在加载。。。[/bold red]")
-            add_fronts(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]字体已存在！[/bold green]")
+            console.print(f"{fonts_list}[bold green]font already exists![/bold green]")
     else:
-        # 字体库不存在，创建字体库
-        console.print("[bold red]字体库不存在，正在创建。。。[/bold red]")
-        add_fronts(fonts_list)  # 创建字体库
+        # 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

@@ -1,15 +1,15 @@
-# PDF管理
-# 创建人：曾逸夫
-# 创建时间：2022-05-05
+# PDF management
+# author: Zeng Yifu
+# creation time: 2022-05-05
 
 from fpdf import FPDF
 
 
-# PDF生成类
+# PDF generation class
 class PDF(FPDF):
-    # 参考：https://pyfpdf.readthedocs.io/en/latest/Tutorial/index.html
+    # 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
@@ -29,7 +29,7 @@ class PDF(FPDF):
     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
@@ -38,20 +38,20 @@ class PDF(FPDF):
         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, "检测结果：", 0, 1, "L", 1)
+        self.cell(0, 6, "Test 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
@@ -66,7 +66,7 @@ class PDF(FPDF):
         self.chapter_body(name)
 
 
-# pdf生成函数
+# pdf generation function
 def pdf_generate(input_file, output_file, title_):
     global title
 
@@ -75,4 +75,4 @@ def pdf_generate(input_file, output_file, title_):
     pdf.set_title(title)
     pdf.set_author("Zeng Yifu")
     pdf.print_chapter(1, "A RUNAWAY REEF", input_file)
-    pdf.output(output_file)
+    pdf.output(output_file)