#4 · 主分类: 模型动物园与中心

PINTO_model_zoo

caffe computer-vision coreml edgetpu keras mediapipe model model-zoo models onnx openvino pretrained-models pytorch tensorflow tensorflow-lite tensorflowjs tf-trt tfjs tflite tflite-models

一个用于存储已在各种框架之间相互转换的模型的仓库。支持的框架包括TensorFlow、PyTorch、ONNX、OpenVINO、TFJS、TFTRT、TensorFlowLite(Float32/16/INT8)、EdgeTPU、CoreML。

项目最后更新:08/19/26

GitHub Stars

4.6K

Forks数量

668

贡献者数量

19

许可证

MIT

收录理由

找过视觉模型的人大概都有这种经历:好不容易相中一个网络,却发现自己要的是TensorFlow Lite的EdgeTPU版本,手头只有PyTorch权重;或者想在Intel神经计算棒上跑,却找不到对应的OpenVINO格式。这个仓库把同一批模型的多种转换版本集中放在一起,从EfficientNet、MobileNet到姿态估计和分割模型,覆盖TensorFlow、PyTorch、ONNX、OpenVINO、TensorFlow.js、TF-TRT、CoreML以及量化后的TensorFlow Lite(Float32/16/INT8)。对边缘设备来说,现成的INT8和EdgeTPU构建尤其省事,省掉了自己动手转换和校准的环节。每个模型文件夹里都附带了对应的许可证说明和转换脚本,不放心的话可以自己重新生成一遍,而不是盲信现成文件。示例代码在部分模型上有所欠缺,这一点作者也坦率承认,并且欢迎志愿者通过pull request补充。

相关文章

暂无针对该项目的文章。

如需补充新闻资讯、使用教程、应用测评等内容,欢迎 联系我们推荐选题或投稿。

同类热门项目