200 lines
6.7 KiB
Python
200 lines
6.7 KiB
Python
import sys
|
||
import random
|
||
import cv2
|
||
from cvzone.FaceDetectionModule import FaceDetector
|
||
from PySide6.QtCore import QTimer
|
||
from PySide6.QtGui import QImage, QPixmap
|
||
from PySide6.QtWidgets import QApplication, QWidget, QLabel, QPushButton
|
||
|
||
|
||
class FaceRandomApp(QWidget):
|
||
def __init__(self):
|
||
super().__init__()
|
||
|
||
self.setWindowTitle("Face Random Selector")
|
||
|
||
# 状态：normal → random
|
||
self.state = "normal"
|
||
self.selected_face_index = -1
|
||
self.static_frame = None # 保存随机状态下的静态画面
|
||
self.all_faces_snapshot = [] # 保存所有人脸信息快照
|
||
|
||
# ---------------- 摄像头 ----------------
|
||
self.cap = cv2.VideoCapture(0)
|
||
|
||
# 设置摄像头分辨率
|
||
self.cap.set(3, 1280)
|
||
self.cap.set(4, 720)
|
||
|
||
# 设置窗口初始大小
|
||
self.resize(1280, 720)
|
||
|
||
# 使用长距离模型和更低的人脸检测阈值以提高检测率
|
||
self.detector = FaceDetector(minDetectionCon=0.25, modelSelection=1)
|
||
|
||
# ---------------- 创建界面 ----------------
|
||
self.setup_ui()
|
||
|
||
self.faces = []
|
||
|
||
def setup_ui(self):
|
||
# 视频标签 - 填充整个窗口
|
||
self.video_label = QLabel(self)
|
||
self.video_label.setStyleSheet("background:#000;")
|
||
self.video_label.setGeometry(0, 0, self.width(), self.height())
|
||
|
||
# 按钮 - 叠加在视频上
|
||
self.btn = QPushButton("随机", self)
|
||
self.btn.setFixedSize(140, 55)
|
||
self.btn.setStyleSheet("""
|
||
QPushButton {
|
||
background-color: rgba(0, 0, 0, 120);
|
||
color: white;
|
||
font-size: 20px;
|
||
padding: 10px 20px;
|
||
border-radius: 8px;
|
||
}
|
||
QPushButton:hover {
|
||
background-color: rgba(30, 30, 30, 180);
|
||
}
|
||
""")
|
||
self.btn.clicked.connect(self.button_clicked)
|
||
|
||
# 初始按钮位置
|
||
self.update_button_position()
|
||
|
||
def resizeEvent(self, event):
|
||
"""窗口大小改变时自动调整视频和按钮位置"""
|
||
self.video_label.setGeometry(0, 0, self.width(), self.height())
|
||
self.update_button_position()
|
||
super().resizeEvent(event)
|
||
|
||
def update_button_position(self):
|
||
"""更新按钮位置"""
|
||
btn_w = 140
|
||
btn_h = 55
|
||
margin = 20
|
||
self.btn.setGeometry(
|
||
margin,
|
||
self.height() - btn_h - margin,
|
||
btn_w,
|
||
btn_h
|
||
)
|
||
|
||
# ---------------------------------------------------------
|
||
# 逻辑按钮：随机 ↔ 重置
|
||
# ---------------------------------------------------------
|
||
def button_clicked(self):
|
||
if self.state == "normal":
|
||
# 切换到随机状态
|
||
self.state = "random"
|
||
self.btn.setText("重置")
|
||
|
||
# 随机选择一个人脸
|
||
if len(self.faces) > 0:
|
||
self.selected_face_index = random.randint(0, len(self.faces) - 1)
|
||
|
||
# 保存当前帧作为静态画面
|
||
ret, img = self.cap.read()
|
||
if ret:
|
||
img = cv2.flip(img, 1)
|
||
# 调整图像大小以适应窗口
|
||
img = cv2.resize(img, (self.video_label.width(), self.video_label.height()))
|
||
self.static_frame = img.copy()
|
||
self.all_faces_snapshot = self.faces.copy()
|
||
else: # self.state == "random"
|
||
# 切换回初始状态
|
||
self.state = "normal"
|
||
self.selected_face_index = -1
|
||
self.static_frame = None
|
||
self.all_faces_snapshot = []
|
||
self.btn.setText("随机")
|
||
|
||
# ---------------------------------------------------------
|
||
# 刷视频帧
|
||
# ---------------------------------------------------------
|
||
def update_frame(self):
|
||
# ============================
|
||
# 状态：normal（检测所有人脸并显示绿框）
|
||
# ============================
|
||
if self.state == "normal":
|
||
ret, img = self.cap.read()
|
||
if not ret:
|
||
return
|
||
|
||
img = cv2.flip(img, 1)
|
||
|
||
# 调整图像大小以适应窗口
|
||
img = cv2.resize(img, (self.video_label.width(), self.video_label.height()))
|
||
|
||
# 检测人脸
|
||
img, faces = self.detector.findFaces(img, draw=False)
|
||
self.faces = faces if faces else []
|
||
|
||
# 绘制所有人脸为绿色
|
||
for face in self.faces:
|
||
x, y, w, h = face["bbox"]
|
||
score = face["score"][0]
|
||
cv2.rectangle(img, (x, y), (x + w, y + h), (0, 255, 0), 3)
|
||
cv2.putText(img, f"{score:.2f}", (x, y - 10),
|
||
cv2.FONT_HERSHEY_SIMPLEX, 0.6, (0, 255, 0), 2)
|
||
|
||
# 显示图像
|
||
self.display_image(img)
|
||
|
||
# ============================
|
||
# 状态：random（显示静态画面，选中的人脸为红色，其他为绿色）
|
||
# ============================
|
||
elif self.state == "random" and self.static_frame is not None:
|
||
img = self.static_frame.copy()
|
||
|
||
# 绘制所有人脸，选中的为红色，其他为绿色
|
||
for i, face in enumerate(self.all_faces_snapshot):
|
||
x, y, w, h = face["bbox"]
|
||
score = face["score"][0]
|
||
|
||
if i == self.selected_face_index:
|
||
color = (0, 0, 255) # 红色
|
||
else:
|
||
color = (0, 255, 0) # 绿色
|
||
|
||
cv2.rectangle(img, (x, y), (x + w, y + h), color, 3)
|
||
cv2.putText(img, f"{score:.2f}", (x, y - 10),
|
||
cv2.FONT_HERSHEY_SIMPLEX, 0.6, color, 2)
|
||
|
||
# 显示图像
|
||
self.display_image(img)
|
||
|
||
# ---------------------------------------------------------
|
||
# 显示图像到 QLabel
|
||
# ---------------------------------------------------------
|
||
def display_image(self, img):
|
||
rgb = cv2.cvtColor(img, cv2.COLOR_BGR2RGB)
|
||
h, w, ch = rgb.shape
|
||
bytes_per_line = ch * w
|
||
qimg = QImage(rgb.data, w, h, bytes_per_line, QImage.Format.Format_RGB888)
|
||
self.video_label.setPixmap(QPixmap.fromImage(qimg))
|
||
|
||
# ---------------------------------------------------------
|
||
def closeEvent(self, event):
|
||
if self.cap.isOpened():
|
||
self.cap.release()
|
||
event.accept()
|
||
|
||
|
||
def main():
|
||
app = QApplication(sys.argv)
|
||
window = FaceRandomApp()
|
||
window.show()
|
||
|
||
# 启动定时器
|
||
timer = QTimer()
|
||
timer.timeout.connect(window.update_frame)
|
||
timer.start(30)
|
||
|
||
sys.exit(app.exec())
|
||
|
||
|
||
if __name__ == "__main__":
|
||
main()
|