只含person的coco数据集

COCO数据集包含有人的图像，但不是所有图像都只包含人。如果您只想要包含人的图像，可以使用COCO数据集中的标注文件来过滤出只包含人的图像。您可以使用以下步骤来实现：

1. 下载COCO数据集和标注文件。

2. 解析标注文件，找到只包含人的图像的ID。

3. 使用ID过滤图像并将其保存到新的数据集中。

以下是Python代码示例：

import json
import os
from shutil import copyfile

# COCO数据集文件夹路径
coco_folder = "/path/to/coco"

# COCO标注文件路径
ann_file = os.path.join(coco_folder, "annotations/person_keypoints_train2017.json")

# 新数据集保存路径
new_folder = "/path/to/new_dataset"

# 如果新数据集文件夹不存在，则创建它
if not os.path.exists(new_folder):
    os.makedirs(new_folder)

# 解析标注文件
with open(ann_file, "r") as f:
    ann_data = json.load(f)

# 找到只包含人的图像的ID
person_images = []
for ann in ann_data["annotations"]:
    if ann["category_id"] == 1:  # 1表示人的类别ID
        person_images.append(ann["image_id"])

person_images = list(set(person_images))  # 去重

# 将只包含人的图像复制到新数据集中
for img in ann_data["images"]:
    if img["id"] in person_images:
        src_path = os.path.join(coco_folder, "train2017", img["file_name"])
        dst_path = os.path.join(new_folder, img["file_name"])
        copyfile(src_path, dst_path)

此代码将从COCO训练集中筛选出只包含人的图像，并将它们复制到新的数据集中。请注意，此代码仅复制图像文件，而不包括标注文件。您可能需要修改此代码来复制标注文件以及其他所需文件。