Spaces:
Build error
Build error
Matteo Sirri
commited on
Commit
·
a67969d
1
Parent(s):
d41f3b7
fix: fix typo
Browse files
app.py
CHANGED
|
@@ -37,7 +37,6 @@ def frcnn_motsynth(image):
|
|
| 37 |
image_tensor = transformEval(image, None)[0]
|
| 38 |
image_tensor = image_tensor.to(device)
|
| 39 |
prediction = model([image_tensor])[0]
|
| 40 |
-
prediction = prediction.to(device)
|
| 41 |
image_w_bbox = add_bbox(image_tensor, prediction, 0.80)
|
| 42 |
torchvision.io.write_png(image_w_bbox, "custom_out.png")
|
| 43 |
return "custom_out.png"
|
|
@@ -49,7 +48,6 @@ def frcnn_coco(image):
|
|
| 49 |
image_tensor = transformEval(image, None)[0]
|
| 50 |
image_tensor = image_tensor.to(device)
|
| 51 |
prediction = model([image_tensor])[0]
|
| 52 |
-
prediction = prediction.to(device)
|
| 53 |
image_w_bbox = add_bbox(image_tensor, prediction, 0.80)
|
| 54 |
torchvision.io.write_png(image_w_bbox, "baseline_out.png")
|
| 55 |
return "baseline_out.png"
|
|
|
|
| 37 |
image_tensor = transformEval(image, None)[0]
|
| 38 |
image_tensor = image_tensor.to(device)
|
| 39 |
prediction = model([image_tensor])[0]
|
|
|
|
| 40 |
image_w_bbox = add_bbox(image_tensor, prediction, 0.80)
|
| 41 |
torchvision.io.write_png(image_w_bbox, "custom_out.png")
|
| 42 |
return "custom_out.png"
|
|
|
|
| 48 |
image_tensor = transformEval(image, None)[0]
|
| 49 |
image_tensor = image_tensor.to(device)
|
| 50 |
prediction = model([image_tensor])[0]
|
|
|
|
| 51 |
image_w_bbox = add_bbox(image_tensor, prediction, 0.80)
|
| 52 |
torchvision.io.write_png(image_w_bbox, "baseline_out.png")
|
| 53 |
return "baseline_out.png"
|