search

rockchip-linux / rknn-toolkit2

BSD 3-Clause "New" or "Revised" License
828 stars 151 forks source link

Does rknn support five-dimensional data input (video)? #239

Open Z1740220020 opened 8 months ago

Z1740220020 commented 8 months ago

My model needs to input 5 consecutive frames of images and detect moving targets in the picture, which uses some 3D convolution. Now it can be successfully converted to the ONNX model, but when converting the RKNN model, an error occurs saying that the input size is incorrect. It seems that it only supports image input. The official yolo example only processes a single image and inputs 4-dimensional data.

Is there any way to make RKNN support 5-dimensional data? (Add one more time dimension on the basis of B, C, H, and W)

minocrafft commented 6 months ago

What version did you use?

kejndan commented 2 months ago

Has anyone found a solution to this problem?