目标检测 YOLOv5 - 模型的输出

目标检测 YOLOv5 - 模型的输出

flyfish

版本: YOLOv5:v5

模型的输出

 y = model(img) 
1

y就是输出，y的样子如下

torch.Size([1, 3, 80, 80, 85])
torch.Size([1, 3, 40, 40, 85])
torch.Size([1, 3, 20, 20, 85])
1
2
3

按照stride来划分，以640 × 640像素图像为例
stride分别是8，16，32

640 /  8 = 80，这层网格大小是80 × 80
640 / 16 = 40，这层网格大小是40 × 40
640 / 32 = 20，这层网格大小是20 × 20
1
2
3

B是每个单元格的预测框的数量
C是类别的数量
这里通常是B=3，C=80（COCO dataset）
输出有3块，分别是output,405,419

output

405

419