利用pycocotools库计算MAP:生成coco格式 json文件数据集和计算map值

文章目录

    • 1.划分val数据集
    • 2. xml to txt
    • 3. coco格式json文件
    • 4. 生成coco格式json文件
    • 5.使用pycocotools计算map
    • 6. 讨论

在目标检测任务中,需要通过Map指标判断模型的精度。为了测试engine文件推理结果的精度,本文介绍了如何使用pycocotools库计算Map,在此之前需要根据coco格式生成json文件。
必须按照coco格式生成json,顺序都要保持一致才行,否则报错不通过。

1.划分val数据集

在做验证时,需要提前划分好val数据集。本文采用labelimg工具画框,因此需要对xml和image文件进行划分。

# coding:utf-8

import os
import random
import argparse

你可能感兴趣的:(深度学习,json,python)