visdrone+yolov5

最新推荐文章于 2025-03-08 23:55:34 发布

小灰啾

最新推荐文章于 2025-03-08 23:55:34 发布

阅读量5k

点赞数 8

分类专栏： yolo

本文链接：https://blog.youkuaiyun.com/qq_42932308/article/details/115217935

版权

本文介绍了如何将VisDrone2019数据集转换为YOLOv5格式，并详细阐述了转换步骤，包括数据集下载、转换工具的使用、YOLOv5环境的搭建、训练配置的修改以及训练过程。通过训练，可以在VisDrone数据集上应用YOLOv5进行目标检测。

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

1.数据集转换
VisDrone2019-DET-train
下载地址

http://aiskyeye.com/download/object-detection/

解压后只有images、annotations两个文件夹
使用数据集转换工具visdrone2yolo.py

只要修改root_dir就可以直接运行。运行前先装一下pip包。如果没有labels文件夹生成就手动新建个labels文件夹，在root_dir下。

import os
from os import getcwd
from PIL import Image
import xml.etree.ElementTree as ET
import random

#root_dir = "train/"
root_dir = "/home/lhw/Gradute/VirDrone/yolov5/VisDrone2019-DET-train/"
annotations_dir = root_dir+"annotations/"
image_dir = root_dir + "images/"
label_dir = root_dir + "labels/"
# label_dir = root_dir + "images/"    # yolo里面要和图片放到一起
xml_dir = root_dir+"annotations_voc/"  #注意新建文件夹。后续改一下名字，运行完成之后annotations这个文件夹就不需要了。把annotations_命名为annotations
data_split_dir = root_dir + "train_namelist/"

sets = ['train', 'test','val']
class_name = ['ignored regions', 'pedestrian','people','bicycle','car', 'van', 'truck', 'tricycle','awning-tricycle', 'bus','motor','others']


def visdrone2voc(annotations_dir, image_dir, xml_dir):
    for filename in os.listdir(annotations_dir):
        fin = open(annotations_dir + filename, 'r')
        image_name = filename.split('.')[0]
        img = Image.open(image_dir + image_name + ".jpg")
        xml_name = xml_dir + image_name + '.xml'
        with open(xml_name, 'w') as fout:
            fout.write('<annotation>' + '\n')

            fout.write('\t' + '<folder>VOC2007</folder>' + '\n')
            fout.write('\t' + '<filename>' + image_name + '.jpg' + '</filename>' + '\n')

            fout.write('\t' + '<source>' + '\n')
            fout.write('\t\t' + '<database>' + 'VisDrone2018 Database' + '</database>' + '\n')
            fout.write('\t\t' + '<annotation>' + 'VisDrone2018' + '</annotation>' + '\n')
            fout.write('\t\t' + '<image>' + 'flickr' + '</image>' + '\n')
            fout.write('\t\t' + '<flickrid>' + 'Unspecified' + '</flickrid>' + '\n')
            fout.write('\t' + '</source>' + '\n')

            fout.write('\t' + '<owner>' +

最低0.47元/天解锁文章