change model path
Browse files- tasks/image.py +2 -2
tasks/image.py
CHANGED
|
@@ -96,7 +96,7 @@ async def evaluate_image(request: ImageEvaluationRequest):
|
|
| 96 |
test_dataset = train_test["test"]
|
| 97 |
|
| 98 |
# model = YOLOv10("best.pt")
|
| 99 |
-
model = YOLOv10("
|
| 100 |
|
| 101 |
#device = "cuda" if torch.cuda.is_available() else "cpu"
|
| 102 |
#model = model.to(device)
|
|
@@ -179,4 +179,4 @@ async def evaluate_image(request: ImageEvaluationRequest):
|
|
| 179 |
}
|
| 180 |
}
|
| 181 |
|
| 182 |
-
return results
|
|
|
|
| 96 |
test_dataset = train_test["test"]
|
| 97 |
|
| 98 |
# model = YOLOv10("best.pt")
|
| 99 |
+
model = YOLOv10("yolov10s.engine")
|
| 100 |
|
| 101 |
#device = "cuda" if torch.cuda.is_available() else "cpu"
|
| 102 |
#model = model.to(device)
|
|
|
|
| 179 |
}
|
| 180 |
}
|
| 181 |
|
| 182 |
+
return results
|