在目标检测任务中,需要通过Map指标判断模型的精度。为了测试engine文件推理结果的精度,本文介绍了如何使用pycocotools库计算Map,在此之前需要根据coco格式生成json文件。
必须按照coco格式生成json,顺序都要保持一致才行,否则报错不通过。
1.划分val数据集
在做验证时,需要提前划分好val数据集。本文采用labelimg工具画框,因此需要对xml和image文件进行划分。
# coding:utf-8
import os
import random
import argparse
from pathlib import Path
import shutil
def