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

编辑推荐精选

博思AIPPT

博思AIPPT

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

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

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

潮际好麦

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

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

iTerms

iTerms

企业专属的AI法律顾问

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

SimilarWeb流量提升

SimilarWeb流量提升

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

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

Sora2视频免费生成

Sora2视频免费生成

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

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

Transly

Transly

实时语音翻译/同声传译工具

Transly是一个多场景的AI大语言模型驱动的同声传译、专业翻译助手,它拥有超精准的音频识别翻译能力,几乎零延迟的使用体验和支持多国语言可以让你带它走遍全球,无论你是留学生、商务人士、韩剧美剧爱好者,还是出国游玩、多国会议、跨国追星等等,都可以满足你所有需要同传的场景需求,线上线下通用,扫除语言障碍,让全世界的语言交流不再有国界。

讯飞绘文

讯飞绘文

选题、配图、成文,一站式创作,让内容运营更高效

讯飞绘文,一个AI集成平台,支持写作、选题、配图、排版和发布。高效生成适用于各类媒体的定制内容,加速品牌传播,提升内容营销效果。

热门AI辅助写作AI工具讯飞绘文内容运营AI创作个性化文章多平台分发AI助手
TRAE编程

TRAE编程

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

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

AI工具TraeAI IDE协作生产力转型热门
商汤小浣熊

商汤小浣熊

最强AI数据分析助手

小浣熊家族Raccoon,您的AI智能助手,致力于通过先进的人工智能技术,为用户提供高效、便捷的智能服务。无论是日常咨询还是专业问题解答,小浣熊都能以快速、准确的响应满足您的需求,让您的生活更加智能便捷。

imini AI

imini AI

像人一样思考的AI智能体

imini 是一款超级AI智能体,能根据人类指令,自主思考、自主完成、并且交付结果的AI智能体。

下拉加载更多