person det example

examples/face_detection/README.md

@@ -0,0 +1,2 @@
+
+#

examples/face_recognition/README.md

@@ -0,0 +1,2 @@
+
+#

examples/person_detection/README.md

@@ -0,0 +1 @@
+# Person Detection Powered by SCRFD

examples/person_detection/scrfd_person.py

@@ -0,0 +1,49 @@
+import datetime
+import numpy as np
+import os
+import os.path as osp
+import glob
+import cv2
+import insightface
+
+assert insightface.__version__>='0.4'
+
+
+def detect_person(img, detector):
+    bboxes, kpss = detector.detect(img)
+    bboxes = np.round(bboxes[:,:4]).astype(np.int)
+    kpss = np.round(kpss).astype(np.int)
+    kpss[:,:,0] = np.clip(kpss[:,:,0], 0, img.shape[1])
+    kpss[:,:,1] = np.clip(kpss[:,:,1], 0, img.shape[0])
+    vbboxes = bboxes.copy()
+    vbboxes[:,0] = kpss[:, 0, 0]
+    vbboxes[:,1] = kpss[:, 0, 1]
+    vbboxes[:,2] = kpss[:, 4, 0]
+    vbboxes[:,3] = kpss[:, 4, 1]
+    return bboxes, vbboxes
+
+if __name__ == '__main__':
+    import glob
+    detector = insightface.model_zoo.get_model('scrfd_person_2.5g.onnx', download=True)
+    detector.prepare(0, nms_thresh=0.5, input_size=(640, 640))
+    img_paths = glob.glob('data/images/*.jpg')
+    for img_path in img_paths:
+        img = cv2.imread(img_path)
+        bboxes, vbboxes = detect_person(img, detector)
+        for i in range(bboxes.shape[0]):
+            bbox = bboxes[i]
+            vbbox = vbboxes[i]
+            x1,y1,x2,y2 = bbox
+            vx1,vy1,vx2,vy2 = vbbox
+            cv2.rectangle(img, (x1,y1)  , (x2,y2) , (0,255,0) , 1)
+            alpha = 0.8
+            color = (255, 0, 0)
+            for c in range(3):
+                img[vy1:vy2,vx1:vx2,c] = img[vy1:vy2, vx1:vx2, c]*alpha + color[c]*(1.0-alpha)
+            cv2.circle(img, (vx1,vy1) , 1, color , 2)
+            cv2.circle(img, (vx1,vy2) , 1, color , 2)
+            cv2.circle(img, (vx2,vy1) , 1, color , 2)
+            cv2.circle(img, (vx2,vy2) , 1, color , 2)
+        filename = img_path.split('/')[-1]
+        cv2.imwrite('./outputs/%s'%filename, img)
+