Initial commit

main.py

@@ -9,7 +9,7 @@ import supervision as sv
 from ultralytics import YOLO
 
 app = FastAPI()
-model = YOLO("models/best_v3.pt", task="detect")
+model = YOLO("models/best_v2.pt", task="detect")
 
 
 def parse_detection(detections):
@@ -56,7 +56,7 @@ def infer(image):
     image_arr = np.frombuffer(image, np.uint8)
     image = cv2.imdecode(image_arr, cv2.IMREAD_COLOR)
     image = cv2.resize(image, (640, 640))
-    results = model(image)[0]
+    results = model(image, conf=0.6, iou=0.25, imgsz=640)[0]
     width, height = results.orig_shape[1], results.orig_shape[0]
     print(results.speed)
     detections = sv.Detections.from_ultralytics(results)