Guill-Bla commited on
Commit
f4ec0e7
·
verified ·
1 Parent(s): 39afbbb

Update tasks/image.py

Browse files
Files changed (1) hide show
  1. tasks/image.py +4 -1
tasks/image.py CHANGED
@@ -116,7 +116,10 @@ async def evaluate_image(request: ImageEvaluationRequest):
116
  true_boxes_list = []
117
 
118
  for batch_idx, (images, annotations) in enumerate(dataloader):
119
- print(f"Processing batch {batch_idx + 1}")
 
 
 
120
 
121
  # Parse true labels and boxes
122
  batch_true_labels = []
 
116
  true_boxes_list = []
117
 
118
  for batch_idx, (images, annotations) in enumerate(dataloader):
119
+ batch_size_current = len(images)
120
+ images_processed += batch_size_current
121
+
122
+ print(f"Processing batch {batch_idx + 1}: {images_processed}/{n_examples} images")
123
 
124
  # Parse true labels and boxes
125
  batch_true_labels = []