tf. 函数库与np.函数库的转换

前言:

        在对算法模型进行部署的时候,往往需要做一些工作。以xilinx开发板部署tf框架模型为例,首先需要对训练好的ckpt模型进行freeze得到.pb模型,之后,对其进行量化、编译生成elf文件,然后在板卡上通过main.cc函数调用pb模型的输入/输出节点对网络模型进行计算,注意这里的计算分为两部分,其中模型的输入数据前处理和输出数据后处理的计算部分由板卡中的ARM处理器进行计算,模型的卷积计算来自板卡中的高性能算力。

        这里主要针对算法在后处理部分实现时,由于ARM中不支持tf的大量依赖库,只能依靠于numpy的数据库进行测试。因此就有了对inference中的tf.函数库转为np.函数库的需求。

案例:

1.tf.shape&np.shape

2.tf.reshape&np.reshape

3.tf.range&np.arange

4.tf.tile&np.tile

5.tf.concat&np.concatenate

6.tf.exp&np.exp

7.tf.sigmoid(x)&1/(1+np.exp(-x))

后记:

突然感觉np的确很np!

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值