yolov5目标检测gpu加速

该项目使用yolov5进行目标检测，并且可以选择使用手机或者电脑摄像头，以及是否使用GPU加速推理。
博客最后附有代码下载链接。

界面相关：

视频效果：

yolov5 gpu目标检测

b站地址

yolov5主要是参考这位巨佬的代码，如有需要可前往自行下载（无界面版）。

下面分享一下主要的界面代码。

import sys, cv2
from PyQt5.QtGui import *
from PyQt5.QtWidgets import *
from PyQt5.QtCore import *
from PyQt5 import QtGui,QtWidgets
import sys, cv2, time
from untitled import Ui_Form
import cv2 as cv
import numpy as np

import time

import cv2
import numpy as np
from PIL import Image
from yolo import YOLO

class Mywindow(QtWidgets.QWidget,Ui_Form): #这个窗口继承了用QtDesignner 绘制的窗口

    def __init__(self):
        super(Mywindow,self).__init__()
        self.setupUi(self)
        self.use_palette()
        # 设置定时器
        self._timer = QTimer(self)
        self._timer.timeout.connect(self.play)



        #单选按钮分组
        self.bg1=QButtonGroup(self)
        self.bg1.addButton(self.radioButton_3,1)
        self.bg1.addButton(self.radioButton_4,2)
        self.radioButton_3.setChecked(True)

        self.bg2 = QButtonGroup(self)
        self.bg2.addButton(self.radioButton_5, 3)
        self.bg2.addButton(self.radioButton_6, 4)
        self.radioButton_5.setChecked(True)

        self.yolo =