search

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

安全帽颜色检测 #37

Closed Zpadger closed 3 years ago

Zpadger commented 3 years ago

小白请问大佬，如果我想对安全帽的颜色进行检测（红、黄、白、蓝），除了要改变classes，网络的结构需要作何改变吗？

`#训练集和验证集的 labels 和 image 文件的位置 train: ./score/images/train val: ./score/images/val

number of classes

nc: 6

class names

names: ['person', 'no_helmet', 'red_helmet', 'yellow_helmet', 'blue_helmet', 'white_helmet']`

PeterH0323 commented 3 years ago

只要改 classes 就行了，网络不要动

PeterH0323 commented 3 years ago

因长时间为回复，先关闭本 issue 了，如果后续有任何疑问，欢迎继续提交 issue ，感谢您的关注

Zpadger commented 3 years ago

感谢大佬指点(这几天网络不好上不了github)