BDD100K数据集json格式转txt格式

最新推荐文章于 2024-08-04 22:19:56 发布

原创

最新推荐文章于 2024-08-04 22:19:56 发布 · 596 阅读

3 ·

CC 4.0 BY-SA版权

文章标签：

#json #BDD #python #txt

文章提供了一段Python代码，用于将BDD数据集的新格式json文件转换为适用于YOLOv5的txt格式。代码处理了每个图片框注释的转换，包括类别归一化和尺寸调整。主要针对的类别包括person、rider、car等。

BDD数据集地址：Profilehttps://bdd-data.berkeley.edu/portal.html#download

目前网上能找到的BDD数据集json格式转txt格式的代码和目前BDD提供的json文件的格式有所不同。之前提供的json是每张图片一个json文件，现在是将数据集中所有的分割信息放到了一个json文件中了，所以查了很多都不适用，所以自己动手写了一个，供大家参考。

所适用的json格式在下面，能看懂吧

{
    "frames":[
        {
            "name":xxx,
            "url":xxx,
            ...
            "labels":[
                {
                    "id":xxx,
                    "index":xxx,
                    "category":xxx,
                    "box2d":{
                    "x1": 447.0,
                    "y1": 392.0,
                    "x2": 524.0,
                    "y2