Closed flmello closed 4 months ago
So, it comes that my generated model supports just 1 image into the pipeline, against trafficnet that suports more than one. I was setting batch-size with the number of video inputs, and for Yolo I can't do that, it must be set to 1. batch-size=1 at the config file turns to be the correct config.
I have encountered the same problem when using the Python API. May I know how to solve it specifically
It comes that my Yolo model was trained with just 1 batch, and I was setting batch-size, at config file, to 2 or more. I had to keep it to 1. By changing this, it solved my error.
You exported the model with --batch 1
. Set it --dynamic
or set the number of the batch-size using --batch
in the exporter file.
Now I exported the model will --dynamic. I set nvstreammux "batch-size" to 2, and set the nvinfer pgie "batch-size" to 2. Just in case, I also set the config_infer_primary.txt with batch-size=2.
However, the pipeline doesn't run, it says "Backend has maxBatchSize 1 whereas 2 has been requested".
So, either the exporting script is not exporting batch sizes greater than 1, or the is something being wrongly set in the pipeline script. Do you have any suggestions?
I running deepstream-app and a pyton script correctly when I provide just 1 video source, but it crashes when 2 or more streams are provided. It seems that there is something I misunderstood at the config file. Note that, when I change the config file (currently for yolov5) and use another config file (for traficnet, for instance) the pipeline is created OK. This is the error I get:
And this is the script test4_yolov5.py.txt and the config file dstest4_pgie_nvinfer_yolov5_config.txt