search

bubbliiiing / yolo3-pytorch

这是一个yolo3-pytorch的源码，可以用于训练自己的模型。
MIT License
2.01k stars 585 forks source link

关于模型输出的问题 #191

Closed mikawudi closed 6 months ago

mikawudi commented 6 months ago

make_last_layers中的最后一层作为输出,我看网上标准的yolov3,每一个尺度的预测是 (grid cell的数量)(grid cell的数量)(锚框数)(分类数+5),但是看到您这边代码的最后输出是(锚框数)(分类数+5)撒?每个尺度的特征图只输出3个预测么?按标准来说不是每个grid cell负责3个预测然后模型输出后再遍历输出,根据置信度阈值过滤出预测结果撒?

mikawudi commented 6 months ago

跑了下代码看懂了....最后的一个cov2是在每一个尺度上的每一个grid cell上都执行的....