测试流程¶
py/ssd/test.py
测试参数¶
- 数据集:
PASCAL VOC 07 test
- 批量大小:
10
- 目标检测:
- 中心坐标的标准方差(
CENTER_VARIANCE
):0.1
- 长宽的标准方差(
SIZE_VARIANCE
):0.2
- 中心坐标的标准方差(
NMS
阈值(NMS_THRESHOLD
):0.5
- 分类概率阈值(
CONFIDENCE_THRESHOLD
):0.01
- 保留预测框个数(
MAX_PER_IMAGE
):100
测试流程¶
- 载入测试图像
- 图像预处理:缩放到指定大小,减去图像均值,转换成
Tensor
格式 - 模型计算:生成检测框偏移和检测框类别置信度
- 结合先验框、检测框偏移以及目标检测参数得到检测框
- 对逐个图像执行
NMS
操作以及分类阈值过滤 - 保留前几个预测框