biyuehuang

MODIFY y.extend((ones(b-a)*i).tolist()) TO y.extend((ones(int(b-a), dtype=np.int32)*i).tolist())

你好，请教一个小问题。我编译好了RK_NPU script/build_aarch64_linux.sh 和 examples/linux/cross/build_aarch64_linux.sh 都编译成功。 想用 examples/linux/cross/build/demo_arm_imageclassify 进行RKNPU推理。 于是把修改examples/linux/src里的.cc代码，把 TNNComputeUnitsCPU 改成 TNNComputeUnitsRKNPU。重新编译examples/linux/cross/build_aarch64_linux.sh 报错：没有TNNComputeUnitsRKNPU 请问您的代码是从哪里来的呢？TNN怎么用RKNPU呢？

> > 你好，请教一个小问题。我编译好了RK_NPU script/build_aarch64_linux.sh 和 examples/linux/cross/build_aarch64_linux.sh 都编译成功。 想用 examples/linux/cross/build/demo_arm_imageclassify 进行RKNPU推理。 > > 于是把修改examples/linux/src里的.cc代码，把 TNNComputeUnitsCPU 改成 TNNComputeUnitsRKNPU。重新编译examples/linux/cross/build_aarch64_linux.sh 报错：没有TNNComputeUnitsRKNPU > > 请问您的代码是从哪里来的呢？TNN怎么用RKNPU呢？ > > 自己写，device_type和network_type要指定是rknpu的 您好，我尝试用benchmark_armlinux/benchmark_model.sh ，加入-DTNN_RK_NPU_ENABLE:BOOL=ON 编译成功。ARMCPU可以用，但是用不了RK_NPU，请问你用过benchmark RK_NPU吗？ 如果TNN benchmark 无法用RK_NPU，请问您的代码是参考哪个代码写的呢？

I met the same problem. I want to convert yolact_plus to onnx. ``` import DCNv2 def evalimage(net:Yolact, path:str, save_path:str=None): frame = torch.from_numpy(cv2.imread(path)).float() batch = FastBaseTransform()(frame.unsqueeze(0)) preds = net(batch) torch.onnx.export(net, batch,...