changed yolo design
This commit is contained in:
+1
-1
@@ -111,7 +111,7 @@ async def predict_face(websocket: WebSocket):
|
||||
#if time.time() - last < 1:
|
||||
#continue
|
||||
|
||||
last = time.time()
|
||||
#last = time.time()
|
||||
with sem_ai:
|
||||
|
||||
np_arr = np.frombuffer(jpg_bytes, np.uint8)
|
||||
|
||||
@@ -75,7 +75,7 @@ def use_webcam(grid, img_size):
|
||||
|
||||
prediction = model(image)
|
||||
|
||||
bboxes, grid_ob, grid_noob = convert_prediction(prediction.squeeze(0), image.squeeze(0), threshold=0.9)
|
||||
bboxes, grid_ob, grid_noob = convert_prediction(prediction.squeeze(0), image.squeeze(0), threshold=0.95)
|
||||
|
||||
for bbox in grid_noob:
|
||||
xmin = int(bbox[0] * scale_w)
|
||||
|
||||
Reference in New Issue
Block a user