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"/>

编辑推荐精选

QoderWork

QoderWork

阿里Qoder团队推出的桌面端AI智能体

QoderWork 是阿里推出的本地优先桌面 AI 智能体,适配 macOS14+/Windows10+,以自然语言交互实现文件管理、数据分析、AI 视觉生成、浏览器自动化等办公任务,自主拆解执行复杂工作流,数据本地运行零上传,技能市场可无限扩展,是高效的 Agentic 生产力办公助手。

音述AI

音述AI

全球首个AI音乐社区

音述AI是全球首个AI音乐社区,致力让每个人都能用音乐表达自我。音述AI提供零门槛AI创作工具,独创GETI法则帮助用户精准定义音乐风格,AI润色功能支持自动优化作品质感。音述AI支持交流讨论、二次创作与价值变现。针对中文用户的语言习惯与文化背景进行专门优化,支持国风融合、C-pop等本土音乐标签,让技术更好地承载人文表达。

lynote.ai

lynote.ai

一站式搞定所有学习需求

不再被海量信息淹没,开始真正理解知识。Lynote 可摘要 YouTube 视频、PDF、文章等内容。即时创建笔记,检测 AI 内容并下载资料,将您的学习效率提升 10 倍。

AniShort

AniShort

为AI短剧协作而生

专为AI短剧协作而生的AniShort正式发布,深度重构AI短剧全流程生产模式,整合创意策划、制作执行、实时协作、在线审片、资产复用等全链路功能,独创无限画布、双轨并行工业化工作流与Ani智能体助手,集成多款主流AI大模型,破解素材零散、版本混乱、沟通低效等行业痛点,助力3人团队效率提升800%,打造标准化、可追溯的AI短剧量产体系,是AI短剧团队协同创作、提升制作效率的核心工具。

seedancetwo2.0

seedancetwo2.0

能听懂你表达的视频模型

Seedance two是基于seedance2.0的中国大模型,支持图像、视频、音频、文本四种模态输入,表达方式更丰富,生成也更可控。

nano-banana纳米香蕉中文站

nano-banana纳米香蕉中文站

国内直接访问,限时3折

输入简单文字,生成想要的图片,纳米香蕉中文站基于 Google 模型的 AI 图片生成网站,支持文字生图、图生图。官网价格限时3折活动

扣子-AI办公

扣子-AI办公

职场AI,就用扣子

AI办公助手,复杂任务高效处理。办公效率低?扣子空间AI助手支持播客生成、PPT制作、网页开发及报告写作,覆盖科研、商业、舆情等领域的专家Agent 7x24小时响应,生活工作无缝切换,提升50%效率!

堆友

堆友

多风格AI绘画神器

堆友平台由阿里巴巴设计团队创建,作为一款AI驱动的设计工具,专为设计师提供一站式增长服务。功能覆盖海量3D素材、AI绘画、实时渲染以及专业抠图,显著提升设计品质和效率。平台不仅提供工具,还是一个促进创意交流和个人发展的空间,界面友好,适合所有级别的设计师和创意工作者。

图像生成AI工具AI反应堆AI工具箱AI绘画GOAI艺术字堆友相机AI图像热门
码上飞

码上飞

零代码AI应用开发平台

零代码AI应用开发平台,用户只需一句话简单描述需求,AI能自动生成小程序、APP或H5网页应用,无需编写代码。

Vora

Vora

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

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

下拉加载更多