Android使用chaquo框架从Java无法传数组给Python的Numpy的NDArray的问题,com.chaquo.python.PyException:

8 篇文章 1 订阅
5 篇文章 0 订阅

针对这个问题,我也是找了很久才找到答案,现在让我们来看看是如何解答的。

背景:

我的需求是在Android开发的项目中,需要获取蓝牙的数据然后通过chaquo传参数给深度学习的模型,模型得到一个resultcode返回给安卓端。

问题所在

深度学习开发的师兄提供给我preidict函数的接口,该函数用于预测行为。函数的参数是python中大名鼎鼎的numpy的NDArray对象,在本项目中要求是要20*3维的numpy数组,也就是:array[20][3]

第一次做法:

第一次我没想太多,我想着numpy对应的不也是数组嘛,所以我用java简单整理了数据发了一个float array[20][3]数组给python端,结果却爆了错。我第一次尝试的代码如下:

PyObject obj1 = py.getModule("Deployment").callAttr("predict"
                                                ,conbinedArray );
                                                                          

代码解析:其实就是调用了Deployment.py脚本下的predict方法,并且传了一个二维数组(20*3)conbinedArray

但噩梦此拉开帷幕。

报错

此时通过debug,try-catch等一系列操作下来发现,chaquo报TypeError错误如下(复制未完全)。

com.chaquo.python.PyException: TypeError: predict() takes 1 positional argument but 3 were given.

意思就是,这个函数仅仅接受一个参数,而我提供了3个。

疑问

这下子就会很纳闷了,predict函数确实是一个参数没错,但是是一个numpy的二维数组呀,我用java传一个二维数组过去难道不正确吗?
这个疑问一直伴随了我一天,后面搜到以下方法,能够解决这个问题。

解决方案

先贴代码

PyObject np = py.getModule("numpy"); // 初始化numpy模块
// 调用numpy.array()函数,例如numpy.array([1,2,3]) --> [1,2,3]
PyObject conbined_final = np.callAttr("array",conbinedArray[0]);
// 调用numpy.expand_dims(array,0 or 1)函数,例如
// numpy.expand_dims([1,2,3],0) --> [[1,2,3]]
// numpy.expand_dims([1,2,3],1) --> [[1],[2],[3]]
// 这个函数就是给array添加维度
conbined_final = np.callAttr("expand_dims", conbined_final, 0);
//这个for循环就是把java中的二维数组中的每一行都转换为一个array例如[1,2,3]
// 然后调用numpy的append方法,一个个添加到conbined_final数组中,我在后面放了张图片,大家可以理解一下
for (int i=1; i < conbinedArray.length; i++){
	// 假设下面这行获得一个[[1,2,3]]
    PyObject temp_arr = np.callAttr("expand_dims", conbinedArray[i], 0);
    // 假设已存在conbined_final=[[6,6,6]]
    // 那么下面这行得到的结果就是[[6,6,6],[1,2,3]]
    //也就是说,我们用java代码调用python numpy库合成了一个predict的函数参数
    conbined_final = np.callAttr("append", conbined_final, temp_arr, 0);
}
// 此时我们就可以放心大胆地把生成的对象发送给脚本了
PyObject obj1 = py.getModule("Deployment").callAttr("predict",conbined_final );
// 获取结果
int result = obj1.toJava(int.class);

谢谢大家的阅读,欢迎评论交流。

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

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值