深度学习模型部署

1.模型应用场景

我只在PC电脑和嵌入式板卡上部署

2.部署环境

包括原生态darknet,原生态caffe,tensorflow,onnx,opencv,trt等环境

3.选择方式

        之前是在PC电脑上运行程序,经常使用的是darknet和caffe,环境部署简单,前面的文章有提到如何安装环境。

        后来用opencv的dnn模块,发现非常方便,可以加载yolo,caffe,tensorflow部分模型,onnx部分模型,所以碰到一些模型优先会想到opencv来加载调用,甚至不需要做模型转换,非常方便。

        现在在jetson板子上加载模型,加载的tf2.0的模型,opencv没办法直接加载,需要做模型转换,对于基础类网络结果如resnet,ssd,mobilenet,maskrcnn等可以转成tf1.0的模型,然后再用opencv加载,也很方便。但是有些模型没办法转换,比如ssd_efficient等,我是没有转换成功。jetson上直接用tf2.0的环境也可以,但是也有问题,比如加载save_model模型时间比较长,有人说是驱动版本问题,这个我没有去试,有人说重装protobuf,对我来说没有用,所以想到tf2.0=》onnx=>TRT

4.TF2转onnx

        参考官网教程就好https://github.com/onnx/tensorflow-onnx,有一点儿需要注意的是,

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

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值