crowdhuman数据集转yolo txt格式

CrowdHuman数据集是一个人群检测数据集，包含了大量的图片和标注信息。如果想要将CrowdHuman数据集转换为YOLO格式的txt文件，可以按照以下步骤进行操作：

1. 安装CrowdHuman数据集的Python API

CrowdHuman数据集提供了Python API，可以方便地读取数据集中的图片和标注信息。首先需要安装该Python API，可以使用以下命令进行安装：

pip install crowdhuman

2. 下载CrowdHuman数据集

在进行转换之前，需要先下载CrowdHuman数据集。可以在CrowdHuman官网上进行下载，也可以使用以下命令进行下载：

wget https://storage.googleapis.com/crowdhumandata/crowdhuman_train.zip

3. 编写转换脚本

接下来需要编写Python脚本，将CrowdHuman数据集中的标注信息转换为YOLO格式的txt文件。可以参考以下代码：

import os
import cv2
import crowdhuman as ch

# CrowdHuman数据集的根目录
root = '/path/to/crowdhuman'

# YOLO格式的txt文件保存路径
save_path = '/path/to/save'

# CrowdHuman数据集中的类别
classes = ['person']

# 遍历CrowdHuman数据集中的所有图片
for img_info in ch.iter_images(root):
    img_path = img_info['file_name']
    img = cv2.imread(img_path)

    # 获取图片的尺寸
    height, width, _ = img.shape

    # 获取图片中的所有标注框
    for ann in img_info['annotations']:
        category = ann['category']

        # 只处理类别为person的标注框
        if category not in classes:
            continue

        bbox = ann['bbox']
        x, y, w, h = bbox

        # 将标注框转换为YOLO格式
        x_center = (x + w / 2) / width
        y_center = (y + h / 2) / height
        x_width = w / width
        y_height = h / height

        # 保存标注框的信息到txt文件中
        save_file = os.path.splitext(os.path.basename(img_path))[0] + '.txt'
        save_file = os.path.join(save_path, save_file)
        with open(save_file, 'a') as f:
            f.write(f'{classes.index(category)} {x_center:.6f} {y_center:.6f} {x_width:.6f} {y_height:.6f}\n')

上述代码中，首先遍历CrowdHuman数据集中的所有图片，然后获取图片中的所有标注框。接着将标注框转换为YOLO格式的坐标，并保存到txt文件中。

4. 运行转换脚本

编写完转换脚本后，就可以运行该脚本，将CrowdHuman数据集中的标注信息转换为YOLO格式的txt文件。可以使用以下命令进行运行：

python convert.py

上述命令中，convert.py是转换脚本的文件名。在运行之前，需要将root和save_path分别替换为CrowdHuman数据集的根目录和YOLO格式的txt文件保存路径。