code

app.py

@@ -7,6 +7,20 @@ import torch
 import numpy as np
 import gradio as gr
 
+# 함수 정의: 사각형 그리고 정확도 표시
+def draw_rect_with_conf(image, detections):
+    for detection in detections:
+        # 탐지된 객체의 경계 상자 좌표 (NumPy 배열)와 신뢰도 점수
+        box, _, conf, _, _ = detection
+        x1, y1, x2, y2 = box.astype(int)
+
+        # 사각형 그리기
+        cv2.rectangle(image, (x1, y1), (x2, y2), (0, 255, 0), 2)
+
+        # 신뢰도 표시
+        cv2.putText(image, f'Accuracy: {conf:.2f}', (x1, y1 - 10), cv2.FONT_HERSHEY_SIMPLEX, 0.5, (0, 255, 0), 2)
+    return image
+
 def detect_faces(input_img):
     # download model
     model_path = hf_hub_download(