3 Star 2 Fork 0

Gitee 极速下载/wechat-tfcc

加入 Gitee
与超过 1200万 开发者一起发现、参与优秀开源项目,私有仓库也完全免费 :)
免费加入
此仓库是为了提升国内下载速度的镜像仓库,每日同步一次。 原始仓库: https://github.com/Tencent/WeChat-TFCC
该仓库未声明开源许可证文件(LICENSE),使用请关注具体项目描述及其代码上游依赖。
克隆/下载
贡献代码
同步代码
取消
提示: 由于 Git 不支持空文件夾,创建文件夹后会生成空的 .keep 文件
Loading...
README

TFCC

TFCC is a C++ deep learning inference framework.

TFCC provides the following toolkits that faciliate your development and deployment of your trained DL models:

Porject Source Description
TFCC ./tfcc The core of deep learning inference library. It provides friendly interfaces for model deployment, as well as the implementation of diverse operations in both MKL and CUDA environments.
TFCC Code Generator ./tfcc_code_generator An automatic generator that can optimize the structure of your high-level models (tensorflows, pytorch, etc.) and generate the TFCC model.
TFCC Runtime ./tfcc_runtime An runtime to load TFCC model and inference.

BUILD

Run

./build.sh ${INSTALL_PREFIX_PATH}

Quick Start

  1. Convert Model

    The script generator.py can convert onnx model or tensorflow model to tfcc model. The docs Convert ONNX Model and Convert TF Model show the details.

  2. Load Model

    There is a simple way to load a model as following code:

    // load tfcc model to a string.
    std::string modelData = load_data_from_file(path);
    tfcc::runtime::Model model(modelData);
    
  3. Inference

    Finally run the model

    tfcc::runtime::data::Inputs inputs;
    tfcc::runtime::data::Outputs outputs;
    
    // set inputs
    auto item = inputs.add_items();
    item->set_name("The input name");
    item->set_dtype(tfcc::runtime::common::FLOAT);
    std::vector<float> data = {1.0, 2.0};
    item->set_data(data.data(), data.size() * sizeof(float));
    
    model.run(inputs, outputs);
    

    Complete code

空文件

简介

WeChat TFCC 是微信技术架构部后台团队研发的服务端深度学习通用推理框架,也是腾讯深度学习与加速Oteam云帆的开源协同成果,具有高性能、易用、通用的特点,已在微信视频号、 展开 收起
取消

发行版

暂无发行版

贡献者

全部

近期动态

不能加载更多了
马建仓 AI 助手
尝试更多
代码解读
代码找茬
代码优化
C/C++
1
https://gitee.com/mirrors/wechat-tfcc.git
git@gitee.com:mirrors/wechat-tfcc.git
mirrors
wechat-tfcc
wechat-tfcc
master

搜索帮助