yoloV5查看anchors

最新推荐文章于 2024-08-22 09:21:39 发布

原创最新推荐文章于 2024-08-22 09:21:39 发布 · 1.6k 阅读

9 ·

CC 4.0 BY-SA版权

文章标签：

#pytorch

目标检测专栏收录该内容

54 篇文章

订阅专栏

这篇博客介绍了如何利用PyTorch加载预训练的YOLOv5模型，并展示了如何检查模型中生成的Anchors，这些Anchors是通过autoanchorcheck对数据集的groundtruth boxes进行聚类得到，旨在提升模型的检测精度。

部署运行你感兴趣的模型镜像

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)

1. 查看anchors
yolov5模型在训练自己的数据集之前会进行autoanchor check，autoanchor check会自动对数据集中的ground truth boxes进行key means聚类，产生新的anchors以提高模型检测的mAP，可以通过以下代码查看新的生成的anchors：

您可能感兴趣的与本文相关的镜像