Spaces:
Sleeping
Sleeping
Update tasks/image.py
Browse files- tasks/image.py +1 -3
tasks/image.py
CHANGED
@@ -152,10 +152,8 @@ def compute_max_iou(true_boxes, pred_box):
|
|
152 |
max_iou = max(max_iou, iou)
|
153 |
return max_iou
|
154 |
|
155 |
-
|
156 |
-
# description=DESCRIPTION)
|
157 |
async def evaluate_image(model_path: str = "models/camera_models/yolo11n_default.engine",request: ImageEvaluationRequest = ImageEvaluationRequest()):
|
158 |
-
# def evaluate_image(model_path: str, request: ImageEvaluationRequest = ImageEvaluationRequest()):
|
159 |
"""
|
160 |
Evaluate image classification and object detection for forest fire smoke.
|
161 |
|
|
|
152 |
max_iou = max(max_iou, iou)
|
153 |
return max_iou
|
154 |
|
155 |
+
@router.post(ROUTE, tags=["Image Task"], description=DESCRIPTION)
|
|
|
156 |
async def evaluate_image(model_path: str = "models/camera_models/yolo11n_default.engine",request: ImageEvaluationRequest = ImageEvaluationRequest()):
|
|
|
157 |
"""
|
158 |
Evaluate image classification and object detection for forest fire smoke.
|
159 |
|