58 lines
1.8 KiB
Python
58 lines
1.8 KiB
Python
|
import cv2
|
|||
|
|
|
|||
|
|
class FaceDetector:
|
|||
|
|
def __init__(self, cascade_path='haarcascade_frontalface_default.xml'):
|
|||
|
|
self.face_cascade = cv2.CascadeClassifier(cv2.data.haarcascades + cascade_path)
|
|||
|
|
|
|||
|
|
def find_faces(self, img, draw=True):
|
|||
|
|
gray = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)
|
|||
|
|
faces = self.face_cascade.detectMultiScale(gray, 1.3, 5)
|
|||
|
|
|
|||
|
|
bboxs = []
|
|||
|
|
for i, (x, y, w, h) in enumerate(faces):
|
|||
|
|
bbox = (x, y, w, h)
|
|||
|
|
bboxs.append([i, bbox, None]) # 'None' 用于保持与原代码结构一致，因为OpenCV不提供置信度分数
|
|||
|
|
|
|||
|
|
if draw:
|
|||
|
|
img = self.fancy_draw(img, bbox)
|
|||
|
|
|
|||
|
|
return img, bboxs
|
|||
|
|
|
|||
|
|
def fancy_draw(self, img, bbox, l=30, t=5, rt=1):
|
|||
|
|
x, y, w, h = bbox
|
|||
|
|
x1, y1 = x + w, y + h
|
|||
|
|
|
|||
|
|
# 绘制矩形
|
|||
|
|
cv2.rectangle(img, bbox, (0, 255, 255), rt)
|
|||
|
|
|
|||
|
|
# 绘制角落
|
|||
|
|
# 左上角
|
|||
|
|
cv2.line(img, (x, y), (x+l, y), (255, 0, 255), t)
|
|||
|
|
cv2.line(img, (x, y), (x, y+l), (255, 0, 255), t)
|
|||
|
|
# 右上角
|
|||
|
|
cv2.line(img, (x1, y), (x1-l, y), (255, 0, 255), t)
|
|||
|
|
cv2.line(img, (x1, y), (x1, y+l), (255, 0, 255), t)
|
|||
|
|
# 左下角
|
|||
|
|
cv2.line(img, (x, y1), (x+l, y1), (255, 0, 255), t)
|
|||
|
|
cv2.line(img, (x, y1), (x, y1-l), (255, 0, 255), t)
|
|||
|
|
# 右下角
|
|||
|
|
cv2.line(img, (x1, y1), (x1-l, y1), (255, 0, 255), t)
|
|||
|
|
cv2.line(img, (x1, y1), (x1, y1-l), (255, 0, 255), t)
|
|||
|
|
|
|||
|
|
return img
|
|||
|
|
|
|||
|
|
# 使用示例
|
|||
|
|
if __name__ == "__main__":
|
|||
|
|
detector = FaceDetector()
|
|||
|
|
cap = cv2.VideoCapture(0) # 使用默认摄像头
|
|||
|
|
|
|||
|
|
while True:
|
|||
|
|
success, img = cap.read()
|
|||
|
|
img, bboxs = detector.find_faces(img)
|
|||
|
|
|
|||
|
|
cv2.imshow("Image", img)
|
|||
|
|
if cv2.waitKey(1) & 0xFF == ord('q'):
|
|||
|
|
break
|
|||
|
|
|
|||
|
|
cap.release()
|
|||
|
|
cv2.destroyAllWindows()
|