测试流程

  • 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操作以及分类阈值过滤
  • 保留前几个预测框