#4 · 主分类: 模型动物园与中心
PINTO_model_zoo
一个用于存储已在各种框架之间相互转换的模型的仓库。支持的框架包括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补充。
相关文章
暂无针对该项目的文章。
如需补充新闻资讯、使用教程、应用测评等内容,欢迎 联系我们推荐选题或投稿。