智慧工地项目:安全帽是工人在施工现场的重要防护工具,但许多工人因安全帽缺乏舒适感而选择不佩戴,这将危及工人的生命安全。因此实时监控工人是否正确佩戴安全帽十分重要,但工地上作业环境危险,不适合用人力迚行实时监控,所以考虑将机器视觉应用其中,代替人力迚行安全帽佩戴检测。这可以一定程度上预防安全事故的収生,保证工人的安全。
0.
环境配置
环境配置部分,还是我们的老朋友pycharm和anaconda,使用pytorch1.10.0。
1.
数据集搜集
数据集方面通过爬虫来进行搜集,搜集并用labelme标注住2000张高质量数据。
2.
模型训练
使用主播写好的go_train.py进行训练,包含4组不同规格的yolov5模型,方便快捷。
3.
模型测试
模型测试部分我们主要使用目标检测的评价指标mAP,高达87.5!
4.
系统构建
最后使用pyqt5封装我们的算法,有模型切换,图片检测和视频检测功能,方便快捷。
以上需要的小伙伴自取:(粉丝儿仅需9.9,包含4组模型+数据集+图形化界面系统)