devbox
我家自动化
这个屌丝很懒，什么也没留下！
关注作者
热门标签
热门文章
当前位置:   article > 正文

yolov8+tensorrt+pyqt5+多线程+鼠标单机放大缩小多路视频监控系统+账户密码验证登录界面（持续开发完善中）_tensort+pyqt5

作者：我家自动化 | 2024-02-27 07:23:56

赞

踩

tensort+pyqt5

一.最近公司做多路检测项目，b站找了好多类似功能实现，源码都收费，好贵，无奈，自己搞

    1.废话不多说，先看demo吧

                

                  

                

 

        我这里算法用的是官方的yolov8s哈，各位大佬可以结合自己的算法转成engine再部署，另外我这里用的转化工具是参考github：

             https://github.com/triple-Mu/YOLOv8-TensorRT 

        另外我的多线程中的推理部分也用的这个大神中间的部分代码，自己做了一些修改，特此鸣谢。

        另注：我这个还没写完哈，我只有一路是摄像头，其他都是视频文件模拟的，我的是显卡是3080，i5的12代cpu，后面还要加的功能很多，因为落地摄像头是rtsp流，可能还要加cuda硬编解码做优化。基本功能已完成，后面会持续优化而且代码也会有优化，现在只相当于草稿了，需要的留意更新吧。。。。。

v1.1.1

pyqt5 账户密码登录界面

  1. class loginUi(QMainWindow,login_ui):
  2.  
  3.     def __init__(self):
  4.  
  5.         QMainWindow.__init__(self)
  6.         self.setupUi(self)
  7.         self.pushButton.clicked.connect(self.checkLogin)
  8.         self.pushButton_2.clicked.connect(self.exitSys)
  9.  
  10.     def exitSys(self):
  11.         sys.exit()
  12.  
  13.     def checkLogin(self):
  14.  
  15.         self.name = self.lineEdit_2.text()
  16.         self.passwd = self.lineEdit_3.text()
  17.  
  18.         self.getSql()
  19.         if self.name == self.nameSave and self.passwdSave == self.passwd:
  20.             self.myPlayer = mainUi()
  21.             self.myPlayer.show()
  22.             self.close()
  23.         else:
  24.             self.errorUi = errorLogin()
  25.             self.errorUi.show()
  26.             self.close()
  27.  
  28.     def getSql(self):
  29.         with open("lo.txt",'r') as f:
  30.             lines = f.readlines()
  31.             self.nameSave = lines[0][:-1]
  32.             self.passwdSave = lines[1]
  33.             f.close()
  34.  
  35. class errorLogin(QMainWindow,error_ui):
  36.  
  37.     def __init__(self):
  38.  
  39.         QMainWindow.__init__(self)
  40.         self.setupUi(self)
  41.         self.pushButton.clicked.connect(self.returnUi)
  42.  
  43.     def returnUi(self):
  44.         self.log = loginUi()
  45.         self.log.show()
  46.         self.close()

v1.0.0

  1. class mainUi(QMainWindow,convert_ui):
  2.  
  3.     def __init__(self):
  4.  
  5.         QMainWindow.__init__(self)
  6.         self.setupUi(self)
  7.         self.pushButton.clicked.connect(self.openVideo)
  8.         self.pushButton_2.clicked.connect(lambda :self.changeLabel(1))
  9.         self.pushButton_3.clicked.connect(lambda: self.changeLabel(2))
  10.         self.pushButton_4.clicked.connect(lambda: self.changeLabel(3))
  11.         self.pushButton_5.clicked.connect(lambda: self.changeLabel(4))
  12.         self.pushButton_6.clicked.connect(lambda: self.changeLabel(5))
  13.         self.pushButton_7.clicked.connect(lambda: self.changeLabel(6))
  14.         self.pushButton_8.clicked.connect(lambda: self.changeLabel(7))
  15.         self.pushButton_9.clicked.connect(lambda: self.changeLabel(8))
  16.         self.pushButton_10.clicked.connect(self.exitSys)
  17.  
  18.         self.countButton1 = 0
  19.         self.countButton2 = 0
  20.         self.countButton3 = 0
  21.         self.countButton4 = 0
  22.         self.countButton5 = 0
  23.         self.countButton6 = 0
  24.         self.countButton7 = 0
  25.         self.countButton8 = 0
  26.  
  27.     def exitSys(self):
  28.  
  29.         sys.exit()
  30.  
  31.     def getArr(self,arr):
  32.         img = QImage(arr, arr.shape[1], arr.shape[0],QImage.Format_BGR888)
  33.         self.label.setPixmap(QPixmap(img))
  34.         self.label.setScaledContents(True)
  35.  
  36.     def getArr2(self,arr):
  37.         img = QImage(arr, arr.shape[1], arr.shape[0], QImage.Format_BGR888)
  38.         self.label_2.setPixmap(QPixmap(img))
  39.         self.label_2.setScaledContents(True)
  40.  
  41.     def getArr3(self,arr):
  42.         img = QImage(arr, arr.shape[1], arr.shape[0], QImage.Format_BGR888)
  43.         self.label_3.setPixmap(QPixmap(img))
  44.         self.label_3.setScaledContents(True)
  45.  
  46.     def getArr4(self,arr):
  47.         img = QImage(arr, arr.shape[1], arr.shape[0], QImage.Format_BGR888)
  48.         self.label_4.setPixmap(QPixmap(img))
  49.         self.label_4.setScaledContents(True)
  50.  
  51.     def getArr5(self,arr):
  52.         img = QImage(arr, arr.shape[1], arr.shape[0], QImage.Format_BGR888)
  53.         self.label_5.setPixmap(QPixmap(img))
  54.         self.label_5.setScaledContents(True)
  55.  
  56.     def getArr6(self,arr):
  57.         img = QImage(arr, arr.shape[1], arr.shape[0], QImage.Format_BGR888)
  58.         self.label_6.setPixmap(QPixmap(img))
  59.         self.label_6.setScaledContents(True)
  60.  
  61.     def getArr7(self,arr):
  62.         img = QImage(arr, arr.shape[1], arr.shape[0], QImage.Format_BGR888)
  63.         self.label_7.setPixmap(QPixmap(img))
  64.         self.label_7.setScaledContents(True)
  65.  
  66.     def getArr8(self,arr):
  67.         img = QImage(arr, arr.shape[1], arr.shape[0], QImage.Format_BGR888)
  68.         self.label_8.setPixmap(QPixmap(img))
  69.         self.label_8.setScaledContents(True)
  70.  
  71.     def openVideo(self):
  72.  
  73.         self.pushButton.setEnabled(False)
  74.         self.pushButton.setText("请等待")
  75.  
  76.  
  77.         self.imgGet = inferVideo1()
  78.         self.imgGet.updateImg.connect(self.getArr)
  79.         self.imgGet.start()
  80.  
  81.         self.imgGet2 = inferVideo2()
  82.         self.imgGet2.updateImg1.connect(self.getArr2)
  83.         self.imgGet2.start()
  84.  
  85.         self.imgGet3 = inferVideo3()
  86.         self.imgGet3.updateImg2.connect(self.getArr3)
  87.         self.imgGet3.start()
  88.  
  89.         self.imgGet4 = inferVideo4()
  90.         self.imgGet4.updateImg3.connect(self.getArr4)
  91.         self.imgGet4.start()
  92.  
  93.         self.imgGet5 = inferVideo5()
  94.         self.imgGet5.updateImg4.connect(self.getArr5)
  95.         self.imgGet5.start()
  96.  
  97.         self.imgGet6 = inferVideo6()
  98.         self.imgGet6.updateImg5.connect(self.getArr6)
  99.         self.imgGet6.start()
  100.  
  101.         self.imgGet7 = inferVideo7()
  102.         self.imgGet7.updateImg6.connect(self.getArr7)
  103.         self.imgGet7.start()
  104.  
  105.         self.imgGet8 = inferVideo8()
  106.         self.imgGet8.updateImg7.connect(self.getArr8)
  107.         self.imgGet8.start()
  108.  
  109.         self.pushButton.setStyleSheet("border-color : yellow")
  110.         self.pushButton.setText("已播放")
  111.         def returnPos(self):
  112.  
  113.         self.label.resize(501, 221)
  114.         self.label.raise_()
  115.  
  116.         self.label_2.move(501, 0)
  117.         self.label_2.resize(501, 221)
  118.         self.label_2.raise_()
  119.  
  120.         self.label_3.move(0, 221)
  121.         self.label_3.resize(501, 221)
  122.         self.label_3.raise_()
  123.  
  124.         self.label_4.move(501, 221)
  125.         self.label_4.resize(501, 221)
  126.         self.label_4.raise_()
  127.  
  128.         self.label_5.move(0, 442)
  129.         self.label_5.resize(501, 221)
  130.         self.label_5.raise_()
  131.  
  132.         self.label_6.move(501, 442)
  133.         self.label_6.resize(501, 221)
  134.         self.label_6.raise_()
  135.  
  136.         self.label_7.move(0, 663)
  137.         self.label_7.resize(501, 221)
  138.         self.label_7.raise_()
  139.  
  140.         self.label_8.move(501, 663)
  141.         self.label_8.resize(501, 221)
  142.         self.label_8.raise_()
  143.  
  144.     def changeLabel(self,x):
  145.  
  146.         if x == 1:
  147.  
  148.             self.countButton1 += 1
  149.  
  150.             if self.countButton1 % 2 == 0:
  151.  
  152.                 self.returnPos()
  153.  
  154.             else:
  155.  
  156.                 self.label.resize(self.label.width() + 501, self.label.height() + 663)
  157.                 self.label.raise_()
  158.  
  159.         elif x == 2:
  160.  
  161.             self.countButton2 += 1
  162.  
  163.             if self.countButton2 % 2 == 0:
  164.  
  165.                 self.returnPos()
  166.  
  167.             else:
  168.  
  169.                 self.label_2.move(0, 0)
  170.                 self.label_2.resize(self.label_2.width() + 501, self.label_2.height() + 663)
  171.                 self.label_2.raise_()
  172.  
  173.         elif x == 3:
  174.  
  175.             self.countButton3 += 1
  176.  
  177.             if self.countButton3 % 2 == 0:
  178.  
  179.                 self.returnPos()
  180.  
  181.             else:
  182.  
  183.                 self.label_3.move(0, 0)
  184.                 self.label_3.resize(self.label_3.width() + 501, self.label_3.height() + 663)
  185.                 self.label_3.raise_()
  186.  
  187.         elif x == 4:
  188.  
  189.             self.countButton4 += 1
  190.  
  191.             if self.countButton4 % 2 == 0:
  192.  
  193.                 self.returnPos()
  194.  
  195.             else:
  196.  
  197.                 self.label_4.move(0, 0)
  198.                 self.label_4.resize(self.label_4.width() + 501, self.label_4.height() + 663)
  199.                 self.label_4.raise_()
  200.  
  201.         elif x == 5:
  202.  
  203.             self.countButton5 += 1
  204.  
  205.             if self.countButton5 % 2 == 0:
  206.  
  207.                 self.returnPos()
  208.  
  209.             else:
  210.  
  211.                 self.label_5.move(0, 0)
  212.                 self.label_5.resize(self.label_5.width() + 501, self.label_5.height() + 663)
  213.                 self.label_5.raise_()
  214.  
  215.         elif x == 6:
  216.  
  217.             self.countButton6 += 1
  218.  
  219.             if self.countButton6 % 2 == 0:
  220.  
  221.                 self.returnPos()
  222.  
  223.             else:
  224.  
  225.                 self.label_6.move(0, 0)
  226.                 self.label_6.resize(self.label_6.width() + 501, self.label_6.height() + 663)
  227.                 self.label_6.raise_()
  228.  
  229.         elif x == 7:
  230.  
  231.             self.countButton7 += 1
  232.  
  233.             if self.countButton7 % 2 == 0:
  234.  
  235.                 self.returnPos()
  236.  
  237.             else:
  238.  
  239.                 self.label_7.move(0, 0)
  240.                 self.label_7.resize(self.label_7.width() + 501, self.label_7.height() + 663)
  241.                 self.label_7.raise_()
  242.  
  243.         elif x == 8:
  244.  
  245.             self.countButton8 += 1
  246.  
  247.             if self.countButton8 % 2 == 0:
  248.  
  249.                 self.returnPos()
  250.  
  251.             else:
  252.  
  253.                 self.label_8.move(0, 0)
  254.                 self.label_8.resize(self.label_8.width() + 501, self.label_8.height() + 663)
  255.                 self.label_8.raise_()

   

  1. def det_postprocess(data: Tuple[Tensor, Tensor, Tensor, Tensor]):
  2.  
  3.     assert len(data) == 4
  4.     num_dets, bboxes, scores, labels = data[0][0], data[1][0], data[2][
  5.         0], data[3][0]
  6.     nums = num_dets.item()
  7.     if nums == 0:
  8.         return bboxes.new_zeros((0, 4)), scores.new_zeros((0, )), labels.new_zeros((0, ))
  9.     bboxes = bboxes[:nums]
  10.     scores = scores[:nums]
  11.     labels = labels[:nums]
  12.  
  13.     return bboxes, scores, labels
  1. def capRun(Img,W,H,device,Engine):
  2.     draw = Img.copy()
  3.     bgr, ratio, dwdh = letterbox(Img, (W, H))
  4.     tensor = blob(bgr, return_seg=False)
  5.     dwdh = torch.asarray(dwdh * 2, dtype=torch.float32, device=device)
  6.     tensor = torch.asarray(tensor, device=device)
  7.     data = Engine(tensor)
  8.     bboxes, scores, labels = det_postprocess(data)
  9.     if bboxes.numel() == 0:
  10.         # if no bounding box
  11.         return draw
  12.     bboxes -= dwdh
  13.     bboxes /= ratio
  14.     for (bbox, score, label) in zip(bboxes, scores, labels):
  15.         bbox = bbox.round().int().tolist()
  16.         cls_id = int(label)
  17.         cls = CLASSES[cls_id]
  18.         color = COLORS[cls]
  19.         cv2.rectangle(draw, bbox[:2], bbox[2:], color, 2)
  20.         cv2.putText(draw, f'{cls}:{score:.3f}', (bbox[0], bbox[1] - 2), cv2.FONT_HERSHEY_SIMPLEX,
  21.                     0.75, [225, 255, 255], thickness=2)
  22.     return draw
  1. class inferVideo1(QThread):
  2.  
  3.     updateImg = pyqtSignal(np.ndarray)
  4.  
  5.     def __init__(self):
  6.  
  7.         super(inferVideo1,self).__init__()
  8.         self.device = torch.device("cuda:0")
  9.         self.Engine = TRTModule("yolov8s.engine", self.device)
  10.         self.H, self.W = self.Engine.inp_info[0].shape[-2:]
  11.         self.Engine.set_desired(['num_dets', 'bboxes', 'scores', 'labels'])
  12.  
  13.     def run(self):
  14.  
  15.         cap = cv2.VideoCapture(0)
  16.  
  17.         if not cap.isOpened():
  18.  
  19.             print("Unable to open camera")
  20.  
  21.         else:
  22.  
  23.             while True:
  24.  
  25.                 ret, Img = cap.read()
  26.  
  27.                 if ret:
  28.  
  29.                     draw = capRun(Img,self.W,self.H,self.device,self.Engine)
  30.                     self.updateImg.emit(draw)
  31.  
  32.                 else:
  33.  
  34.                     break
  35.  
  36.         cap.release()

上面是一个线程的代码哈，需要的自己复制粘贴8个就行。我的i5的12代 16g内存 完美运行。。。

需要源码的可以私信我哈。

声明：本文内容由网友自发贡献，不代表【wpsshop博客】立场，版权归原作者所有，本站不承担相应法律责任。如您发现有侵权的内容，请联系我们。转载请注明出处：https://www.wpsshop.cn/w/我家自动化/article/detail/151042?site
推荐阅读
相关标签

Copyright © 2003-2013 www.wpsshop.cn 版权所有，并保留所有权利。

  

闽ICP备14008679号