Spaces:
Runtime error
Runtime error
Zengyf-CVer
commited on
Commit
•
ec8e77a
1
Parent(s):
0931150
app update
Browse files- app.py +5 -2
- requirements.txt +0 -1
app.py
CHANGED
@@ -7,6 +7,7 @@
|
|
7 |
import argparse
|
8 |
import csv
|
9 |
import json
|
|
|
10 |
import sys
|
11 |
from pathlib import Path
|
12 |
|
@@ -18,6 +19,8 @@ from PIL import Image, ImageDraw, ImageFont
|
|
18 |
from util.fonts_opt import is_fonts
|
19 |
from util.pdf_opt import pdf_generate
|
20 |
|
|
|
|
|
21 |
ROOT_PATH = sys.path[0] # 根目录
|
22 |
|
23 |
# 模型路径
|
@@ -37,7 +40,7 @@ FONTSIZE = 25
|
|
37 |
|
38 |
|
39 |
def parse_args(known=False):
|
40 |
-
parser = argparse.ArgumentParser(description="Gradio YOLOv5 Det v0.2")
|
41 |
parser.add_argument(
|
42 |
"--model_name", "-mn", default="yolov5s", type=str, help="model name"
|
43 |
)
|
@@ -320,7 +323,7 @@ def main(args):
|
|
320 |
outputs = [outputs_img, outputs02_json, outputs03_pdf]
|
321 |
|
322 |
# 标题
|
323 |
-
title = "基于Gradio的YOLOv5通用目标检测系统v0.2"
|
324 |
# 描述
|
325 |
description = "<div align='center'>可自定义目标检测模型、安装简单、使用方便</div><div align='center'>Customizable object detection model, easy to install and easy to use</div>"
|
326 |
|
|
|
7 |
import argparse
|
8 |
import csv
|
9 |
import json
|
10 |
+
import os
|
11 |
import sys
|
12 |
from pathlib import Path
|
13 |
|
|
|
19 |
from util.fonts_opt import is_fonts
|
20 |
from util.pdf_opt import pdf_generate
|
21 |
|
22 |
+
os.system("pip install gradio==2.9b40")
|
23 |
+
|
24 |
ROOT_PATH = sys.path[0] # 根目录
|
25 |
|
26 |
# 模型路径
|
|
|
40 |
|
41 |
|
42 |
def parse_args(known=False):
|
43 |
+
parser = argparse.ArgumentParser(description="Gradio YOLOv5 Det v0.2.2")
|
44 |
parser.add_argument(
|
45 |
"--model_name", "-mn", default="yolov5s", type=str, help="model name"
|
46 |
)
|
|
|
323 |
outputs = [outputs_img, outputs02_json, outputs03_pdf]
|
324 |
|
325 |
# 标题
|
326 |
+
title = "基于Gradio的YOLOv5通用目标检测系统v0.2.2"
|
327 |
# 描述
|
328 |
description = "<div align='center'>可自定义目标检测模型、安装简单、使用方便</div><div align='center'>Customizable object detection model, easy to install and easy to use</div>"
|
329 |
|
requirements.txt
CHANGED
@@ -8,7 +8,6 @@ requests>=2.23.0
|
|
8 |
scipy>=1.4.1
|
9 |
torch>=1.7.0
|
10 |
torchvision>=0.8.1
|
11 |
-
gradio==2.9b40
|
12 |
tqdm>=4.41.0
|
13 |
wget>=3.2
|
14 |
rich>=12.2.0
|
|
|
8 |
scipy>=1.4.1
|
9 |
torch>=1.7.0
|
10 |
torchvision>=0.8.1
|
|
|
11 |
tqdm>=4.41.0
|
12 |
wget>=3.2
|
13 |
rich>=12.2.0
|