airockchip / rknn-toolkit2

Other
996 stars 104 forks source link

4 dimensions input is required at runtime but not at build time #82

Open medvedev opened 4 months ago

medvedev commented 4 months ago

It is allowed to build a rknn file with batch size > 1 and input array having less than 4 dimensions. However at runtime I get an error:

E RKNN: [12:17:25.781] only support input n_dims == 4 when batch_size > 1

Ideally, I'd like to have a possibility to use any number of dimensions. But at the moment it would be nice to get an error if I'm trying to build a rknn file that won't be accepted by a runtime. rknn-lite version: 2.0.0b1