请写出完整的python代码,实现利用ylov5对包含任意数量香蕉的照片进行计数,并且利用pyqt设计一个界面,实现从文件夹选择图像,显示所选择的图像,点击按钮开始计数,输出计数结果和结果图像四个功能

时间: 2024-04-11 20:30:06 浏览: 10
import sys import os from PyQt5.QtWidgets import QApplication, QMainWindow, QLabel, QVBoxLayout, QFileDialog, QPushButton, QMessageBox from PyQt5.QtGui import QPixmap import cv2 import torch import numpy as np from yolov5 import detect class BananaCounter(QMainWindow): def __init__(self): super().__init__() self.setWindowTitle("Banana Counter") self.label_image = QLabel(self) self.label_image.setAlignment(Qt.AlignCenter) self.btn_select_image = QPushButton("Select Image", self) self.btn_select_image.clicked.connect(self.select_image) self.btn_count_bananas = QPushButton("Count Bananas", self) self.btn_count_bananas.clicked.connect(self.count_bananas) layout = QVBoxLayout() layout.addWidget(self.label_image) layout.addWidget(self.btn_select_image) layout.addWidget(self.btn_count_bananas) widget = QWidget(self) widget.setLayout(layout) self.setCentralWidget(widget) def select_image(self): file_dialog = QFileDialog() file_dialog.setNameFilter("Image files (*.jpg *.png)") if file_dialog.exec_(): file_path = file_dialog.selectedFiles()[0] self.show_image(file_path) def show_image(self, file_path): pixmap = QPixmap(file_path) self.label_image.setPixmap(pixmap.scaledToWidth(self.label_image.width())) def count_bananas(self): file_path = self.label_image.pixmap().toImage().save("temp_image.jpg") result = detect("temp_image.jpg") os.remove("temp_image.jpg") if result is not None: num_bananas = len(result) message = f"Number of bananas: {num_bananas}" QMessageBox.information(self, "Banana Count Result", message) # Display the result image with bounding boxes image_with_boxes = draw_boxes(result, cv2.imread(file_path)) cv2.imshow("Banana Count Result", image_with_boxes) cv2.waitKey(0) else: QMessageBox.warning(self, "Banana Count Result", "No bananas found in the image.") def draw_boxes(result, image): for box in result: x1, y1, x2, y2, _ = box cv2.rectangle(image, (x1, y1), (x2, y2), (0, 255, 0), 2) return image def detect(image_path): weights = 'path_to_yolov5_weights' # Replace with your yolov5 weights path device = torch.device("cuda" if torch.cuda.is_available() else "cpu") model = torch.hub.load('ultralytics/yolov5', 'custom', path=weights).to(device).eval() img = cv2.imread(image_path) img = cv2.cvtColor(img, cv2.COLOR_BGR2RGB) results = model(img) if results.pandas().xyxy[0] is not None: return results.pandas().xyxy[0].values.tolist() else: return None if __name__ == "__main__": app = QApplication(sys.argv) window = BananaCounter() window.show() sys.exit(app.exec_())

相关推荐

最新推荐

recommend-type

Python PyQt5模块实现窗口GUI界面代码实例

主要介绍了Python PyQt5模块实现窗口GUI界面代码实例,文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需要的朋友可以参考下
recommend-type

用python+pyqt5手工编写一个含交互界面的简易的词法分析器

在本文我将用pyqt5写一个简单的交互界面,也修改了部分代码使得程序更加完整。 具体实验分析 本文全部源码见本文末尾,上次编写的不含交互界面的源码也可点此处下载 交互界面可以用QtDesigner快速编写,或者自己写,...
recommend-type

python GUI库图形界面开发之PyQt5简单绘图板实例与代码分析

主要介绍了python GUI库图形界面开发之PyQt5简单绘图板实例与代码分析,需要的朋友可以参考下
recommend-type

python GUI库图形界面开发之PyQt5窗口布局控件QStackedWidget详细使用方法

主要介绍了python GUI库图形界面开发之PyQt5窗口布局控件QStackedWidget详细使用方法,需要的朋友可以参考下
recommend-type

完美解决PyQt5登录界面跳转主界面方法

然后新建一个Python文件,在文件中分别将这两个界面UI引入,分别创建两个类,一个是实现登录显示的类,另一个是实现主界面显示的类。   创建两个类,主窗口类  登录窗口类  最后使用一个main函数,在这个函数...
recommend-type

zigbee-cluster-library-specification

最新的zigbee-cluster-library-specification说明文档。
recommend-type

管理建模和仿真的文件

管理Boualem Benatallah引用此版本:布阿利姆·贝纳塔拉。管理建模和仿真。约瑟夫-傅立叶大学-格勒诺布尔第一大学,1996年。法语。NNT:电话:00345357HAL ID:电话:00345357https://theses.hal.science/tel-003453572008年12月9日提交HAL是一个多学科的开放存取档案馆,用于存放和传播科学研究论文,无论它们是否被公开。论文可以来自法国或国外的教学和研究机构,也可以来自公共或私人研究中心。L’archive ouverte pluridisciplinaire
recommend-type

MATLAB柱状图在信号处理中的应用:可视化信号特征和频谱分析

![matlab画柱状图](https://img-blog.csdnimg.cn/3f32348f1c9c4481a6f5931993732f97.png) # 1. MATLAB柱状图概述** MATLAB柱状图是一种图形化工具,用于可视化数据中不同类别或组的分布情况。它通过绘制垂直条形来表示每个类别或组中的数据值。柱状图在信号处理中广泛用于可视化信号特征和进行频谱分析。 柱状图的优点在于其简单易懂,能够直观地展示数据分布。在信号处理中,柱状图可以帮助工程师识别信号中的模式、趋势和异常情况,从而为信号分析和处理提供有价值的见解。 # 2. 柱状图在信号处理中的应用 柱状图在信号处理
recommend-type

用Spring boot和vue写一个登录注册界面

好的,这是一个比较常见的技术问题,我可以为您提供一些参考资料和思路。首先,您可以使用Spring Boot作为后端框架来处理用户认证和注册的逻辑,比如使用Spring Security实现用户登录认证。同时,您还需要设计相应的数据模型和数据库表结构来存储用户信息。在前端方面,您可以使用Vue.js作为框架来构建登录注册页面,使用Axios来发起API请求并和后端进行交互。当然,在实现过程中,还需要考虑一些具体细节,比如数据校验、安全性和用户体验等方面。希望这些信息能够帮助到您。
recommend-type

JSBSim Reference Manual

JSBSim参考手册,其中包含JSBSim简介,JSBSim配置文件xml的编写语法,编程手册以及一些应用实例等。其中有部分内容还没有写完,估计有生之年很难看到完整版了,但是内容还是很有参考价值的。