TNN

TNN

轻量级、高效能、多平台支持的开源深度学习框架

TNN,腾讯优图实验室开源的神经网络推理框架,提供针对移动设备和X86/NV GPUs的高效性能优化。该框架已被QQ、微视等多款应用使用,并支持各大平台包括TensorFlow、Pytorch、MxNet。

TNN人工智能模型转换跨平台性能优化Github开源项目

TNN 项目介绍

TNN 是腾讯优图实验室开源的一个高性能、轻量级的神经网络推理框架。它不仅具有跨平台支持、高性能表现,还有模型压缩、代码裁剪等很多优秀的特性。TNN 在原有 Rapidnet 和 ncnn 框架的基础上,进一步强化了对移动设备的支持和性能优化。同时,TNN 借鉴了业内主流开源框架的高性能和良好扩展性,扩展了对 X86 和 NV GPU 的支持。目前,TNN 已被应用于手机 QQ、微视和 P 图等多款移动应用中,作为腾讯云 AI 的基础加速框架,TNN 也为众多业务提供了加速支持。我们欢迎大家参与到 TNN 的协作建设中,推动 TNN 推理框架的进一步完善。

效果示例

TNN 框架支持多种神经网络模型,并提供了多个应用场景的视觉效果示例,包括:

  • 人脸检测(BlazeFace):通过简单高效的模型进行快速人脸定位。
  • 人脸对齐:提供面部特征点检测,支持精准的人脸解析。
  • 头发分割:用于图像中头发区域的检测与分割。
  • 姿态估计(来自腾讯广流实验室和 BlazePose 模型):提供人体姿态的识别和定位。
  • 中文 OCR:实现轻量化中文文本识别,支持多种方位的文本解析。
  • 目标检测(YOLOv5s 和 MobilenetV2-SSD):用于快速、多目标的识别与定位。
  • 阅读理解:支持自然语言处理中的阅读理解任务。

快速入门

使用 TNN 非常简单,即使大家已经拥有了训练完成的模型,也可以通过以下三个步骤将模型部署到目标平台上。

  1. 模型转换:将训练好的模型转换为 TNN 模型,TNN 提供了丰富的工具,能够支持 Tensorflow、Pytorch 或 Caffe 的无缝转换。
  2. 编译 TNN 引擎:根据目标平台选择合适的加速方案进行编译,如 ARM/OpenCL/Metal/NPU/X86/CUDA 等。
  3. 推理使用:在应用程序中调用 TNN 引擎,非常适合各种开发环境。

技术解决方案

TNN 的技术特点主要包括计算优化、低精度计算加速和内存优化:

  • 计算优化:利用 Winograd、Tile-GEMM 等多种算法实现高效计算。
  • 低精度计算加速:支持 INT8/FP16 模式,通过算法提升计算速度和降低内存使用。
  • 内存优化:通过高效的内存管理策略,显著降低模型的内存占用。

贡献与协作

TNN 框架致力于成为行业最优秀的推理框架,期待各位开发者和研究人员的贡献与合作。大家可以通过技术交流QQ群(704900079,答案:TNN)参与讨论,与我们共同推动 TNN 的发展。扫描下方二维码加入讨论。

<div align=left ><img src="TNN-QQ.png"/>

编辑推荐精选

Vora

Vora

免费创建高清无水印Sora视频

Vora是一个免费创建高清无水印Sora视频的AI工具

Refly.AI

Refly.AI

最适合小白的AI自动化工作流平台

无需编码,轻松生成可复用、可变现的AI自动化工作流

酷表ChatExcel

酷表ChatExcel

大模型驱动的Excel数据处理工具

基于大模型交互的表格处理系统,允许用户通过对话方式完成数据整理和可视化分析。系统采用机器学习算法解析用户指令,自动执行排序、公式计算和数据透视等操作,支持多种文件格式导入导出。数据处理响应速度保持在0.8秒以内,支持超过100万行数据的即时分析。

AI工具使用教程AI营销产品酷表ChatExcelAI智能客服
TRAE编程

TRAE编程

AI辅助编程,代码自动修复

Trae是一种自适应的集成开发环境(IDE),通过自动化和多元协作改变开发流程。利用Trae,团队能够更快速、精确地编写和部署代码,从而提高编程效率和项目交付速度。Trae具备上下文感知和代码自动完成功能,是提升开发效率的理想工具。

热门AI工具生产力协作转型TraeAI IDE
AIWritePaper论文写作

AIWritePaper论文写作

AI论文写作指导平台

AIWritePaper论文写作是一站式AI论文写作辅助工具,简化了选题、文献检索至论文撰写的整个过程。通过简单设定,平台可快速生成高质量论文大纲和全文,配合图表、参考文献等一应俱全,同时提供开题报告和答辩PPT等增值服务,保障数据安全,有效提升写作效率和论文质量。

数据安全AI助手热门AI工具AI辅助写作AI论文工具论文写作智能生成大纲
博思AIPPT

博思AIPPT

AI一键生成PPT,就用博思AIPPT!

博思AIPPT,新一代的AI生成PPT平台,支持智能生成PPT、AI美化PPT、文本&链接生成PPT、导入Word/PDF/Markdown文档生成PPT等,内置海量精美PPT模板,涵盖商务、教育、科技等不同风格,同时针对每个页面提供多种版式,一键自适应切换,完美适配各种办公场景。

热门AI工具AI办公办公工具智能排版AI生成PPT博思AIPPT海量精品模板AI创作
潮际好麦

潮际好麦

AI赋能电商视觉革命,一站式智能商拍平台

潮际好麦深耕服装行业,是国内AI试衣效果最好的软件。使用先进AIGC能力为电商卖家批量提供优质的、低成本的商拍图。合作品牌有Shein、Lazada、安踏、百丽等65个国内外头部品牌,以及国内10万+淘宝、天猫、京东等主流平台的品牌商家,为卖家节省将近85%的出图成本,提升约3倍出图效率,让品牌能够快速上架。

iTerms

iTerms

企业专属的AI法律顾问

iTerms是法大大集团旗下法律子品牌,基于最先进的大语言模型(LLM)、专业的法律知识库和强大的智能体架构,帮助企业扫清合规障碍,筑牢风控防线,成为您企业专属的AI法律顾问。

SimilarWeb流量提升

SimilarWeb流量提升

稳定高效的流量提升解决方案,助力品牌曝光

稳定高效的流量提升解决方案,助力品牌曝光

Sora2视频免��费生成

Sora2视频免费生成

最新版Sora2模型免费使用,一键生成无水印视频

最新版Sora2模型免费使用,一键生成无水印视频

下拉加载更多