2201_75538081的博客

原创 通过pycocotools获取每个类别的COCO指标

save_path = r'C:\Users\wang/' # 指定生成的json文件的存放路径。# 通过gt的json文件和pred的json文件计算map。# 这里 voc 为数据集文件名字，可以改成自己的路径。# 部分同学都用的autodl, 用antodl举例。# 获取类别（单类）对应的所有图片id。# 获取多个类别对应的所有图片的id。# 打印COCO API的总体AP。# 计算每个类别的recall。# xx.json生成之后存放的地址。# 计算每个类别的平均AP。# 打印每个类别的结果。