ssd300模型训练代码.py

时间:2022-08-02 04:42:37
【文件属性】:
文件名称:ssd300模型训练代码.py
文件大小:18KB
文件格式:PY
更新时间:2022-08-02 04:42:37
目标检测 SSD训练过程中的关键点是如何将ground truth信息和网络预测输出的信息匹配上,一旦信息匹配上就能端到端的训练了(这里可以理解为预测值和GT值为loss function的两个输入,需要进行相应值的转换才才能符合loss function的输入标准)。 匹配策略: 在SSD的训练过程中,当预测bbox和任一gt bbox的jaccard overlap(IOU值)大于0.5时都被当做正样本(由于图像中GT很少,目的是防止负样本过多这个样本过少,学习难度大),其实即使是这样训练过程中由于ground truth太少,正负样本还是极度不平衡; 训练目标(loss函数):

网友评论