266 lines
9.1 KiB
Python
266 lines
9.1 KiB
Python
import sys
|
||
import random
|
||
import cv2
|
||
from cvzone.FaceDetectionModule import FaceDetector
|
||
from PySide6.QtCore import QTimer, Qt
|
||
from PySide6.QtGui import QImage, QPixmap
|
||
from PySide6.QtWidgets import QApplication, QWidget, QLabel, QPushButton, QVBoxLayout, QProgressBar
|
||
|
||
|
||
class LoadingScreen(QWidget):
|
||
def __init__(self):
|
||
super().__init__()
|
||
self.setWindowTitle("加载中...")
|
||
self.setFixedSize(300, 150)
|
||
self.setWindowFlags(Qt.FramelessWindowHint | Qt.WindowStaysOnTopHint)
|
||
self.setStyleSheet("background-color: #1E90FF;") # 蓝色背景
|
||
|
||
# 创建布局
|
||
layout = QVBoxLayout()
|
||
layout.setAlignment(Qt.AlignCenter)
|
||
layout.setSpacing(15)
|
||
|
||
# 加载提示文字
|
||
self.loading_label = QLabel("人脸模型加载中")
|
||
self.loading_label.setAlignment(Qt.AlignCenter)
|
||
self.loading_label.setStyleSheet("color: white; font-size: 18px; font-weight: bold;")
|
||
layout.addWidget(self.loading_label)
|
||
|
||
# 进度条 - 设置为不确定模式，显示加载动画
|
||
self.progress_bar = QProgressBar()
|
||
self.progress_bar.setRange(0, 0) # 设置为0,0表示不确定模式
|
||
self.progress_bar.setFixedHeight(10)
|
||
self.progress_bar.setStyleSheet("""
|
||
QProgressBar {
|
||
border: 1px solid rgba(255, 255, 255, 100);
|
||
border-radius: 5px;
|
||
background-color: rgba(255, 255, 255, 50);
|
||
}
|
||
QProgressBar::chunk {
|
||
background-color: white;
|
||
border-radius: 4px;
|
||
}
|
||
""")
|
||
layout.addWidget(self.progress_bar)
|
||
|
||
self.setLayout(layout)
|
||
|
||
|
||
class FaceRandomApp(QWidget):
|
||
def __init__(self):
|
||
super().__init__()
|
||
|
||
# 先显示加载页面
|
||
self.loading_screen = LoadingScreen()
|
||
self.loading_screen.show()
|
||
|
||
# 延迟初始化主界面，让加载页面先显示
|
||
QTimer.singleShot(100, self.initialize_app)
|
||
|
||
def initialize_app(self):
|
||
# 设置窗口标题
|
||
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)
|
||
|
||
# 使用长距离模型和更低的人脸检测阈值以提高检测率
|
||
# 这里会加载模型，可能需要一些时间
|
||
try:
|
||
self.detector = FaceDetector(minDetectionCon=0.25, modelSelection=1)
|
||
except Exception as e:
|
||
print(f"初始化人脸检测器时出错: {e}")
|
||
# 如果初始化失败，使用一个空的检测器
|
||
self.detector = None
|
||
|
||
# ---------------- 创建界面 ----------------
|
||
self.setup_ui()
|
||
|
||
self.faces = []
|
||
|
||
# 模拟加载完成
|
||
QTimer.singleShot(2000, self.finish_loading)
|
||
|
||
def finish_loading(self):
|
||
# 关闭加载页面
|
||
self.loading_screen.close()
|
||
# 显示主窗口
|
||
self.show()
|
||
|
||
# 启动定时器更新视频帧
|
||
self.timer = QTimer()
|
||
self.timer.timeout.connect(self.update_frame)
|
||
self.timer.start(30)
|
||
|
||
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()))
|
||
|
||
# 检测人脸
|
||
if self.detector is not None:
|
||
img, faces = self.detector.findFaces(img, draw=False)
|
||
self.faces = faces if faces else []
|
||
else:
|
||
self.faces = []
|
||
|
||
# 绘制所有人脸为绿色
|
||
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()
|
||
|
||
sys.exit(app.exec())
|
||
|
||
|
||
if __name__ == "__main__":
|
||
main() |