写在前面:本文不详细介绍SSD,只记录一些大体流程,具体细节可见参考1,换用数据集训练SSD可见参考2
参考1:SSD代码细节讲解
参考2:SSD换数据集训练
给定图片如何得到对应的GT
SSD网络若指定了先验框的minsize和maxsize,则整个图片的所有先验框位置均固定。拿300SSD来说,可以得到8372个位置确定的先验框(存于model_data/prior_boxes_ssd300.pkl)
图片所标注的物体位置(真实框)和所属类别(存于2007_train.txt)
计算真实框对应的先验框,与这个先验框应当有的预测结果。
【输入:8372先验框和图片标注的信息,输出:assig
1