[Python]Cityscapes目标检测标注转YOLO格式

最新推荐文章于 2024-08-14 23:17:29 发布

身披白袍

最新推荐文章于 2024-08-14 23:17:29 发布

阅读量3.4k

点赞数 6

分类专栏： Python|Matlab 机器学习|数据挖掘|数学模型文章标签： python 深度学习 yolo 目标检测

本文链接：https://blog.csdn.net/Shenpibaipao/article/details/111240711

版权

该博客介绍了如何将Cityscapes数据集的目标检测标注转化为适用于YOLO（yolo3和yolov5的pytorch版本）的格式。作者提供了数据集下载链接，并强调了在处理过程中应避免使用含有中文路径的Linux环境。

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

文件目录

数据集下载：https://www.cityscapes-dataset.com/downloads/
在这里插入图片描述
下载 leftImg8bit_trainvaltest.zip 和 gtFine_trainvaltest.zip ，解压后放成如下的格式：

─ {root}
── getfine
── leftImg8bit
── 处理脚本.py

执行:

python 处理脚本.py

生成：

─ {root}
── getfine
── leftImg8bit
── 处理脚本.py
──  labels  # 存放框图的标签
──  images # yolov3+版本的脚本将会拷贝输出图片到这个位置
──  yolo_train.txt  # 训练样本的路径
──  yolo_val.txt # 验证样本的路径
──  yolo_test.txt # 测试样本的路径

尽量用linux，路径中不要有中文。
其它的看看下文的代码就好了，注释够详细了
yolo3、yolov5的pytorch版本，请用这个代码。

处理脚本.py

import json
import os
from pathlib import Path
import re
from tqdm import tqdm
from functools import reduce


def convert_annotation(image_id, paths):
    global label_map

    def find_box(points):  # 该函数用来找出xmin, xmax, ymin ,ymax 即bbox包围框
        _x, _y = [float(pot[0]) for pot in points], [float(pot[1]) for pot in points]
        return min(_x), max(_x), min(_y), max(_y)

    def<