Advertisement

086 基于卷积神经网络的安全带的佩戴检测

阅读量:

关于演示资源的获取方式,请参考086期的相关内容:

到此一游7758258的个人空间_哔哩哔哩_bilibili

以下为效果展示示意图:

以下为代码文件的展示内容:

执行01数据集文本生成制作.py脚本后,可将图片路径信息存储至txt文本文件中。

通过运行02train.py脚本,能够从txt文本中读取图片路径信息,并用于模型训练过程。

在02train.py中,用户可选择十余种不同的模型进行训练与对比,涵盖alexnet、DenseNet、DLA、GoogleNet、Mobilenet、ResNet、ResNeXt、ShuffleNet、VGG、EfficientNet以及Swin transformer等多种主流网络结构。

![](ht

全部评论 (0)

还没有任何评论哟~