Spaces:
Sleeping
Sleeping
Update tasks/image.py
Browse files- 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 |
-
|
|
|
|
|
|
|
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 = []
|