yoloV5查看anchors

最新推荐文章于 2023-09-22 19:04:43 发布

老三是只猫

最新推荐文章于 2023-09-22 19:04:43 发布

阅读量1.3k

点赞数 3

分类专栏：目标检测文章标签： pytorch

本文链接：https://blog.csdn.net/zhonglongshen/article/details/120591344

版权

目标检测专栏收录该内容

54 篇文章 3 订阅

订阅专栏

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

摘要由CSDN通过智能技术生成

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：

老三是只猫

关注

3
点赞
踩
6

收藏

觉得还不错? 一键收藏
2
评论
yoloV5查看anchors

import torchfrom models.experimental import attempt_loadmodel = 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. 查看anchorsyolo.
复制链接

扫一扫

专栏目录