sarim commited on
Commit
cf572c0
·
1 Parent(s): c354df6

more dubugging

Browse files
Files changed (1) hide show
  1. app.py +10 -9
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