将COCO2017标注文件中的bbox显示在图片上

最新推荐文章于 2024-08-05 12:55:50 发布

努力奋斗-不断进化

最新推荐文章于 2024-08-05 12:55:50 发布

阅读量1w

点赞数 5

分类专栏： Python 文章标签： coco bbox 显示

本文链接：https://blog.csdn.net/zhuoyuezai/article/details/84315113

版权

该博客介绍了如何将COCO2017数据集的bbox边界框显示到对应的图片上，提到了imageidFile的数据格式要求，并展示了实现效果。

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

from pycocotools.coco import COCO
import cv2
import pandas as pd


def showNimages(imageidFile, annFile, imageFile, resultFile):
    """
    :param imageidFile: 要查看的图片imageid，存储一列在csv文件里 （目前设计的imageid需要为6位数，如果少于6位数，可以在前面加多个0）
    :param annFile:使用的标注文件
    :param imageFile:要读取的image所在文件夹
    :param resultFile:画了标注之后的image存储文件夹
    :return:
    """
    data = pd.read_csv(imageidFile)
    list = data.values.tolist()
    image_id = []  # 存储的是要提取图片id
    for i in range(len(list)):
        image_id.append(list[i][0])
    print(image_id)
    print(len(image_id))
    coco = COCO(annFile)

    for i in range(len(image_id)):
        image = cv2.imread(imageFile + '000000' + str(image_id[i]) + '.jpg')
        annIds = coco.getAnnIds(imgIds=image_id[i], iscrowd=None)
        anns = coco.loadAnns(annIds)
        for n in