Tract是Sonos开发的一款神经网络推理工具包。其主要功能是读取ONNX和NNEF格式的神经网络,进行优化并运行推理。该项目最初名为tfdeploy,或Tensorflow-deploy-rust。
Tract支持多种神经网络格式,包括ONNX和NNEF。用户可以通过Tract对这些格式进行加载、优化和运行,支持的网络示例有:
Tract在ONNX后端测试中约85%的测试均成功通过,尤其在一些实际应用集成测试中表现优异,例如:bvlc_alexnet、densenet121、inception_v1等。不过,一些与张量序列和可选张量相关的运算符仍未得到支持,这一限制构成了Tract的架构特征。这些功能的支持可能会影响性能,因此暂未列入计划。
NNEF(神经网络交换格式)提供了一种面向生产环境的网络描述格式。Tract支持NNEF,并且能够执行大多数NNEF规范中的操作。不过,像ROI操作符这样的少数特性目前还未被支持。此外,Tract引入了称为tract-OPL的NNEF扩展,以便支持额外的运算符。
即便Tract目前尚无法支持所有任意的TensorFlow模型,它仍能够运行一些重要的模型如Google Inception v3和Snips唤醒词模型。Tract甚至支持将TensorFlow 2模型转换为ONNX格式以便加载和运行。
Tract性能的演进可通过其对一些网络模型的持续测试来监控,诸如:
Tract项目在Apache 2.0和MIT双许可协议下发布,用 户可以选择任意一个进行使用。项目中的一些文件目录是从TensorFlow和ONNX项目中复制而来,并不受此许可证的约束。
Tract项目欢迎各方贡献,所有贡献默认在Apache-2.0许可证下进行双许可,遵循项目的整体开源精神和约定。目前,Tract在支持更多运算符和网络格式方面仍在不断努力和拓展。
AI辅助编程,代码自动修复
Trae是一种自适应的集成开发环境(IDE),通过自动化和多元协作改变开发流程。利用Trae,团队能够更快速、精确地编写和部署代码,从而提高编程效率和项目交付速度。Trae具备上下文感知和代码自动完成功能,是提升开发效率的理想工具。
AI小说写作 助手,一站式润色、改写、扩写
蛙蛙写作—国内先进的AI写作平台,涵盖小说、学术、社交媒体等多场景。提供续写、改写、润色等功能,助力创作者高效优化写作流程。界面简洁,功能全面,适合各类写作者提升内容品质和工作效率。
全能AI智能助手,随时解答生活与工作的多样问题
问小白,由元石科技研发的AI智能助手,快速准确地解答各种生活和工作问题,包括但不限于搜索、规划和社交互动,帮助用户在日常生活中提高效率,轻松管理个人事务。