diff --git a/src/Guide_stick_system/image_recognition/window.py b/src/Guide_stick_system/image_recognition/window.py index 1567185..4370487 100644 --- a/src/Guide_stick_system/image_recognition/window.py +++ b/src/Guide_stick_system/image_recognition/window.py @@ -59,18 +59,18 @@ class MainWindow(QTabWidget): ***模型初始化*** ''' @torch.no_grad() - def model_load(self, weights="", # model.pt path(s) - device='', # cuda device, i.e. 0 or 0,1,2,3 or cpu - half=False, # use FP16 half-precision inference - dnn=False, # use OpenCV DNN for ONNX inference + def model_load(self, weights="", # 模型文件路径 + device='', # 指定CUDA设备,例如 '0' 或 '0,1,2,3' 或 'cpu' + half=False, # 是否使用半精度浮点数(FP16)进行推理 + dnn=False, # 是否使用OpenCV DNN进行ONNX模型的推理 ): device = select_device(device) - half &= device.type != 'cpu' # half precision only supported on CUDA + half &= device.type != 'cpu' # 只有CUDA支持半精度 device = select_device(device) model = DetectMultiBackend(weights, device=device, dnn=dnn) stride, names, pt, jit, onnx = model.stride, model.names, model.pt, model.jit, model.onnx - # Half - half &= pt and device.type != 'cpu' # half precision only supported by PyTorch on CUDA + # 半精度处理 + half &= pt and device.type != 'cpu' # 确保PyTorch在CUDA上支持半精度 if pt: model.model.half() if half else model.model.float() print("模型加载完成!") @@ -106,13 +106,13 @@ class MainWindow(QTabWidget): det_img_button.clicked.connect(self.detect_img) up_img_button.setFont(font_main) det_img_button.setFont(font_main) - up_img_button.setStyleSheet("QPushButton{color:white}" - "QPushButton:hover{background-color: rgb(2,110,180);}" - "QPushButton{background-color:rgb(48,124,208)}" - "QPushButton{border:2px}" - "QPushButton{border-radius:5px}" - "QPushButton{padding:5px 5px}" - "QPushButton{margin:5px 5px}") + up_img_button.setStyleSheet("QPushButton{color:white}"#颜色 + "QPushButton:hover{background-color: rgb(2,110,180);}"#鼠标悬停时的样式,鼠标在按钮上悬停时,背景变为蓝色 + "QPushButton{background-color:rgb(48,124,208)}"#设置按钮的默认背景色为另一种蓝色。 + "QPushButton{border:2px}"#宽2像素 + "QPushButton{border-radius:5px}"#圆角5像素 + "QPushButton{padding:5px 5px}"#四周边距5像素 + "QPushButton{margin:5px 5px}")#外边距5像素 det_img_button.setStyleSheet("QPushButton{color:white}" "QPushButton:hover{background-color: rgb(2,110,180);}" "QPushButton{background-color:rgb(48,124,208)}" @@ -177,16 +177,16 @@ class MainWindow(QTabWidget): # todo 关于界面 about_widget = QWidget() about_layout = QVBoxLayout() - about_title = QLabel('欢迎使用目标检测系统\n\n 提供付费指导:有需要的好兄弟加下面的QQ即可') # todo 修改欢迎词语 + about_title = QLabel('欢迎使用') # todo 修改欢迎词语 about_title.setFont(QFont('楷体', 18)) about_title.setAlignment(Qt.AlignCenter) about_img = QLabel() about_img.setPixmap(QPixmap('images/UI/qq.png')) about_img.setAlignment(Qt.AlignCenter) - # label4.setText("如何调整学习率") - label_super = QLabel() # todo 更换作者信息 - label_super.setText("或者你可以在这里找到我-->肆十二") + # + label_super = QLabel() + label_super.setFont(QFont('楷体', 16)) label_super.setOpenExternalLinks(True) # label_super.setOpenExternalLinks(True) @@ -201,10 +201,7 @@ class MainWindow(QTabWidget): self.left_img.setAlignment(Qt.AlignCenter) self.addTab(img_detection_widget, '图片检测') self.addTab(vid_detection_widget, '视频检测') - self.addTab(about_widget, '联系我') - self.setTabIcon(0, QIcon('images/UI/lufei.png')) - self.setTabIcon(1, QIcon('images/UI/lufei.png')) - self.setTabIcon(2, QIcon('images/UI/lufei.png')) + ''' ***上传图片*** @@ -377,7 +374,7 @@ class MainWindow(QTabWidget): ### 视频开启事件 ### ''' - # 视频和摄像头的主函数是一样的,不过是传入的source不同罢了 + # 视频和摄像头的主函数是一样的,传入的source不同 def detect_vid(self): # pass model = self.model