pointnet语义分割内batch_inference.py问题解决

在运行batch_inference.py是遇到的一些问题,在这里记录一下,希望可以给大家带来帮助
  • 报错内容:
unsupported operand type(s) for + :'range' and 'list' 
  • 修改部分代码:在indoor3d_util.py中
def sample_data(data, num_sample):
    """ data is in N x ...
        we want to keep num_samplexC of them.
        if N > num_sample, we will randomly keep num_sample of them.
        if N < num_sample, we will randomly duplicate samples.
    """
    N = data.shape[0]
    if 
我的错误类型是输入/输出形状不匹配 在进行某些操作时数据形状不匹配,我现在使用了onnx作为中间转换工具,但是在onnx转换为rknn模型的过程中依旧报错Traceback (most recent call last): File "/home/liuxin/PycharmProjects/PythonProject/onnx2rknn.py", line 33, in <module> ret = rknn.build(do_quantization=False) File "/home/liuxin/miniconda3/envs/py38/lib/python3.8/site-packages/rknn/api/rknn.py", line 198, in build return self.rknn_base.build(do_quantization=do_quantization, dataset=dataset, expand_batch_size=rknn_batch_size, auto_hybrid=auto_hybrid) File "rknn/api/rknn_log.py", line 349, in rknn.api.rknn_log.error_catch_decorator.error_catch_wrapper File "rknn/api/rknn_log.py", line 95, in rknn.api.rknn_log.RKNNLog.e ValueError: Traceback (most recent call last): File "rknn/api/rknn_log.py", line 344, in rknn.api.rknn_log.error_catch_decorator.error_catch_wrapper File "rknn/api/rknn_base.py", line 1990, in rknn.api.rknn_base.RKNNBase.build File "rknn/api/graph_optimizer.py", line 942, in rknn.api.graph_optimizer.GraphOptimizer.fold_constant File "rknn/api/session.py", line 34, in rknn.api.session.Session.__init__ File "rknn/api/session.py", line 131, in rknn.api.session.Session.sess_build File "/home/liuxin/miniconda3/envs/py38/lib/python3.8/site-packages/onnxruntime/capi/onnxruntime_inference_collection.py", line 419, in __init__ self._create_inference_session(providers, provider_options, disabled_optimizers) File "/home/liuxin/miniconda3/envs/py38/lib/python3.8/site-packages/onnxruntime/capi/onnxruntime_inference_collection.py", line 482, in _create_inference_session sess = C.InferenceSession(session_options, self._model_bytes, False, self._read_config_from_model) onnxruntime.capi.onnxruntime_pybind11_state.Fail: [ONNXRuntimeError] : 1 : FAIL : Node (sequential/transformer_encoder/positional_embedding/add) Op (Add) [ShapeInferenceError] Incompatible dimensions
最新发布
08-10
评论 11
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值