coco数据集中筛选person

在COCO数据集中，如果你只想要与人相关的图像，你可以使用以下步骤来筛选出包含person类别的图像：

1. 下载COCO数据集并安装COCO API。
2. 使用COCO API中的`pycocotools`模块加载COCO标注文件。
3. 获取标注文件中的所有图像ID和类别ID。
4. 获取person类别的ID。
5. 遍历所有图像ID，使用`loadImgs`函数加载图像信息，并使用`loadAnns`函数获取该图像中所有的标注信息。
6. 判断标注信息中是否包含person类别，如果包含则将该图像的ID加入到筛选结果中。

下面是一个示例代码：

from pycocotools.coco import COCO

# 初始化COCO API
dataDir = '/path/to/coco/dataset'
dataType = 'train2017' # 数据集类型，可以是train2017、val2017等
annFile = '{}/annotations/instances_{}.json'.format(dataDir,dataType)
coco = COCO(annFile)

# 获取person类别的ID
catIds = coco.getCatIds(catNms=['person'])

# 获取所有图像ID和类别ID
imgIds = coco.getImgIds()

# 筛选包含person类别的图像
personImgIds = []
for imgId in imgIds:
    img = coco.loadImgs(imgId)[0]
    annIds = coco.getAnnIds(imgIds=img['id'])
    anns = coco.loadAnns(annIds)
    for ann in anns:
        if ann['category_id'] in catIds:
            personImgIds.append(img['id'])
            break

这样就可以得到所有包含person类别的图像ID了。你可以进一步使用这些图像ID来获取图像并进行后续处理。