问一下大佬，数据集分类的问题

PeterH0323 / Smart_Construction

Base on YOLOv5 Head Person Helmet Detection on Construction Sites，基于目标检测工地安全帽和禁入危险区域识别系统，🚀😆附 YOLOv5 训练自己的数据集超详细教程🚀😆2021.3新增可视化界面❗❗

GNU General Public License v3.0

2.19k stars 464 forks source link

Closed jhyscode closed 3 years ago

jhyscode commented 3 years ago

原始的数据集只有people和hat两个分类，我看readme文件里面的custom_data.yaml的里面分类是三个分类，还有测试的图片也是三个分类，不知道大佬数据集这块是加入了自己标注的还是其他怎么处理的，希望解答一下，万分感谢！

PeterH0323 commented 3 years ago

我自己加了人体，详细看最后一小节，谢谢

jhyscode commented 3 years ago

感谢大佬的答疑和回复！

PeterH0323 commented 3 years ago

谢谢您的认可，如果帮到您请帮我点个星星 🌟 🌟 🌟 谢谢！