Re: Which version of pytorch is available with nnstreamer-pytorch

MyungJoo Ham

... ! video/x-raw,format=RGB,width=640,height=640 ! tensor_converter ! tensor_filter framework=pytorch model=../../tf_model/ 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 { to automatically receive all group messages.