pyqt5+AI健身+CPU实时检测mediapipe 可视化界面opencv实现人体关键点检测，AI健身计数等功能

import cv2 import numpy as np import time import PoseModule as pm from PyQt5 import QtWidgets from csdn import Ui_Form from PyQt5.QtWidgets import QFileDialog import cv2 import time from PyQt5.QtGui import QImage, QPixmap from PyQt5.QtWidgets import QMainWindow, QApplication, QGraphicsScene, QGraphicsPixmapItem from PyQt5 import QtCore, QtGui, QtWidgets from PyQt5.QtCore import * detector = pm.poseDetector() class mywindow(QtWidgets.QWidget, Ui_Form): def __init__(self): super(mywindow, self).__init__() self.setupUi(self) self.pushButton.clicked.connect(self.read_file) self.pushButton_2.clicked.connect(self.read_voc) self.pushButton_3.clicked.connect(self.stop_voc) self.cap = [] self.timer_camera = QtCore.QTimer() self.count= 0 self.dir = 0 self.pTime = 0 def read_file(self): # 选取文件 filename, filetype = QFileDialog.getOpenFileName(self, "选取文件", "./", "Images (*.png *.jpeg *.jpg)") image = cv2.imread(filename) image = image[:, :, ::-1].copy() image = detector.findPose(image, False) lmList = detector.findPosition(image, False) if len(lmList) != 0: # Right Arm angle = detector.findAngle(image, 12, 14, 16) height = image.shape[0] width = image.shape[1] frame = QImage(image, width, height, width * 3, QImage.Format_RGB888) pix = QPixmap.fromImage(frame) self.item = QGraphicsPixmapItem(pix) self.scene = QGraphicsScene() # 创建场景 self.scene.addItem(self.item) self.graphicsView.setScene(self.scene) def read_voc(self): # 选取文件 videoName, _ = QFileDialog.getOpenFileName(self, "Open", "", "*.mp4;;*.avi;;All Files(*)") self.cap = cv2.VideoCapture(videoName) self.timer_camera.start(100) self.timer_camera.timeout.connect(self.voc) def voc(self): if (self.cap.isOpened()): ret, self.frame = self.cap.read() if ret: frame = cv2.cvtColor(self.frame, cv2.COLOR_BGR2RGB) frame = cv2.resize(frame, (960, 576)) # frame = cv2.imread("AiTrainer/test.jpg") frame = detector.findPose(frame, False) lmList = detector.findPosition(frame, False) # print(lmList) if len(lmList) != 0: # Right Arm angle = detector.findAngle(frame, 12, 14, 16) # # Left Arm # angle = detector.findAngle(frame, 11, 13, 15) # per = np.interp(angle, (210, 310), (0, 100)) per = np.interp(angle, (0, 180), (0, 100)) bar = np.interp(angle, (0, 180), (100, 550)) # bar = np.interp(angle, (210, 310), (650, 100)) # Check for the dumbbell curls color = (255, 0, 255) if per >= 80: color = (0, 255, 0) if self.dir == 0: self.count += 0.5 self.dir = 1 if per <= 40: color = (0, 255, 0) if self.dir == 1: self.count += 0.5 self.dir = 0 print(per, self.count) # Draw Bar cv2.rectangle(frame, (800, 100), (875, 550), color, 3) cv2.rectangle(frame, (800, int(bar)), (875, 550), color, cv2.FILLED) cv2.putText(frame, f'{int(100 - per)} %', (800, 75), cv2.FONT_HERSHEY_PLAIN, 4, color, 4) # Draw Curl Count cv2.rectangle(frame, (0, 376), (250, 576), (0, 255, 0), cv2.FILLED) cv2.putText(frame, str(int(self.count)), (45, 550), cv2.FONT_HERSHEY_PLAIN, 15, (255, 0, 0), 25) cTime = time.time() fps = 1 / (cTime - self.pTime) self.pTime = cTime cv2.rectangle(frame, (0, 0), (200, 150), (0, 255, 0), cv2.FILLED) cv2.putText(frame, str(int(fps)), (30, 100), cv2.FONT_HERSHEY_PLAIN, 8, (255, 0, 0), 15) height, width, bytesPerComponent = frame.shape bytesPerLine = bytesPerComponent * width q_image = QImage(frame.data, width, height, bytesPerLine, QImage.Format_RGB888) pix = QPixmap.fromImage(q_image) self.item = QGraphicsPixmapItem(pix) self.scene = QGraphicsScene() # 创建场景 self.scene.addItem(self.item) self.graphicsView.setScene(self.scene) else: self.cap.release() self.timer_camera.stop() # 停止计时器 def stop_voc(self): if self.cap != []: self.cap.release() self.count= 0 self.dir = 0 self.pTime = 0 self.timer_camera.stop() # 停止计时器 ################################################################# if __name__ == "__main__": import sys app = QtWidgets.QApplication(sys.argv) ui = mywindow() ui.show() sys.exit(app.exec_())