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 = []
|