more dubugging
Browse files
app.py
CHANGED
@@ -29,19 +29,20 @@ def read_image(image_file: bytes = File(...)):
|
|
29 |
target_sizes = torch.tensor([image.size[::-1]])
|
30 |
results = processor.post_process_object_detection(outputs, target_sizes=target_sizes, threshold=0.9)[0]
|
31 |
print("results pushed")
|
32 |
-
for score, label, box in zip(results["scores"], results["labels"], results["boxes"]):
|
33 |
-
box = [round(i, 2) for i in box.tolist()]
|
34 |
-
print(
|
35 |
-
f"Detected {model.config.id2label[label.item()]} with confidence "
|
36 |
-
f"{round(score.item(), 3)} at location {box}"
|
37 |
-
)
|
38 |
-
print(results["scores"])
|
39 |
-
print(results["labels"])
|
40 |
-
print(results["boxes"])
|
41 |
response = {}
|
42 |
response['scores'] = results['scores']
|
43 |
response['labels'] = results['labels']
|
44 |
response['boxes'] = results['boxes']
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
45 |
return response
|
46 |
|
47 |
|
|
|
29 |
target_sizes = torch.tensor([image.size[::-1]])
|
30 |
results = processor.post_process_object_detection(outputs, target_sizes=target_sizes, threshold=0.9)[0]
|
31 |
print("results pushed")
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
32 |
response = {}
|
33 |
response['scores'] = results['scores']
|
34 |
response['labels'] = results['labels']
|
35 |
response['boxes'] = results['boxes']
|
36 |
+
for score, label, box in zip(results["scores"], results["labels"], results["boxes"]):
|
37 |
+
box = [round(i, 2) for i in box.tolist()]
|
38 |
+
print(model.config.id2label[label.item()])
|
39 |
+
print(score.item())
|
40 |
+
print(box)
|
41 |
+
# print(
|
42 |
+
# f"Detected {model.config.id2label[label.item()]} with confidence "
|
43 |
+
# f"{round(score.item(), 3)} at location {box}"
|
44 |
+
# )
|
45 |
+
|
46 |
return response
|
47 |
|
48 |
|