ddgrcf / convert_tools

datasets convert tools

数据转化

重要介绍

仅仅对于labelme标注工具产生的文件

本仓库包含以下内容：

1. 在label_adjust.py中能够对错误标签进行转化。
2. 在label_adjust.py中能够对类别数量进行统计。
3. 在rotated_adjust能够将十字标注转化为旋转的进行框，针对已存在标签和未存在标签都适用。

内容列表

• 安装
• 使用说明

安装

创建虚拟环境

$ conda create -n convert python=3.8 -y

安装依赖

$ pip install -r requirement.txt

使用说明

输出每个类的数量(--paths_adjust --suffix tif，进行校正)

$ python labels_adjust.py --dir your dataset file --paths_adjust --suffix tif

将十字标注方式转化为旋转的矩形框(这个针对从头开始标注的)

$ python rotated_adjust.py --dir your dataset file

将十字标注方式转化为旋转的矩形框(这个针对数据集原本就有矩形标注框的)

$ python rotated_adjust.py --dir your dataset file --use-labels --temp-labels null-label

这中标注方式是在原来有矩形标注的对象上标注十字，标注的label就是temp-labels指定的(这里为null-label)，会根据null-label所标注十字所生成的旋转矩形框与原来所有的正矩形框求IOU，把IOU最大的label赋给这个临时的null-label