查看自己模型的anchor

一.用yolov5官方的训练代码可能会根据自己的数据集生成新的anchor，并没有用到配置文件中的anchor，如果不清楚这个问题，就会给后续的工作带来一些麻烦

下面是查看自己训练好的模型中的anchor：

import torch
import sys
weights = 'runs/train/test/weights/best.pt'
model = torch.load(str(weights[0] if isinstance(weights, list) else weights), map_location='cpu')
model1 = model['ema' if model.get('ema') else 'model']
model2 = model1.float().fuse().model.state_dict()

for k,v in model2.items():
    if 'anchor' in k:
        # print(k)
        # print(v)
        print(v.numpy().flatten().tolist())

这样你就可以直接查看自己模型的anchor了。