submission-template-yolo

Sleeping

Guill-Bla commited on 26 days ago

Commit

3f1f9b9

verified ·

1 Parent(s): e24aace

Update tasks/image.py

Files changed (1) hide show

tasks/image.py CHANGED Viewed

@@ -102,7 +102,7 @@ async def evaluate_image(request: ImageEvaluationRequest):
     from ultralytics import YOLO
     # Load the trained YOLOv8 model
-    yolov8_model = YOLO("./../yolov8n.pt")
     predictions = []
     true_labels = []

     from ultralytics import YOLO
     # Load the trained YOLOv8 model
+    model = YOLO("./../best.pt")
     predictions = []
     true_labels = []