Spaces:
Build error
Build error
Matteo Sirri
commited on
Commit
·
369cd4c
1
Parent(s):
b0c3cbd
fix: fix description
Browse files
app.py
CHANGED
@@ -29,7 +29,7 @@ def load_model(baseline: bool = False):
|
|
29 |
|
30 |
|
31 |
def frcnn_motsynth(image):
|
32 |
-
model = load_model(
|
33 |
transformEval = presets.DetectionPresetEval()
|
34 |
image_tensor = transformEval(image, None)[0]
|
35 |
image_tensor = image_tensor.to(device)
|
@@ -51,7 +51,7 @@ def frcnn_coco(image):
|
|
51 |
|
52 |
|
53 |
title = "Domain shift adaption on pedestrian detection with Faster R-CNN"
|
54 |
-
description = "School in AI: Deep Learning, Vision and Language for Industry - second edition final project work by Matteo Sirri
|
55 |
examples = ["001.jpg", "002.jpg", "003.jpg",
|
56 |
"004.jpg", "005.jpg", "006.jpg", "007.jpg", ]
|
57 |
|
@@ -62,4 +62,4 @@ io_custom = gr.Interface(frcnn_motsynth, gr.Image(type="pil"), gr.Image(
|
|
62 |
type="file", shape=(1920, 1080), label="Faster R-CNN trained on MOTSynth + FT on MOT17"))
|
63 |
|
64 |
gr.Parallel(io_baseline, io_custom, title=title,
|
65 |
-
description=description, examples=examples
|
|
|
29 |
|
30 |
|
31 |
def frcnn_motsynth(image):
|
32 |
+
model = load_model()
|
33 |
transformEval = presets.DetectionPresetEval()
|
34 |
image_tensor = transformEval(image, None)[0]
|
35 |
image_tensor = image_tensor.to(device)
|
|
|
51 |
|
52 |
|
53 |
title = "Domain shift adaption on pedestrian detection with Faster R-CNN"
|
54 |
+
description = '<p style="text-align:center">School in AI: Deep Learning, Vision and Language for Industry - second edition final project work by Matteo Sirri.</p> '
|
55 |
examples = ["001.jpg", "002.jpg", "003.jpg",
|
56 |
"004.jpg", "005.jpg", "006.jpg", "007.jpg", ]
|
57 |
|
|
|
62 |
type="file", shape=(1920, 1080), label="Faster R-CNN trained on MOTSynth + FT on MOT17"))
|
63 |
|
64 |
gr.Parallel(io_baseline, io_custom, title=title,
|
65 |
+
description=description, examples=examples).launch(enable_queue=True)
|