yoloV5查看anchors
import torch
from models.experimental import attempt_load
model = attempt_load('./weights/yolov5s.pt', map_location=torch.device('cpu'))
m = model.module.model[-1] if hasattr(model, 'module') else model.model[-1]
print(m.anchor_grid)
- 查看anchors
yolov5模型在训练自己的数据集之前会进行autoanchor check,autoanchor check会自动对数据集中的ground truth boxes进行key means聚类,产生新的anchors以提高模型检测的mAP,可以通过以下代码查看新的生成的anchors:
还没有评论,来说两句吧...