如何从 py torch 转换为 tflite

问题描述 投票:0回答:1

我一直在使用yolov8s模型,并从google colab生成了best.pt模型,但我现在在google colab中遇到了问题,所以,我想在vscode上将pt模型转换为tensorflow lite。 有什么想法吗?

python tensorflow pytorch onnx yolov8
1个回答
0
投票

Google 有一个新的 PyTorch 到 TFLite 转换库:https://github.com/google-ai-edge/ai-edge-torch

© www.soinside.com 2019 - 2024. All rights reserved.