PeterH0323 / Smart_Construction

Base on YOLOv5 Head Person Helmet Detection on Construction Sites,基于目标检测工地安全帽和禁入危险区域识别系统,🚀😆附 YOLOv5 训练自己的数据集超详细教程🚀😆2021.3新增可视化界面❗❗
GNU General Public License v3.0
2.2k stars 464 forks source link

计算anchors #17

Closed zhongwenkun886 closed 3 years ago

zhongwenkun886 commented 3 years ago

您好。你的说明里面需要自己计算新数据的anchors,但YOLOv5训练代码里面不是有自动计算新anchors的功能吗?训练参数 --noautoanchor ,utils/utils.py 里的 check_anchors 函数,这里应该会重新计算model里的 anchors。 不知道我的理解是否正确,谢谢指教。

PeterH0323 commented 3 years ago

代码里面是有的,不需要执行那一个可选步骤