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