使用 golang xml 标准库转换 SeaShips(7000) 数据集标注文件格式到 YOLO数据格式

驭鲸·环球

已于 2022-03-04 11:56:58 修改

阅读量1k

点赞数

分类专栏： Darknet 文章标签： xml golang 目标检测

于 2022-02-21 15:42:32 首次发布

知识因分享而具有价值，回忆昨天，把握今天，期待明天。 —— 驭鲸·环球

本文链接：https://blog.csdn.net/HighproX/article/details/123048337

版权

本文介绍如何利用Golang的xml标准库将包含7000张图片的SeaShips数据集的标注文件转换为YOLO格式，以供Yolov4训练目标检测模型使用。通过160行代码实现转换，并提供了转换后标注文件的验证方法。

摘要由CSDN通过智能技术生成

使用 golang xml 标准库 160 行代码转换 7000 张图片的标注文件。

rect => <object-class> <x_center> <y_center> <width> <height>

SeaShips 标注文件

最近需要使用 yolov4 训练该数据集，然后输出一个船舶识别数据模型，所以写了一个小小工具来进行标注文件格式转换。转换后的效果可以使用上次写的标注工具查看一下。

在这里插入图片描述

package main

import (
	"encoding/xml"
	"flag"
	"fmt"
	"log"
	"os"
	"strings"
)

// 定义一些 xml 结构和 SeaShips(7000) 数据集对应
type Size struct {
	Width int `xml:"width"`
	Height int `xml:"height"`
	Depth int `xml:"depth"`
}

type BndBox struct {
	Xmin int `xml:"xmin"`
	Ymin int `xml: