Hugging Face Blog·· 2022-07-25AI 评分33
使用 TF Serving 在 Hugging Face 部署 TensorFlow 视觉模型
Deploying TensorFlow Vision Models in Hugging Face with TF Serving
AI 导读
Hugging Face 团队展示了如何使用 TensorFlow Serving 在本地部署 TensorFlow 视觉模型,将模型暴露为 REST 或 gRPC 端点。文章以 Vision Transformer 为例,演示了通过 save_pretrained 将模型保存为 SavedModel 格式,并详细说明了如何将图像预处理和后处理逻辑嵌入计算图以优化部署流程。
来源:Hugging Face Blog · huggingface.co