YOLO模型常见输出参数解惑

根据不同输出参数编写后处理,否则会报错

第一种

  • Concatoutput_dim_0 :变量,表示预测目标的数量,
  • 7:表示每个目标的七个参数:batch_id,x0,y0,x1,y1,cls_id,score

YOLO模型常见输出参数解惑_第1张图片

第二种

  • 85:每一行85个数值,5个center_x,center_y, width, height,score ,80个标签类别得分
  • 25200:三个尺度上的预测框总和 ( 80∗80∗3 + 40∗40∗3 + 20∗20∗3 ),每个网格三个预测框,后续需要非极大值抑制NMS处理
  • 1:没有批量预测推理,即每次输入推理一张图片

YOLO模型常见输出参数解惑_第2张图片

第三种

YOLO模型常见输出参数解惑_第3张图片

第四种

YOLO模型常见输出参数解惑_第4张图片

资料

YOLO模型常见输出参数解惑_第5张图片

你可能感兴趣的:(YOLO,深度学习,人工智能)