Guill-Bla commited on
Commit
61936a9
·
verified ·
1 Parent(s): 678ffd5

Update tasks/image.py

Browse files
Files changed (1) hide show
  1. tasks/image.py +4 -1
tasks/image.py CHANGED
@@ -115,7 +115,10 @@ async def evaluate_image(request: ImageEvaluationRequest):
115
  predictions = []
116
  pred_boxes = []
117
  true_boxes_list = []
118
-
 
 
 
119
  for batch_idx, (images, annotations) in enumerate(dataloader):
120
  batch_size_current = len(images)
121
  images_processed += batch_size_current
 
115
  predictions = []
116
  pred_boxes = []
117
  true_boxes_list = []
118
+
119
+ n_examples = len(test_dataset)
120
+ images_processed = 0
121
+
122
  for batch_idx, (images, annotations) in enumerate(dataloader):
123
  batch_size_current = len(images)
124
  images_processed += batch_size_current