Re: Which version of pytorch is available with nnstreamer-pytorch 2.1.0.0-0~202111300837~ubuntu18.04.1?


MyungJoo Ham
 

... ! video/x-raw,format=RGB,width=640,height=640 ! tensor_converter ! tensor_filter framework=pytorch model=../../tf_model/yolov5s.torchscript.pt input=3:640:640:1 inputname=x inputtype=float32 output=1:25200:85 outputname=416 outputtype=float32 ...

The actual tensor you've entered for tensor_filter is 3:640:640:1 (uint8) and you've stated that the tensor_filter needs to get 3:640:640:1 (float32). In other words, you are pushing uint8[640][640][3] into float32[640][640][3] array.

You need to transform uint8 stream to float32 stream with tensor_transform before you feed it to tensor_filter.

nnstreamer-pytorch 2.1.0 is tested with pytorch 1.1, 1.3, and 1.6; but this is not related with version compatibility.

Join {nnstreamer-technical-discuss@lists.lfaidata.foundation to automatically receive all group messages.