前几天用pytorch训练了一个目标检测的模型,但是因为要部署的嵌入式板子只支持tensorlfow、caffe等模型,最重要的是我从来没有使用过这两框架,坑跌的是tensorflow也只有1.9版本支持。想到了四种方案,第一种tensorlfow重新训练编写模型、第二种使用onnx框架进行模型转换、第三种使用caffe重写整个模型、第四使用板子支持的框架只重写inference阶段的模型,同时将pytorch训练好的权重进行迁移



博客内容遵循 署名-非商业性使用-相同方式共享 4.0 国际 (CC BY-NC-SA 4.0) 协议

本站使用 Material X 作为主题 , 总访问量为 次 。
载入天数...载入时分秒...