Mask RCNN裁剪后移植到Android端,报错解决方案

模型是在tensorflow r1.12中训练的,所以使用tensorflow r1.12的源码编译libtensorflow_inference.so来配套移植到Android端,结果报了以下两个错误:

1. Op type not registered 'NonMaxSuppressionV3' in binary running on localhost. Make sure the Op and Kernel are registered in the binary running in this process.

参考:https://www.jianshu.com/p/056ba3cbf02a

2. No OpKernel was registered to support Op 'DenseToDenseSetOperation' with these attrs.  Registered devices: [CPU], Registered kernels:

1) 在tensorflow/contrib/makefile/tf_op_files.txt中添加set_kernels.cc

2) 在tensorflow/core/kernels/BUILD中找到:

filegroup(
    name = "android_extended_ops_group2",
    srcs = [

    ]

在srcs里面加上set_kernels.cc

3) 重新编译libtensorflow_inference.so

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

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值