55、记录标注的关键点和目标检测数据集，进行目标检测标注文件和关键点文件分离以及yolov5.txt标注格式转xml数据集

2023-05-16

基本思想：手中存在一份关键点检测模型，因为客户把检测框和目标关键点框都标注在一起了，所以进行一下剥离

代码一：从关标注数据集剥离目标检测

# -*- coding: utf-8 -*-
import cv2
import json
import io
import os
import glob
from xml.dom.minidom import Document

path = r'F:\trash\test'
destPath = r"F:\trash\test"




m_folder = ""
print('m_folder=', m_folder)
m_database = 'Unknown'
print('m_database=', m_database)
m_depth = 3
print('m_depth=', m_depth)
m_segmented = 0
print('m_segmented=', m_segmented)

m_pose = 'Unspecified'
print('m_pose=', m_pose)
m_truncated = 0
print('m_truncated=', m_truncated)
m_difficult = 0
print('m_difficult=', m_difficult)
m_segmented = 0
print('m_segmented=', m_segmented)

path_list = glob.glob(path+"\*.json")
for name in enumerate(path_list):
    m_path = name[1]
    print("m_path=",m_path)
    dir = os.path.dirname(m_path)
    name__,__=os.path.split(name[1])
    ff,ext=os.path.splitext(__)
    print('dir=', dir)

    file_json = io.open(m_path, 'r', encoding='utf-8')
    json_data = file_json.read()
    data = json.loads(json_data)
    m_filename = ff+".jpg"
    print('m_filename=', m_filename)
    m_path = os.path.join(dir, m_filename)
    #print('m_path=', m_path)
    m_width = data['imageWidth']
    #print('m_width=', m_width)
    m_height = data['imageHeight']
    #print('m_height=', m_height)
    object_name = os.path.splitext(m_filename)[0]
    new_object_name = object_name + '.xml'
    img_name = object_name + ".jpg"

    #print(new_object_name)
    doc = Document()  # 创建DOM文档对象
    DOCUMENT = doc.createElement('annotation')  # 创建根元素

    folder = doc.createElement('folder')
    folder_text = doc.createTextNode(m_folder)
    folder.appendChild(folder_text)
    DOCUMENT.appendChild(folder)
    doc.appendChild(DOCUMENT)

    filename = doc.createElement('filename')
    filename_text = doc.createTextNode(m_filename)
    filename.appendChild(filename_text)
    DOCUMENT.appendChild(filename)
    doc.appendChild(DOCUMENT)

    path = doc.createElement('path')
    path_text = doc.createTextNode(m_filename)
    path.appendChild(path_text)
    DOCUMENT.appendChild(path)
    doc.appendChild(DOCUMENT)

    source = doc.createElement('source')
    database = doc.createElement('database')
    database_text = doc.createTextNode(m_database)  # 元素内容写入
    database.appendChild(database_text)
    source.appendChild(database)
    DOCUMENT.appendChild(source)
    doc.appendChild(DOCUMENT)

    size = doc.createElement('size')
    width = doc.createElement('width')
    width_text = doc.createTextNode(str(m_width))  # 元素内容写入
    width.appendChild(width_text)
    size.appendChild(width)

    height = doc.createElement('height')
    height_text = doc.createTextNode(str(m_height))
    height.appendChild(height_text)
    size.appendChild(height)

    depth = doc.createElement('depth')
    depth_text = doc.createTextNode(str(m_depth))
    depth.appendChild(depth_text)
    size.appendChild(depth)

    DOCUMENT.appendChild(size)

    segmented = doc.createElement('segmented')
    segmented_text = doc.createTextNode(str(m_segmented))
    segmented.appendChild(segmented_text)
    DOCUMENT.appendChild(segmented)
    doc.appendChild(DOCUMENT)
    for i in range(len(data['shapes'])):
        if data['shapes'][i]['shape_type'] !="rectangle":
            continue
        m_xmin_0 = data['shapes'][i]['points'][0][0]
        #print('m_xmin_0=', m_xmin_0)
        m_ymin_0 = data['shapes'][i]['points'][0][1]
        #print('m_ymin_0=', m_ymin_0)
        m_xmax_0 = data['shapes'][i]['points'][1][0]
        #print('m_xmax_0=', m_xmax_0)
        m_ymax_0 = data['shapes'][i]['points'][1][1]
        #print('m_ymax_0=', m_ymax_0)
        m_name_0 ="tilapia"# data['shapes'][i]['label']
        #print('m_name_0=', m_name_0)
        object = doc.createElement('object')
        name = doc.createElement('name')
        name_text = doc.createTextNode(m_name_0)
        name.appendChild(name_text)
        object.appendChild(name)

        pose = doc.createElement('pose')
        pose_text = doc.createTextNode(m_pose)
        pose.appendChild(pose_text)
        object.appendChild(pose)

        truncated = doc.createElement('truncated')
        truncated_text = doc.createTextNode(str(m_truncated))
        truncated.appendChild(truncated_text)
        object.appendChild(truncated)

        difficult = doc.createElement('difficult')
        difficult_text = doc.createTextNode(str(m_difficult))
        difficult.appendChild(difficult_text)
        object.appendChild(difficult)

        bndbox = doc.createElement('bndbox')
        xmin = doc.createElement('xmin')
        xmin_text = doc.createTextNode(str(int(m_xmin_0)))
        xmin.appendChild(xmin_text)
        bndbox.appendChild(xmin)

        ymin = doc.createElement('ymin')
        ymin_text = doc.createTextNode(str(int(m_ymin_0)))
        ymin.appendChild(ymin_text)
        bndbox.appendChild(ymin)

        xmax = doc.createElement('xmax')
        xmax_text = doc.createTextNode(str(int(m_xmax_0)))
        xmax.appendChild(xmax_text)
        bndbox.appendChild(xmax)

        ymax = doc.createElement('ymax')
        ymax_text = doc.createTextNode(str(int(m_ymax_0)))
        ymax.appendChild(ymax_text)
        bndbox.appendChild(ymax)
        object.appendChild(bndbox)

        DOCUMENT.appendChild(object)
    new_path_filename = os.path.join(destPath, new_object_name)
    print('new_path_filename=', new_path_filename)
    f = open(new_path_filename, 'w')

    doc.writexml(f, indent='\t', newl='\n', addindent='\t', encoding='utf-8')
    f.close()

txt数据集格式转xml

# -*- coding: utf-8 -*-
import glob
import json
import io
import os
import cv2
from xml.dom.minidom import Document

path = '/home/ubuntu/Downloads/RetinaFace/images'
destPath = "/home/ubuntu/Downloads/RetinaFace/images"


m_folder = os.path.basename(path)
print('m_folder=', m_folder)
m_database = 'Unknown'
print('m_database=', m_database)
m_depth = 3
print('m_depth=', m_depth)
m_segmented = 0
print('m_segmented=', m_segmented)

m_pose = 'Unspecified'
print('m_pose=', m_pose)
m_truncated = 0
print('m_truncated=', m_truncated)
m_difficult = 0
print('m_difficult=', m_difficult)
m_segmented = 0
print('m_segmented=', m_segmented)

path_list = glob.glob(path+"/*.txt")
for name in path_list:
    m_path,name_ = os.path.split(name)
    name_p, ext_ = os.path.splitext(name_)
    img_jpg=".".join([name_p,"jpg"])
    image_src=cv2.imread(os.path.join(m_path,img_jpg))
    m_width,m_height,_=image_src.shape[0],image_src.shape[1],image_src.shape[2]
    m_filename = img_jpg
    print('m_path=', m_path)
    print('m_height=', m_height)
    object_name = os.path.splitext(m_filename)[0]
    new_object_name = object_name + '.xml'
    img_name = object_name + ".jpg"

    print(new_object_name)
    doc = Document()  # 创建DOM文档对象
    DOCUMENT = doc.createElement('annotation')  # 创建根元素

    folder = doc.createElement('folder')
    folder_text = doc.createTextNode(m_folder)
    folder.appendChild(folder_text)
    DOCUMENT.appendChild(folder)
    doc.appendChild(DOCUMENT)

    filename = doc.createElement('filename')
    filename_text = doc.createTextNode(m_filename)
    filename.appendChild(filename_text)
    DOCUMENT.appendChild(filename)
    doc.appendChild(DOCUMENT)

    path = doc.createElement('path')
    path_text = doc.createTextNode(m_filename)
    path.appendChild(path_text)
    DOCUMENT.appendChild(path)
    doc.appendChild(DOCUMENT)

    source = doc.createElement('source')
    database = doc.createElement('database')
    database_text = doc.createTextNode(m_database)  # 元素内容写入
    database.appendChild(database_text)
    source.appendChild(database)
    DOCUMENT.appendChild(source)
    doc.appendChild(DOCUMENT)

    size = doc.createElement('size')
    width = doc.createElement('width')
    width_text = doc.createTextNode(str(m_width))  # 元素内容写入
    width.appendChild(width_text)
    size.appendChild(width)

    height = doc.createElement('height')
    height_text = doc.createTextNode(str(m_height))
    height.appendChild(height_text)
    size.appendChild(height)

    depth = doc.createElement('depth')
    depth_text = doc.createTextNode(str(m_depth))
    depth.appendChild(depth_text)
    size.appendChild(depth)

    DOCUMENT.appendChild(size)

    segmented = doc.createElement('segmented')
    segmented_text = doc.createTextNode(str(m_segmented))
    segmented.appendChild(segmented_text)
    DOCUMENT.appendChild(segmented)
    doc.appendChild(DOCUMENT)
    with open(name ,encoding='utf-8') as f:
        lines = f.readlines()
        for j, line in enumerate(lines):
            class_id, x, y, w, h = line.strip().split(' ')
            class_id, x, y, w, h = int(class_id), float(x), float(y), float(w), float(h)  # 将字符串类型转为可计算的int和float类型

            m_xmin_0 = (x - w / 2) * m_width  # 坐标转换
            m_ymin_0 = (y - h / 2) * m_height
            m_xmax_0 = (x + w / 2) * m_width
            m_ymax_0 = (y + h / 2) * m_height

            print('m_ymax_0=', m_ymax_0)
            m_name_0 = str(class_id)
            print('m_name_0=', m_name_0)
            object = doc.createElement('object')
            name = doc.createElement('name')
            name_text = doc.createTextNode(m_name_0)
            name.appendChild(name_text)
            object.appendChild(name)

            pose = doc.createElement('pose')
            pose_text = doc.createTextNode(m_pose)
            pose.appendChild(pose_text)
            object.appendChild(pose)

            truncated = doc.createElement('truncated')
            truncated_text = doc.createTextNode(str(m_truncated))
            truncated.appendChild(truncated_text)
            object.appendChild(truncated)

            difficult = doc.createElement('difficult')
            difficult_text = doc.createTextNode(str(m_difficult))
            difficult.appendChild(difficult_text)
            object.appendChild(difficult)

            bndbox = doc.createElement('bndbox')
            xmin = doc.createElement('xmin')
            xmin_text = doc.createTextNode(str(int(m_xmin_0)))
            xmin.appendChild(xmin_text)
            bndbox.appendChild(xmin)

            ymin = doc.createElement('ymin')
            ymin_text = doc.createTextNode(str(int(m_ymin_0)))
            ymin.appendChild(ymin_text)
            bndbox.appendChild(ymin)

            xmax = doc.createElement('xmax')
            xmax_text = doc.createTextNode(str(int(m_xmax_0)))
            xmax.appendChild(xmax_text)
            bndbox.appendChild(xmax)

            ymax = doc.createElement('ymax')
            ymax_text = doc.createTextNode(str(int(m_ymax_0)))
            ymax.appendChild(ymax_text)
            bndbox.appendChild(ymax)
            object.appendChild(bndbox)

            DOCUMENT.appendChild(object)
    new_path_filename = os.path.join(destPath, new_object_name)
    print('new_path_filename=', new_path_filename)
    f = open(new_path_filename, 'w')

    doc.writexml(f, indent='\t', newl='\n', addindent='\t', encoding='utf-8')
    f.close()

本文内容由网友自发贡献，版权归原作者所有，本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容，请联系:hwhale#tublm.com(使用前将#替换为@)

yolov5

txt

xml

记录标注的关键点和目标检测数据集

进行目标检测标注文件和关键点文件分离以及

55、记录标注的关键点和目标检测数据集，进行目标检测标注文件和关键点文件分离以及yolov5.txt标注格式转xml数据集的相关文章

C# XML 反序列化。将节点中的所有内部文本读取到字符串属性中

我目前正在尝试修改我的类以便我的模型上的文本属性包含某个节点的所有内部文本 text node 给我带来问题的 xml 示例是
xml 拉解析器资产 xml

如何使用拉解析器解析资产文件夹中的本地 XML 文件我无法让拉解析器工作它总是抛出 io 异常我想我无法获取文件的路径或连接到该文件 mixm 我正在尝试各种方法来从资产和资源加载本地文件但要按要求回答您的问题因为其他人似
Xslt 到 xsl-fo 转换

我想将 xslt 转换为 xsl fo 但我不太确定我能做到这一点我尝试将 XML 列表转换为 xsl fo 列表谁能告诉我在哪里可以找到我在谷歌上搜索了很长时间没有很多这样的例子我的XML是这样的 p TEXT p ul li It
Android 键盘清单未显示在设置中

我正在制作我的第一个 Android 应用程序我需要它作为键盘服务据我所知清单看起来不错并且我有一个文件 WifiJoy java 在 com zwad3 wifijoy 包中以及所有其他文件
如何使用scrapy抓取xml url

你好我正在使用 scrapy 来抓取 xml url 假设下面是我的 Spider py 代码 class TestSpider BaseSpider name test allowed domains www example com s
c#Registry to XML无效字符问题

我在尝试从注册表创建 XML 文件时遇到问题在我的笔记本电脑 W7 64b 上它工作正常生成了 xml 文件但在另一台计算机 Xp 32b 上抛出异常 System ArgumentException 十六进制值 0x00 是无效字符
允许 .NET WebApi 忽略 DOCTYPE 声明

我正在尝试通过 WebApi 方法将 XML 反序列化为对象我有以下课程 XmlRoot IsNullable false public class MyObject XmlElement Name public string Name
& 在 xml 文件中算作一个还是多个字符？

我正在使用的 XML 模式具有特定的字符串字符长度所以我可能有一个类似的字符串 Jim Mary 在 C 中是 10 个字符但是当它写入 xml 时它会变成 Jim amp Mary 如果 XML 模式规定字符串最多只能有 10 个字
文本视图不显示全文

我正在使用 TableLayout 和 TableRow 创建一个简单的布局其中包含两个 TextView 这是代码的一部分
为 XSD 中的元素指定命名空间

我有一个 XML 需要为其生成 XSD 我的 XML 如下实例
Eclipse 在处理 xml（传入或传出）时挂起。此时CPU+磁盘尖峰

我每天必须处理许多 XML 这极大地打断了我的工作流程我尝试降级到 Indigo 但所有附加组件都被删除打开xml时出现问题切换到 xml 切换出一个 xml 甚至切换到另一个 xml 到底发生了什么事情让 XML 如此烦人我可以禁
将巨大的模式编译成Java

有两个主要工具提供了将 XSD 模式编译为 Java 的方法 xmlbeans 和 JAXB 问题是 XSD 模式确实很大 30MB 的 XML 文件大部分模式在我的项目中没有使用所以我可以注释掉大部分代码但这不是一个好的解决方案目
有人可以推荐一个免费的 xslt 工具吗？ [关闭]

就目前情况而言这个问题不太适合我们的问答形式我们希望答案得到事实参考资料或专业知识的支持但这个问题可能会引发辩论争论民意调查或扩展讨论如果您觉得这个问题可以改进并可能重新开放访问帮助中心 help reopen questi
VBA XML V6.0 如何让它等待页面加载？

我一直在努力寻找答案但似乎找不到任何有用的东西基本上我是从一个网站上拉取的当您在该页面上时该网站会加载更多项目我希望我的代码在加载完成后提取最终数据但不知道如何让 XML httprequest 等待 Edited Sub p
XML 创建 - 错误：带有替代方案的重载方法构造函数 UnprefixedAttribute

scala gt val count 7 count Int 7 将其放入 XML 属性中会出现错误 scala gt val x
在 Java 中创建 XML 文件的最佳方法是什么？

我们目前使用 dom4j 来创建 XML 文件不过我猜现在有更好的东西了如果我们使用的是 Java 1 6 或更高版本那么在编写 XML 文件时最好使用什么类运行速度最快使用简单我不需要构建一个 DOM 然后编写整个 DOM
使用 GWT 读取非常大的本地 XML 文件

我正在使用 GWT 构建我的第一个 Java 应用程序它必须从一个非常大的 XML 文件中读取数据当我尝试发送对文件中信息的请求时遇到问题并且我不太确定它是否与文件的大小或我的语义有关在我的程序中我有以下内容 static fin
CakePHP Xml 实用程序库触发 DOMDocument 警告

我正在使用 CakePHP 在视图中生成 XMLXML核心库 http book cakephp org 2 0 en core utility libraries xml html xml Xml build data array ret
如何在 Spring 属性中进行算术运算？
如何在Android中解析xml类型的HTTPResponse

我有一个 Android 应用程序我使用 POST 方法来获取响应这是我的代码 HttpResponse httpResponse httpclient execute httppost HttpEntity resEntity htt

随机推荐

【踩坑】通过端口转发解决Jupyter远程无法连接ERR_CONNECTION_TIMED_OUT

转载请注明出处 xff1a 小锋学长生活大爆炸 xfxuezhang cn 目录方式一 xff1a SSH 方式二 xff1a Mobaxterm 如果你也试了各种方法 xff0c 配置文件也改了 xff0c 还是无法连接上 xff0c
【教程】Jupyter相关记录

转载请注明出处 xff1a 小锋学长生活大爆炸 xfxuezhang cn 踩坑类踩坑 Jupyter无法找到Conda创建的虚拟环境踩坑通过端口转发解决Jupyter远程无法连接ERR CONNECTION TIMED OUT 配置
【教程】使用Nvidia System Monitor GUI监控GPU的运行状态

github xff1a Task Manager for Linux for Nvidia graphics cards sudo apt install git qtdeclarative5 dev cmake git clone ht
FreeRTOS 软定时器实现

原文地址 xff1a http blog csdn net qq 18150497 article details 52874310 简述考虑平台硬件定时器个数限制的 xff0c FreeRTOS 通过一个 Daemon 任务 xff08
【教程】手机远程连接另一台手机的termux进行拍照

主要通过MQTT实现通信图片的上传和显示使用了服务器对于没有服务器的同学 xff0c 可以进入官网领取免费1个月的轻量云服务器 xff1a 云产品免费试用 xff1b 需要选购的进 xff1a 轻量应用服务器专场 xff1b 不清楚怎么
【踩坑】Openssl升级后，版本与Library版本不一致的问题

转载请注明出处 xff1a 小锋学长生活大爆炸 xfxuezhang cn openssl和openssh的详细升级教程可参考 xff1a 一些服务器常见漏洞的修复方法可以通过ldd usr local openssl bin opens
【教程】保姆级红米AX6000刷UBoot和OpenWrt固件

转载请注明出处 xff1a 小锋学长生活大爆炸 xfxuezhang cn 目录开启SSH 刷入UBoot 刷入Openwrt 设置Openwrt 刷回小米原厂固件相关文件已为大家上传到国内云盘 xff1a https xfxuezha
【教程】如何从 PowerPoint 导出高分辨率(高 dpi)的图片 [附脚本]

如何从 PowerPoint 导出高分辨率 xff08 高 dpi xff09 幻灯片 Office Microsoft Learn 总结为 xff1a 改注册表顺手写个脚本 xff1a import tkinter as tk from
【踩坑】PyG报错Undefined symbol: xxxxxx

首先确保pytorch pyg cuda的版本对应其次 xff0c 所有步骤的安装 xff0c 都是用pip 不要pip跟conda混用 xff01 xff01 比如 xff0c 这里是PyTorch 2 0 1 CUDA11 7 xff
【教程】安装VSCode-Server

转载请注明出处 xff1a 小锋学长生活大爆炸 xfxuezhang cn 前景提要 jupyter虽然好用 xff0c 但他只能运行ipynb文件 xff0c 对于py文件还是只能通过命令行来运行 xff0c 不是很方便因此 xff0c
48、OAK通过共享内存传递变长结构体(Rapidjson)进行数据和图片交互

基本思想 xff1a 主要学习一下在共享内存中传递变长的数据 xff0c 这样在c 调用c 43 43 dll也可以雷同操作 xff0c 以oak的检测和共享内存为代码整合 xff0c 集成了rapidjson的使用代码自己摘要和参考吧
49、OAK测试官方的IMU模块和SpatialLocationCalculator节点

基本思想 xff1a 不太懂IMU是干嘛的 xff0c 不像图像那么容易可视化 xff0c 参考官方demo的 xff0c 记录一下 xff0c 后续这篇需要补充 xff0c 参考的IMU的介绍 xff0c 原理不懂 xff0c 先占个坑
50、OAK的VideoEncoder节点编码h265和使用ffmpeg进行解码播放(用在POE的OAK效果更好)

基本思想 xff1a 幸亏会一点点ffmpeg的视频编解码进行了视频播放本篇博客教你如何直接解码h265进行视频播放 xff0c 还得立下flag 要学如何硬解码 cmakelists txt cmake minimum required
51、部署PaddleSeg的pp_liteseg到MNN框架、OpenVINO框架和OAK框架、NPU(RK3399 PRO)框架

基本思想 xff1a 需要一个快的实例分割模型 xff0c 由于需要配置oak使用 xff0c 所以就记录和实现一下微软社区提供的思路 xff0c 去部署PaddleSeg的轻量级实际是语义分割模型所有的实验模型 xff0c 花了两天
关于头文件中的 static inline函数

原文地址 xff1a http blog csdn net huanghui167 article details 41346663 关于头文件中的 static inline函数头文件中常见static inline函数 xff0c 于
52、训练paddleSeg模型，部署自己的模型到OAK相机上

基本思想 xff1a 简单记录一下训练过程 xff0c 数据集在coco基础上进行 xff0c 进行筛选出杯子的数据集 xff0c 然后进行训练 xff0c 比较简单 xff0c 从coco数据集中筛选出杯子的数据集 xff0c 然后在la
52、人脸检测部署RK3399 PRO，完成RKNN的人脸检测

基本思想 xff1a 帮助好友部署一个人脸检测模型 xff0c 也是自己业务需求 xff0c 部分代码来自好友陈同学 xff0c 自己改了改c 43 43 的代码可以部署rk3399pro上了 xff0c 其它资料见附录吧 xff0c 官方
53、正确导出onnx模型，以及替换对应的算子支持

基本思想 xff1a 学习tensorRT教程 xff0c 来自bilibi 参考附录一和在实际算法使用中遇到的总结一代码 xff1a import torch import torch nn as nn class Model nn
54、记录yolov7 训练、部署ncnn、部署mnn、部署rk3399pro npu、部署openvino、部署oak vpu、部署TensorRT

基本思想 xff1a 记录一下yolov7训练人头检测和部署oak的vpu相机上和rk3399 pro开发板上 xff0c 完成需求 ubuntu 64 ubuntu yolov7 pip install torch 61 61 1 12
55、记录标注的关键点和目标检测数据集，进行目标检测标注文件和关键点文件分离以及yolov5.txt标注格式转xml数据集

基本思想 xff1a 手中存在一份关键点检测模型 xff0c 因为客户把检测框和目标关键点框都标注在一起了 xff0c 所以进行一下剥离代码一 xff1a 从关标注数据集剥离目标检测 coding utf 8 import cv2 imp

55、记录标注的关键点和目标检测数据集，进行目标检测标注文件和关键点文件分离以及yolov5.txt标注格式转xml数据集

55、记录标注的关键点和目标检测数据集，进行目标检测标注文件和关键点文件分离以及yolov5.txt标注格式转xml数据集 的相关文章

随机推荐

热门标签

55、记录标注的关键点和目标检测数据集，进行目标检测标注文件和关键点文件分离以及yolov5.txt标注格式转xml数据集的相关文章