yoloV5查看anchors

浅浅的花香味﹌ 2022-08-28 10:51 241阅读 0赞
  1. import torch
  2. from models.experimental import attempt_load
  3. model = attempt_load('./weights/yolov5s.pt', map_location=torch.device('cpu'))
  4. m = model.module.model[-1] if hasattr(model, 'module') else model.model[-1]
  5. print(m.anchor_grid)
  1. 查看anchors
    yolov5模型在训练自己的数据集之前会进行autoanchor check,autoanchor check会自动对数据集中的ground truth boxes进行key means聚类,产生新的anchors以提高模型检测的mAP,可以通过以下代码查看新的生成的anchors:

发表评论

表情:
评论列表 (有 0 条评论,241人围观)

还没有评论,来说两句吧...

相关阅读

    相关 YOLOV5训练

    简介 `YOLOv4`还没有退热,`YOLOv5`就已经来了! 6月9日,`Ultralytics`公司开源了`YOLOv5`,离上一次`YOLOv4`发布不到50天,不过