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(400, 250) # 增大窗口大小以容纳错误信息 self.setWindowFlags(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.error_label = QLabel("") self.error_label.setAlignment(Qt.AlignCenter) self.error_label.setStyleSheet("color: #FFD700; font-size: 12px; font-weight: normal;") self.error_label.setWordWrap(True) self.error_label.setHidden(True) # 初始隐藏 layout.addWidget(self.error_label) # 进度条 - 设置为不确定模式，显示加载动画 self.progress_bar = QProgressBar() self.progress_bar.setRange(0, 0) # 设置为0,0表示不确定模式 self.progress_bar.setFixedHeight(10) layout.addWidget(self.progress_bar) # 关闭按钮 - 初始隐藏 self.close_button = QPushButton("关闭") self.close_button.setFixedSize(100, 35) self.close_button.setStyleSheet(""" QPushButton { background-color: #FF6347; color: white; font-size: 14px; border-radius: 5px; } QPushButton:hover { background-color: #FF4500; } """) self.close_button.clicked.connect(self.close) self.close_button.setHidden(True) # 初始隐藏 layout.addWidget(self.close_button) self.setLayout(layout) def show_error(self, error_message): """显示错误信息并显示关闭按钮，隐藏进度条""" self.loading_label.setText("人脸模型加载失败") self.error_label.setText(f"错误详情: {error_message}") self.error_label.setHidden(False) self.progress_bar.setHidden(True) self.close_button.setHidden(False) self.setWindowTitle("加载失败") # 调整窗口大小以适应错误信息 self.adjustSize() QApplication.processEvents() class FaceRandomApp(QWidget): def __init__(self): super().__init__() # 先显示加载页面 self.loading_screen = LoadingScreen() self.loading_screen.show() # 强制立即处理GUI事件，确保加载页面能立即显示 QApplication.processEvents() # 延迟初始化主界面，让加载页面先显示 QTimer.singleShot(100, self.initialize_app) def initialize_app(self): # 状态：normal → random self.state = "normal" self.selected_face_index = -1 self.static_frame = None # 保存随机状态下的静态画面 self.all_faces_snapshot = [] # 保存所有人脸信息快照 # 更新加载页面文字，提示正在进行人脸模型初始化 self.loading_screen.loading_label.setText("正在初始化人脸检测模型...") QApplication.processEvents() # 使用长距离模型和更低的人脸检测阈值以提高检测率 # 这里会加载模型，可能需要一些时间 try: self.detector = FaceDetector(minDetectionCon=0.25, modelSelection=1) self.loading_screen.loading_label.setText("人脸模型加载完成!") # 只有在检测器初始化成功后才继续初始化 # 设置窗口标题 self.setWindowTitle("Face Random Selector") # ---------------- 摄像头 ---------------- self.cap = cv2.VideoCapture(0) # 设置摄像头分辨率 self.cap.set(3, 1280) self.cap.set(4, 720) # 设置窗口初始大小 self.resize(1280, 720) # ---------------- 创建界面 ---------------- self.setup_ui() self.faces = [] # 模拟加载完成 QTimer.singleShot(1500, self.finish_loading) except Exception as e: # 捕获并显示详细的错误信息（仅显示真实的错误内容） error_type = type(e).__name__ error_message = str(e) # 只显示真实的错误类型和错误消息 detailed_error = f"{error_type}: {error_message}" print(f"初始化人脸检测器时出错: {detailed_error}") # 显示错误信息并停止继续初始化 self.loading_screen.show_error(detailed_error) # 不再继续初始化主界面 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()