极道
·
GoFaceRec：基于 Go 的深度学习人脸识别工具
💡
原文中文，约1300字，阅读约需4分钟。
📝
内容提要
GoFaceRec是一个使用Go构建的人脸识别工具，使用MTCNN进行人脸检测和QMagFace进行人脸识别。项目在Ubuntu 20.04上使用Go 1.17进行测试，将QMagFace模型从PyTorch转换为TensorFlow，并使用tfgo进行模型运行。主要挑战是不同类型之间的转换，但在Python中这些转换更简单快速。
🎯
关键要点
-
GoFaceRec是一个用Go构建的人脸识别工具，利用MTCNN进行人脸检测和QMagFace进行人脸识别。
-
该项目旨在将深度学习模型的力量引入Go社区，使用TensorFlow和tfgo进行模型运行。
-
项目在Ubuntu 20.04上使用Go 1.17进行测试，安装的OpenCV版本为4.7。
-
使用ONNX作为中介将QMagFace模型从PyTorch转换为TensorFlow。
-
使用model_converter.py脚本将PyTorch模型转换为ONNX，再转换为TF。
-
提取模型的输入层和输出层名称以便使用tfgo运行模型。
-
项目使用MTCNN进行人脸检测，QMagFace进行人脸识别，分为三个阶段：PNet、RNet、ONet。
-
主要挑战是gocv.Mat、tfgo.Tensor、gonum和Go原生切片之间的转换，Python中这些转换更简单快速。
➡️
