BDD100KtoYOLOv5.txt格式（数据集转换脚本）

最新推荐文章于 2024-06-11 21:56:33 发布

原创

最新推荐文章于 2024-06-11 21:56:33 发布

· 2.7k 阅读

39 ·

版权

文章标签：

#YOLOv5 #BDD100K #目标检测 #json2txt

这篇博客分享了一个将BDD100K数据集的JSON标签格式转换为YOLOv5所需的TXT格式的脚本，解决了现有转换脚本不适用或存在错误的问题，适合新手使用。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

目前网上有许多各大数据集之间的转换脚本，比如BDDtoCOCO、XtoKITTI、XtoVOC等等，但是对于YOLOv5的.txt标签格式，若想使用BDD100K数据集去训练，需要将BDD100K的json标签格式转为.txt格式，现有的一些转换脚本Demo绝大多数都不能直接被拿过来使用，要不就是得经历几次来回转换、要不就是demo存在问题，这对于我们这类新手很不友好，so…花了点时间写了个小demo，在此分享给各位！（自用）

关键词：BDD100K.json to YOLOv5.txt

有问题还请各位多多指正！

import json
import os

def bdd2yolo5(categorys,jsonFile,writepath):
    strs=""
    f = open(jsonFile)
    info = json.load(f)
    #print(len(info))
    #print(info["name"])
    write = open(writepath + "%s.txt" % info["name"], 'w')
    for obj in info["frames"]:
        #print(obj["objects"])
        for