Spaces:
Running
Running
wangli
commited on
Commit
·
36ce8fe
1
Parent(s):
4868d0a
feat: add reference
Browse files
README.md
CHANGED
@@ -11,3 +11,5 @@ license: other
|
|
11 |
---
|
12 |
|
13 |
Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
|
|
|
|
|
|
11 |
---
|
12 |
|
13 |
Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
|
14 |
+
|
15 |
+
参考: https://huggingface.co/spaces/ClassCat/YOLOS-Object-Detection
|
app.py
CHANGED
@@ -10,14 +10,11 @@ import matplotlib.patches as patches
|
|
10 |
import io
|
11 |
from random import choice
|
12 |
|
13 |
-
|
14 |
image_processor_tiny = AutoImageProcessor.from_pretrained("hustvl/yolos-tiny")
|
15 |
model_tiny = AutoModelForObjectDetection.from_pretrained("hustvl/yolos-tiny")
|
16 |
|
17 |
-
|
18 |
import gradio as gr
|
19 |
|
20 |
-
|
21 |
COLORS = ["#ff7f7f", "#ff7fbf", "#ff7fff", "#bf7fff",
|
22 |
"#7f7fff", "#7fbfff", "#7fffff", "#7fffbf",
|
23 |
"#7fff7f", "#bfff7f", "#ffff7f", "#ffbf7f"]
|
|
|
10 |
import io
|
11 |
from random import choice
|
12 |
|
|
|
13 |
image_processor_tiny = AutoImageProcessor.from_pretrained("hustvl/yolos-tiny")
|
14 |
model_tiny = AutoModelForObjectDetection.from_pretrained("hustvl/yolos-tiny")
|
15 |
|
|
|
16 |
import gradio as gr
|
17 |
|
|
|
18 |
COLORS = ["#ff7f7f", "#ff7fbf", "#ff7fff", "#bf7fff",
|
19 |
"#7f7fff", "#7fbfff", "#7fffff", "#7fffbf",
|
20 |
"#7fff7f", "#bfff7f", "#ffff7f", "#ffbf7f"]
|