From 4db3d96369c3eb9a189a54df1f3d53f2d9f0bac6 Mon Sep 17 00:00:00 2001
From: Paul <22284856+ProgramComputer@users.noreply.github.com>
Date: Tue, 19 Sep 2023 14:41:41 -0500
Subject: [PATCH 1/2] Update README.md
---
README.md | 1 +
1 file changed, 1 insertion(+)
diff --git a/README.md b/README.md
index 6c28a0c..eedaa5e 100644
--- a/README.md
+++ b/README.md
@@ -33,3 +33,4 @@ opencv-python==4.0.0.21 to opencv-python==4.6.0.66
Acknowledgement:
Face detection using mtcnn https://github.com/ipazc/mtcnn.
Face detection using retinaface https://github.com/peteryuX/retinaface-tf2
+Face detection using scrfd https://github.com/deepinsight/insightface/tree/master/detection/scrfd
From 1edd313b5930d54d0cdf78e570fbb9cbd89978a4 Mon Sep 17 00:00:00 2001
From: Paul <22284856+ProgramComputer@users.noreply.github.com>
Date: Tue, 19 Sep 2023 14:43:56 -0500
Subject: [PATCH 2/2] Update pose_detection_scrfd.py
---
pose_detection_scrfd.py | 6 +++---
1 file changed, 3 insertions(+), 3 deletions(-)
diff --git a/pose_detection_scrfd.py b/pose_detection_scrfd.py
index be11c5a..0019a39 100644
--- a/pose_detection_scrfd.py
+++ b/pose_detection_scrfd.py
@@ -65,9 +65,9 @@ def detect(self, srcimg):
# inference output
scores_list, bboxes_list, kpss_list = [], [], []
for idx, stride in enumerate(self._feat_stride_fpn):
- scores = outs[idx * self.fmc][0]
- bbox_preds = outs[idx * self.fmc + 1][0] * stride
- kps_preds = outs[idx * self.fmc + 2][0] * stride
+ scores = outs[idx][0]
+ bbox_preds = outs[idx + self.fmc][0] * stride
+ kps_preds = outs[idx + self.fmc * 2][0] * stride
height = blob.shape[2] // stride
width = blob.shape[3] // stride
anchor_centers = np.stack(np.mgrid[:height, :width][::-1], axis=-1).astype(np.float32)