上传者: 42110362
|
上传时间: 2022-03-30 22:36:43
|
文件大小: 33.81MB
|
文件类型: -
基于MobileFacenet的Coral EdgeTPU支持的人脸识别
还提供了带有Tensorflow-2,EdgeTPU模型的Mobilefacenet,用于在Coral EdgeTPU上运行模型
介绍
来自Tensorflow 2版本的mobilefacenet
演示版
在带有Coral TPU的台式机上运行60 fps,在树莓派上运行约24 fps
用法
数据集
使用与使用的数据集相同的数据集进行训练。 CASIA用于训练,LFW用于测试。
训练
在train.py中更改指向图像数据集的目录。 我通过将RESUME设置为False直接使用ArcFace训练了模型,但是值得尝试使用softmax损失进行预训练
我添加了一个示例,以添加额外的标题以使用生成的嵌入来执行分类,在这里,我使用生成的嵌入来预测人是否戴着口罩。 您可以通过使用另一个数据集来获得更多乐趣
结果
经过训练的模型