u版yolov3查看模型anchors

1.查看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)

2.使用默认anchors

如果不想进行anchors的key means聚类，可以在训练时添加--noautoanchor，使用默认的anchors，默认的anchors如下所示：

# anchors
anchors:
  - [10,13, 16,30, 33,23]  # P3/8
  - [30,61, 62,45, 59,119]  # P4/16
  - [116,90, 156,198, 373,326]  # P5/32

 

 

参考链接：https://blog.csdn.net/linghu8812/article/details/110532712