Update tasks/image.py

tasks/image.py

@@ -185,8 +185,9 @@ async def evaluate_image(request: ImageEvaluationRequest):
     true_boxes_list = []  
 
     for batch_images, batch_annotations in dataloader:
-        image_inputs = feature_extractor(images=batch_images, return_tensors="pt", padding=True).pixel_values
-    
+        # image_inputs = feature_extractor(images=batch_images, return_tensors="pt", padding=True).pixel_values
+        image_inputs = feature_extractor(images=[img.permute(1, 2, 0).numpy() for img in batch_images], return_tensors="pt", padding=True).pixel_values
+
         # Perform inference
         with torch.no_grad():
             outputs = model(pixel_values=image_inputs)