visdrone验证集标签可视化

最新推荐文章于 2025-12-19 06:58:17 发布

原创

最新推荐文章于 2025-12-19 06:58:17 发布 · 2.9k 阅读

7 ·

CC 4.0 BY-SA版权

文章标签：

#xml

本文分享了一段用于VisDrone数据集可视化的Python代码，该代码能够将数据集中的XML标签转换为图像上的边界框，并附带类别标签，便于进行目标检测任务的评估与调试。

为了方便实验对比结果，先在将visdrone的验证集可视化出来，脚本是git别人的，实测可行，分享给需要的人吧。
代码是根据xml标签可视化的，需要将visdrone的txt标签转成xml，可参考我另一篇文章：

import os
import os.path
import numpy as np
import xml.etree.ElementTree as xmlET
from PIL import Image, ImageDraw

#'1': 'people', '2': 'people','3': 'bicycle', '4': 'car', '5': 'car',
# 6':'others','7':'others','8':'others','9':'others','10': 'motor','11':'others'

classes = ('__background__', # always index 0
           'ignored regions','pedestrian', 'people','bicycle','car','van','truck','tricycle','awning-tricycle',