Visdrone数据集转YOLO格式

最新推荐文章于 2024-07-09 16:38:52 发布

YOLO歪瓜裂枣

最新推荐文章于 2024-07-09 16:38:52 发布

阅读量668

点赞数 9

分类专栏：公开YOLO改进文章标签： python 开发语言

本文链接：https://blog.csdn.net/u012305034/article/details/134655981

版权

本文介绍了一个Python脚本，用于将从Github下载的VisDrone数据集从VisDrone格式转换为YOLO所需的xywh格式，以便于模型训练。脚本包括读取VisDrone标注文件、转换坐标并保存到新标签文件中。

摘要由CSDN通过智能技术生成

从Github上下载的Visdrone数据集格式不是YOLO的，可以使用下面脚本做转换。

import os
from pathlib import Path

from ultralytics.utils.downloads import download

def visdrone2yolo(dir_path):
    dir = Path(dir_path)
    from PIL import Image
    from tqdm import tqdm

    def convert_box(size, box):
        # Convert VisDrone box to YOLO xywh box
        dw = 1. / size[0]
        dh = 1. / size[1]
        return (box[0] + box[2] / 2) * dw, (box[1] + box[3] / 2) * dh, box[2] * dw, box[3] * dh

    (dir / 'labels').mkdir(parents=True, exist_ok=True)  # make labels directory
    pbar = tqdm((dir / 'annotations').glob('*.txt'), desc=f'Converting {dir}')
    for f in pbar:
        img_size = Image.open((dir / 'images' / f.name).with_suffix('.jpg')).size
        lines = []
        with open(f, 'r') as file:  # read annotation.txt
            for row in [x.split(',')