文件名称:InsightFace-REST:InsightFace REST API 用于在 Docker 中使用 TensorRT 轻松部署人脸识别服务
文件大小:2.2MB
文件格式:ZIP
更新时间:2024-08-24 08:27:13
docker gpu face-recognition face-detection fp16
InsightFace-REST 该存储库旨在为 InsightFace 人脸检测和识别管道提供方便、易于部署和可扩展的 REST API,使用 FastAPI 进行服务,使用 NVIDIA TensorRT 进行优化推理。 代码主要基于官方 DeepInsight InsightFaceAPI。 该存储库提供用于构建人脸识别 REST API 和使用 Docker 将模型转换为 ONNX 和 TensorRT 的源代码。 主要特征: 准备好使用 Docker 和 nvidia-docker2 在支持 NVIDIA GPU 的系统上进行部署。 启动时自动下载模型(使用 Google Drive)。 借助 TensorRT 优化、FP16 推理和使用 ArcFace 模型对检测到的人脸进行批量推理,性能比 MXNet 推理提高多达 3 倍。 支持较旧的 Retinaface 探
【文件预览】:
InsightFace-REST-master
----sample.py(3KB)
----src()
--------Dockerfile_trt(625B)
--------api_trt()
--------converters()
--------Dockerfile_cpu(687B)
--------Dockerfile.converter(501B)
----deploy_trt.sh(4KB)
----LICENSE(11KB)
----README.md(14KB)
----deploy_converter.sh(403B)
----misc()
--------images()
----.gitignore(245B)
----deploy_cpu.sh(3KB)