自己制作VOC转yolo的软件

不多说，直接贴源码，很简单，直接复制下来运行就行

from PyQt5.QtWidgets import (QWidget, QVBoxLayout, QLabel, QLineEdit, QPushButton,QListWidget, QProgressBar, QHBoxLayout, QMainWindow, QFileDialog, QMenu,QDesktopWidget, QSplashScreen, QTimeEdit, QApplication, QStatusBar, QTextEdit, QMessageBox)
from PyQt5.QtCore import Qt, QThread, pyqtSignal, QTimer, QTime
from PyQt5.QtGui import QPixmap, QIcon
import sys
import os
import fnmatch
import subprocess
# from VOCToYOLO import convert_voc_to_yoloimport os
import xml.etree.ElementTree as ET
# VOC to yolo
def convert_voc_to_yolo(voc_path, output_path, class_names, progress_callback=None):"""Convert VOC format annotations to YOLO format annotations.:param voc_path: Directory containing VOC format XML files.:param output_path: Directory to save YOLO format TXT files.:param class_names: List of class names in the same order as their indices.:param progress_callback: Callback function to update progress bar."""if not os.path.exists(output_path):os.makedirs(output_path)xml_files = [f for f in os.listdir(voc_path) if f.endswith('.xml')]total_files = len(xml_files)processed_files = 0for xml_file in xml_files:xml_path = os.path.join(voc_path, xml_file)tree = ET.parse(xml_path)root = tree.getroot()# Extract the image file name without extensionfilename_without_ext = xml_file.split(".xml")[0]img_width = int(root.find('size/width').text)img_height = int(root.find('size/height').text)yolo_lines = []for obj in root.findall('object'):class_name = obj.find('name').textif class_name not in class_names:continueclass_id = class_names.index(class_name)bbox = obj.find('bndbox')xmin = float(bbox.find('xmin').text)ymin = float(bbox.find('ymin').text)xmax = float(bbox.find('xmax').text)ymax = float(bbox.find('ymax').text)# Convert to YOLO formatx_center = (xmin + xmax) / 2 / img_widthy_center = (ymin + ymax) / 2 / img_heightwidth = (xmax - xmin) / img_widthheight = (ymax - ymin) / img_heightyolo_lines.append(f"{class_id} {x_center} {y_center} {width} {height}")# Save to YOLO format with the same name as the image fileyolo_file = os.path.join(output_path, filename_without_ext + '.txt')with open(yolo_file, 'w') as f:f.write("\n".join(yolo_lines))processed_files += 1if progress_callback:progress_callback(processed_files * 100 // total_files)def resource_path(relative_path):""" 获取资源文件的绝对路径 """try:# PyInstaller创建的临时文件夹路径base_path = sys._MEIPASSexcept AttributeError:# 如果不是打包的环境，使用当前文件夹base_path = os.path.dirname(__file__)return os.path.join(base_path, relative_path)class MainWindow(QMainWindow):def __init__(self):super().__init__()self.setWindowTitle('YOLOv10数据集转换器')self.setGeometry(100, 100, 800, 400)self.setWindowIcon(QIcon(resource_path('source/image/logo.ico')))main_layout = QVBoxLayout()# ----------------------------------------------------#   文件加载部分# ----------------------------------------------------load_layout = QHBoxLayout()self.load_button = QPushButton("加载标注文件夹")self.load_button.clicked.connect(self.load_dir)load_layout.addWidget(self.load_button)self.line_edit = QLineEdit()self.line_edit.setPlaceholderText("请输入文本...")load_layout.addWidget(self.line_edit)# 填写输出路径output_layout = QHBoxLayout()self.outputdir_load_button = QPushButton("输出文件夹路径")self.outputdir_load_button.clicked.connect(self.load_output_dir)output_layout.addWidget(self.outputdir_load_button)self.outputdir_edit = QLineEdit()self.outputdir_edit.setPlaceholderText("待输出文本...")output_layout.addWidget(self.outputdir_edit)# 填写VOC类别class_layout = QHBoxLayout()self.write_class_message = QLabel("输入类别信息：")class_layout.addWidget(self.write_class_message)self.classwirte_edit = QLineEdit()self.classwirte_edit.setPlaceholderText("请输入类别信息如：{}；该信息默认".format(["ok", "YiWu", "CanJiao", "CanXi","ShaoJian","BianXing","PaoMianTiePian", "PaoMianLouTie"]))class_layout.addWidget(self.classwirte_edit)main_layout.addLayout(load_layout)main_layout.addLayout(output_layout)main_layout.addLayout(class_layout)# ----------------------------------------------------#   转换部分# ----------------------------------------------------button_layout = QHBoxLayout()self.start_button = QPushButton("开始转换")button_layout.addWidget(self.start_button)self.defor_button = QPushButton("判断格式")self.defor_button.clicked.connect(self.modeDet)button_layout.addWidget(self.defor_button)main_layout.addLayout(button_layout)# ----------------------------------------------------#   进度条部分# ----------------------------------------------------progress_layout = QHBoxLayout()self.progress_bar = QProgressBar()progress_layout.addWidget(self.progress_bar)main_layout.addLayout(progress_layout)# ----------------------------------------------------#   状态栏部分# ----------------------------------------------------system_layout = QVBoxLayout()self.text = QLabel('状态栏')system_layout.addWidget(self.text)self.textbox = QTextEdit(self)self.textbox.setReadOnly(True)system_layout.addWidget(self.textbox)main_layout.addLayout(system_layout)# 创建中心窗口部件central_widget = QWidget()central_widget.setLayout(main_layout)self.setCentralWidget(central_widget)# 创建状态栏self.status_bar = QStatusBar()self.setStatusBar(self.status_bar)# 连接信号到槽self.line_edit.textChanged.connect(self.on_text_changed)self.load_button.clicked.connect(self.on_load_button_clicked)self.start_button.clicked.connect(self.on_start_button_clicked)self.mode = "Unknow"def on_text_changed(self, text):print(f"输入的文本: {text}")def on_load_button_clicked(self):self.status_bar.showMessage("加载标注文件夹...")# 在这里添加加载标注文件夹的逻辑def load_dir(self):print("load_dir is running!")# 打开文件夹选择对话框self.dir_path = QFileDialog.getExistingDirectory(self, 'Select Directory')if self.dir_path:self.line_edit.setText(self.dir_path)# 获取目录中的所有文件self.files = [f for f in os.listdir(self.dir_path) if f.lower().endswith(('.xml'))]if not self.files:self.mode = "Qita"QMessageBox.information(self, 'No message Found', 'No xml files found in the selected directory.')returnself.mode = "VOC"for file_name in self.files:self.textbox.append("文件名为：{}".format(file_name))else:QMessageBox.information(self, 'Selection Cancelled', '您已取消文件夹选择')def load_output_dir(self):self.output_path = QFileDialog.getExistingDirectory(self, 'Select Directory')print(self.output_path)if self.output_path:self.outputdir_edit.setText(self.output_path)else:QMessageBox.information(self, 'Selection Cancelled', '您已取消文件夹选择')def modeDet(self):self.status_bar.showMessage("数据集格式为：{}".format(self.mode))def on_start_button_clicked(self):print("inter is running!")self.status_bar.showMessage("开始转换...")if not self.classwirte_edit.text():class_message = ["ok", "YiWu", "CanJiao", "CanXi","ShaoJian", "BianXing","PaoMianTiePian", "PaoMianLouTie"]else:class_message = self.classwirte_edit.text().split(',')try:if self.dir_path and self.output_path and class_message:self.progress_bar.setValue(0)convert_voc_to_yolo(self.dir_path, self.output_path, class_message, progress_callback=self.update_progress)QMessageBox.information(self, '转换完成！', '嘿嘿')else:QMessageBox.information(self, '缺少信息', '请重新检查以上步骤！')except Exception as e:QMessageBox.information(self, '报错！', '错误为：{}'.format(e))# 在这里添加开始转换的逻辑def update_progress(self, value):self.progress_bar.setValue(value)if __name__ == '__main__':app = QApplication(sys.argv)splash_pix = QPixmap(resource_path('source/image/loading.png'))splash = QSplashScreen(splash_pix)splash.show()window = MainWindow()QTimer.singleShot(500, lambda: (splash.close(), window.show()))screen = QDesktopWidget().screenGeometry()size = window.geometry()window.move((screen.width() - size.width()) // 2, (screen.height() - size.height()) // 2-120)sys.exit(app.exec_())

mainUI.spec

# -*- mode: python ; coding: utf-8 -*-block_cipher = Nonea = Analysis(['mainUI.py'],pathex=['E:\\0_ALL_Cpde\\PyQt5_practice1\\YoloDatasetProcess'],binaries=[],datas=[('source/image/loading.png', 'source/image/'), ('source/image/logo.ico', 'source/image/')],hiddenimports=[],hookspath=[],hooksconfig={},runtime_hooks=[],excludes=[],win_no_prefer_redirects=False,win_private_assemblies=False,cipher=block_cipher,
)
pyz = PYZ(a.pure)exe = EXE(pyz,a.scripts,a.binaries,a.datas,[],name='VOC2YOLO转换器',debug=False,bootloader_ignore_signals=False,strip=False,upx=True,upx_exclude=[],runtime_tmpdir=None,console=False,disable_windowed_traceback=False,argv_emulation=False,target_arch=None,codesign_identity=None,entitlements_file=None,icon='source/image/logo.ico'
)

本文来自互联网用户投稿，该文观点仅代表作者本人，不代表本站立场。本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。如若转载，请注明出处：http://www.mzph.cn/bicheng/52394.shtml

如若内容造成侵权/违法违规/事实不符，请联系多彩编程网进行投诉反馈email:809451989@qq.com，一经查实，立即删除！

相关文章

Kubernetes 中如何对 etcd 进行备份和还原

Kubernetes 中如何对 etcd 进行备份和还原

etcd 是 Kubernetes 集群的重要组件，存储了集群的所有数据，如配置文件、状态信息、以及服务发现数据。因此，定期备份和能够快速还原 etcd 数据是保障 Kubernetes 集群安全与稳定运行的关键步骤。本文将详细介绍如何对 etcd 进行备份与还原的具…
阅读更多...
7z解压crc错误 7-Zip-常见问题解答

7z解压crc错误 7-Zip-常见问题解答

7-Zip 是一个流行的文件压缩和解压缩工具，但在使用过程中，有时会遇到 CRC（循环冗余校验）错误。这通常意味着压缩文件已损坏或未完全下载。以下是一些可能的原因及解决方案。 常见原因 文件损坏：下载过程中出现错误&a…
阅读更多...
2024年【电气试验】找解析及电气试验模拟考试

2024年【电气试验】找解析及电气试验模拟考试

题库来源：安全生产模拟考试一点通公众号小程序 电气试验找解析根据新电气试验考试大纲要求，安全生产模拟考试一点通将电气试验模拟考试试题进行汇编，组成一套电气试验全真模拟考试试题，学员可通过电气试验模拟考试全真模拟&#…
阅读更多...
文件IO和多路复用IO

文件IO和多路复用IO

目录 前言 一、文件 I/O 1.基本文件 I/O 操作 1.1打开文件 1.2读取文件内容 (read) 1.3写入文件 (write) 1.4关闭文件 (close) 2.文件指针 二、多路复用 I/O 1.常用的多路复用 I/O 模型 1.1select 1.2poll 1.3epoll 2.使用 select、poll 和 epoll 进行简单的 I/O…
阅读更多...
C++观察者模式Observer

C++观察者模式Observer

组件协作 –(都是晚绑定的） ----观察者模式 为某些对象建立一种通知依赖的关系， 只要这个对象状态发生改变，观察者对象都能得到通知。 但是依赖关系要松耦合，不要太依赖。 eg：做一个文件分割器，需要一个…
阅读更多...
css实现水滴效果图

css实现水滴效果图

效果图&#xff1a; <template><div style"width: 100%;height:500px;padding:20px;"><div class"water"></div></div> </template> <script> export default {data() {return {};},watch: {},created() {},me…
阅读更多...
B/S架构和C/S架构的区别

B/S架构和C/S架构的区别

B/S架构、C/S架构区别 1. B/S架构 1.1 什么是B/S架构 B/S架构的全称为Browser/Server&#xff0c;即浏览器/服务器结构。Browser指的是Web浏览器&#xff0c;极少数事务逻辑在前端实现&#xff0c;但主要事务逻辑在服务器端实现。B/S架构的系统无须特别安装&#xff0c;只需要…
阅读更多...
C知识扫盲-------文件结束符（EOF）

C知识扫盲-------文件结束符（EOF）

引言&#xff1a; 在文件操作中&#xff0c;EOF 是 “End of File”&#xff08;文件结束&#xff09;的缩写&#xff0c;用于指示文件的结束。EOF 是一个特殊的整型值&#xff08;int&#xff09;&#xff0c;当文件指针到达文件末尾时&#xff0c;许多文件操作函数会返回这个…
阅读更多...
动态内存管理-经典笔试题

动态内存管理-经典笔试题

目录 题目一&#xff1a; 题目二&#xff1a; 题目三&#xff1a; 题目四&#xff1a; 题目一&#xff1a; 结果&#xff1a;程序崩溃 原因&#xff1a; 1、函数是传值调用&#xff0c;出了函数p不存在&#xff0c;str未改变&#xff0c;依旧为空指针&#xff0c;运行时发…
阅读更多...
【CTF Web】CTFShow 版本控制泄露源码2 Writeup（目录扫描+.svn泄漏）

【CTF Web】CTFShow 版本控制泄露源码2 Writeup（目录扫描+.svn泄漏）

版本控制泄露源码2 10 版本控制很重要&#xff0c;但不要部署到生产环境更重要。 解法 用 dirsearch 扫描。 dirsearch -u https://8d22223d-dc2c-419c-b82d-a1d781eda427.challenge.ctf.show/找到 .svn 仓库。 访问&#xff1a; https://8d22223d-dc2c-419c-b82d-a1d781eda…
阅读更多...
ubuntu安装minio

ubuntu安装minio

# 下载MinIO的可执行文件 curl -O https://dl.min.io/server/minio/release/linux-amd64/minio # 添加执行权限 chmod x minio # 运行MinIO (需要先创建存储数据和存储存储桶的目录) nohup ./minio server /home/lighthouse/minioDir/data /home/lighthouse/minioDir/bucke…
阅读更多...
修复线上问题，又造成了个Bug

修复线上问题，又造成了个Bug

项目场景 这是一位朋友给我分享的实际线上问题&#xff0c;看似简单却害了项目再次造出一个bug来。 场景&#xff1a;线上环境中&#xff0c;需要查询某某业务数据&#xff0c;条件是状态&#xff0c;之前产品只要求查两个状态的&#xff0c;但现在让他再多查一个状态的。 自…
阅读更多...
时序预测|基于贝叶斯BO-卷积-双向门控单元-注意力机制的单变量时间序列预测模型BO-CNN-BiGRU-Attention

时序预测|基于贝叶斯BO-卷积-双向门控单元-注意力机制的单变量时间序列预测模型BO-CNN-BiGRU-Attention

时序预测|基于贝叶斯BO-卷积-双向门控单元-注意力机制的单变量时间序列预测模型BO-CNN-BiGRU-Attention 文章目录 前言时序预测|基于贝叶斯BO-卷积-双向门控单元-注意力机制的单变量时间序列预测模型BO-CNN-BiGRU-Attention 一、BO-CNN-BiGRU-Attention模型1. 贝叶斯优化&#…
阅读更多...
Go Convey测试框架入门(go convey gomonkey)

Go Convey测试框架入门(go convey gomonkey)

Go Convey测试框架入门 介绍 GoConvey是一款针对Golang的测试框架&#xff0c;可以管理和运行测试用例&#xff0c;同时提供了丰富的断言函数&#xff0c;并支持很多 Web 界面特性。 Golang虽然自带了单元测试功能&#xff0c;并且在GoConvey框架诞生之前也出现了许多第三方测…
阅读更多...
OSPF路由原理详解与关键点

OSPF路由原理详解与关键点

目录 一. OSPF简介: 二. OSPF原理描述: 三. OSPF的核心内容: 四. OSPF的邻居关系和邻接 五. LSA在各区域中传播的支持情况 一. OSPF简介: 开放式最短路径优先OSPF&#xff08;Open Shortest Path First&#xff09;是IETF组织开发的一个基于链路状态的内部网关协议&…
阅读更多...
技术债务已接管经济

技术债务已接管经济

“技术债务”一词通常指软件开发过程中的捷径或次优方法。它表现为设计不良的代码、缺乏文档和过时的组件。虽然正确编写的代码和文档是永恒的&#xff0c;但组件和方法却不是。随着时间的推移&#xff0c;软件及其组件可能会成为技术债务。自 40 年前的 20 世纪 80 年代软件行…
阅读更多...
Linux判断语句if/case以及标准输出

Linux判断语句if/case以及标准输出

bash shell脚本&#xff1a;把平时在命令行执行的命令放在一个文本文件内&#xff0c;此文件即shell脚本 注意&#xff1a;绝大部分是非交互式命令执行脚本# bash 脚本文件# ./脚本文件# 路径/脚本文件交互式命令敲完回车之后没有直接出结果&#xff0c;并且需要在输入其他内容…
阅读更多...
【嵌入式开发之网络编程】TCP端口和UDP端口

【嵌入式开发之网络编程】TCP端口和UDP端口

目录 网络端口的定义及作用 运输层的作用 运输层的两个主要协议 用户数据报协议UDP (User Datagram Protocol) 传输控制协议TCP (Transmission Control Protocol) 运输层的端口及分类 按照端口号分类 按照协议类型分类 BSD端口 网络端口的定义及作用 在网络技术中…
阅读更多...
《通义千问AI落地—下》：WebSocket详解

《通义千问AI落地—下》：WebSocket详解

一、前言 文本源自 微博客 且已获授权,请尊重版权。 《通义千问AI落地——下篇》如约而至。Websocket在这一类引用中，起到前后端通信的作用。因此，本文将介绍websocket在这类应用场景下的配置、使用、注意事项以及ws连接升级为wss连接等；如下图，本站已经使用了wss连接…
阅读更多...
iphone异常问题常用修复方法

iphone异常问题常用修复方法

作为智能手机的领军者&#xff0c;iPhone凭借其卓越的性能和稳定的系统赢得了全球用户的青睐。然而&#xff0c;就像任何电子设备一样&#xff0c;iPhone在使用过程中也难免会遇到各种异常问题&#xff0c;如卡顿、无法充电、应用闪退等。这些问题虽然令人头疼&#xff0c;但大…
阅读更多...
最新文章

Copyright @ 2022~2023