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

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

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

最近需要使用 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:"ymin"`
	Xmax