fixed websocket image piepline

2026-05-29 18:13:15 +02:00
parent 90e86d6f48
commit 8180767db7
1 changed files with 0 additions and 2 deletions
@@ -115,8 +115,6 @@ async def predict_face(websocket: WebSocket):
        scale_w = W / IMAGE_SIZE_YOLO
        scale_h = H / IMAGE_SIZE_YOLO
        image = cv2.cvtColor(frame, cv2.COLOR_BGR2RGB)
        image = transform_face(image).unsqueeze(0).to(device)
        with torch.no_grad():
            pred = modeL_face(image)