人脸_面具_检测_CNN
基于计算机视觉和深度学习的OpenCV和Pytorch面罩检测系统
数据集
下载数据集: :
训练
训练ResNet50模型:在train_resnet50.ipynb中,选择data_path(安装数据集的位置)和model_dir(应保存训练后的模型的位置)。执行train_resnet50.ipynb
训练MobileNetV2模型:在train_mobilenetv2.ipynb中,选择data_path(安装数据集的位置)和model_dir(应保存经过训练的模型的位置)。执行train_mobilenetv2.ipynb
测验
打开test.ipynb
在导入中:从model_resnet50导入模型中选择以测试ResNet50模型,从model_mobilenetv2导入模型中选择以测试MobileNetV2模型。
选择一个model_p
1