According to the log you've send in the previous message, calling pytorch's load function has failed:
try {
#ifdef PYTORCH_VER_ATLEAST_1_2_0
model = std::make_shared<torch::jit::script::Module> (torch::jit::load (model_path));
#else
model = torch::jit::load (model_path);
#endif
} catch (const std::invalid_argument &ia) {
ml_loge ("Invalid argument while loading the model: %s", ia.what ());
return -1;
} catch (const std::exception &ex) {
ml_loge ("Exception while loading the model: %s", ex.what ());
return -1;
|
the exception is from "Exception while loading the model", which means that the model you've entered
has failed with "load()" function.
Please check if the pytorch invoked by nnstreamer is correct (depending on your own shard-library configurations, different pytorch might be invoked when you call it via nnstreamer/gstreamer). Check if the corresponding pytorch instance is compatible with your model. Then, check if your tensor_filter arguments, "inputname", "outputname" are correct, too.
Cheers,
MyungJoo.
ps. Please keep this thread public.
ps.2 If you want more active & efficient discussion, you can try github/issue
toggle quoted message
Show quoted text
---------
Original Message ---------
Sender : 이병헌 <byungs286@...>
Date : 2021-12-02 15:51 (GMT+9)
Title : Private: Re: [NNStreamer Technical Discuss] Which version of pytorch is available with nnstreamer-pytorch 2.1.0.0-0~202111300837~ubuntu18.04.1?
Thanks for reply
I've tried your advice as like below
const char *string = "rtspsrc location=rtsp://address:port/mount latency=0 protocols=4 ! \
rtph265depay ! \
avdec_h265 ! \
videoscale ! \
videoconvert ! \
video/x-raw,format=RGB,width=640,height=640 ! \
tensor_converter ! \
tensor_transform mode=typecast option=float32 ! \
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 ! \
tensor_sink name=tensor_sink";
but I still have same error as below..
failed to initialize the object: PyTorch
0:00:00.159368517 264 0x564b2683b140 WARN GST_PADS gstpad.c:1149:gst_pad_set_active:<tensorfilter0:sink> Failed to activate pad
** Message: 06:37:13.822: gpu = 0, accl = cpu
** (tester:264): CRITICAL **: 06:37:13.825: Exception while loading the model:
** (tester:264): CRITICAL **: 06:37:13.825: Failed to load model
failed to initialize the object: PyTorch
Is there have any other problem in that pipeline?
|
|
