marvin/CNN_Website
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

added semaphore

Browse Source
This commit is contained in:
marvin
2026-05-29 19:38:29 +02:00
parent 0ca5df8737
commit 0fba1a8a72
1 changed files with 24 additions and 23 deletions
Download Patch File Download Diff File Expand all files Collapse all files
api/src/server.py
+24 -23
View File
@@ -112,36 +112,37 @@ async def predict_face(websocket: WebSocket):
continue continue
last = time.time() last = time.time()
with sem_ai:
np_arr = np.frombuffer(jpg_bytes, np.uint8) np_arr = np.frombuffer(jpg_bytes, np.uint8)
frame = cv2.imdecode(np_arr, cv2.IMREAD_COLOR) frame = cv2.imdecode(np_arr, cv2.IMREAD_COLOR)
frame = cv2.cvtColor(frame, cv2.COLOR_BGR2RGB) frame = cv2.cvtColor(frame, cv2.COLOR_BGR2RGB)
image = Image.fromarray(frame) image = Image.fromarray(frame)
H, W, _ = frame.shape H, W, _ = frame.shape
scale_w = W / IMAGE_SIZE_YOLO scale_w = W / IMAGE_SIZE_YOLO
scale_h = H / IMAGE_SIZE_YOLO scale_h = H / IMAGE_SIZE_YOLO
image = transform_face(image).unsqueeze(0).to(device) image = transform_face(image).unsqueeze(0).to(device)
with torch.no_grad(): with torch.no_grad():
pred = modeL_face(image) pred = modeL_face(image)
bboxes, _, _ = convert_prediction( bboxes, _, _ = convert_prediction(
pred.squeeze(0), pred.squeeze(0),
image.squeeze(0), image.squeeze(0),
threshold=0.9 threshold=0.9
) )
boxes_to_send = [] boxes_to_send = []
for bbox in bboxes: for bbox in bboxes:
xmin = int(bbox[0] * scale_w) xmin = int(bbox[0] * scale_w)
ymin = int(bbox[1] * scale_h) ymin = int(bbox[1] * scale_h)
xmax = int(bbox[2] * scale_w) xmax = int(bbox[2] * scale_w)
ymax = int(bbox[3] * scale_h) ymax = int(bbox[3] * scale_h)
boxes_to_send.append([xmin, ymin, xmax, ymax]) boxes_to_send.append([xmin, ymin, xmax, ymax])
await websocket.send_json({"bboxes": boxes_to_send}) await websocket.send_json({"bboxes": boxes_to_send})
except WebSocketDisconnect: except WebSocketDisconnect:
print("Client disconnected") print("Client disconnected")