一、项目背景
打火机是点香烟不可或缺的工具,在日常生活中常常可以看见它的身影。国内疫情反复,高铁站、地铁等公共交通场所实施严格的安检,该项目如果落地,则从一定程度上提高了安检效率,减少了人力消耗,同时也减少了安检场所的人员聚集现象,有助于疫情防控。对于中小学来说,它助长了香烟对青少年的毒害。对于一些违禁打火机的地方(机场,动车,景区)来说,它威胁了公众的安全及民众的利益,比如在森林景区中,是严禁携带打火机的,一旦旅客使用打火机,那么就有可能造成火灾,甚至大规模的伤亡。
二、数据集简介
本项目数据集来源于其他AI社区公开数据集。注意:该数据集中没有划分预测集,且两个划分数据集的split files中‘/’都是反的,直接使用会找不到路径报错,所以我们需要重新划分数据集。[公共场所安检打火机及危险物品检测_成都锦城学院_畅想一队 - 飞桨AI Studio (baidu.com)]
三、训练模型
(1)模型选择:本项目使用PaddleX套件yolov3_mobilenetv1模型进行训练
(2)调参代码:
num_epochs=120,
train_dataset=train_dataset,
train_batch_size=8,
eval_dataset=eval_dataset,
learning_rate=0.0125,
lr_decay_epochs=[210, 240],
save_dir='output/yolov3_mobilenetv1')
本项目还有很多需要改进之处,比如检测对象不限于打火机,还可以有其他的危险物品;比如模型参数、模型部署还有很多可以调整完善的地方等等。