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
|