search

rockchip-linux / rknn-toolkit

BSD 3-Clause "New" or "Revised" License
771 stars 175 forks source link

输入格式为[4, 512, 512]的相关问题 #421

Open bro-xie opened 7 months ago

bro-xie commented 7 months ago

我使用的是unet网络，单个输入的尺寸是[4, 1072, 1920]。因此在rknn模型的build中，我选择的dataset是numpy下的npy格式（因为通道数有4个），然后在生成rknn模型时出现了提示：input.numpy need NHWC format，但是整个过程顺利完成了；然后我使用该模型做评估，评估直接卡在了init_runtime那一步（体现为一直在运行，无警告和报错），最下面一行的提示是 I NPUTransfer：Starting NPU Transfer Client，Transfer version 2.1.0。想问一下关于numpy的那个提示是否会对模型有不好的影响，甚至后面评估中卡在init_runtime也是由它导致的。

bro-xie commented 7 months ago

标题中的尺寸写错了

eRaul commented 6 months ago

该告警不影响后面的推理。 请贴下完整的推理日志。