Spaces:
Build error
Build error
Matteo Sirri
commited on
Commit
·
49a08b1
1
Parent(s):
99dceec
style: refactoring
Browse files
app.py
CHANGED
|
@@ -35,7 +35,6 @@ def frcnn_motsynth(image):
|
|
| 35 |
model = load_model()
|
| 36 |
transformEval = presets.DetectionPresetEval()
|
| 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")
|
|
@@ -46,7 +45,6 @@ def frcnn_coco(image):
|
|
| 46 |
model = load_model(baseline=True)
|
| 47 |
transformEval = presets.DetectionPresetEval()
|
| 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")
|
|
|
|
| 35 |
model = load_model()
|
| 36 |
transformEval = presets.DetectionPresetEval()
|
| 37 |
image_tensor = transformEval(image, None)[0]
|
|
|
|
| 38 |
prediction = model([image_tensor])[0]
|
| 39 |
image_w_bbox = add_bbox(image_tensor, prediction, 0.80)
|
| 40 |
torchvision.io.write_png(image_w_bbox, "custom_out.png")
|
|
|
|
| 45 |
model = load_model(baseline=True)
|
| 46 |
transformEval = presets.DetectionPresetEval()
|
| 47 |
image_tensor = transformEval(image, None)[0]
|
|
|
|
| 48 |
prediction = model([image_tensor])[0]
|
| 49 |
image_w_bbox = add_bbox(image_tensor, prediction, 0.80)
|
| 50 |
torchvision.io.write_png(image_w_bbox, "baseline_out.png")
|