Closed Im-JimmyHu closed 2 years ago
Could you try to modify the source code and rebuild the "yolo_layer" plugin, to see if it resolves the problem?
Replace the above line of code with the following:
bool supportsFormatCombination(int pos, const PluginTensorDesc* inOut, int nbInputs, int nbOutputs) const NOEXCEPT override { return inOut[pos].format == PluginFormat::kLINEAR && inOut[pos].type == DataType::kFLOAT; }
Does my previous comment help?
Does my previous comment help?
yeah ,tks,jk
Hi jk.while Demo #5: YOLOv4 runs well on jetpack4.5.1 and trt7.1.3 ,after I upgrade to jetpack4.6 and tensorrt8.0 ,the error above happened.the error occurred when i run the command:
python3 onnx_to_tensorrt.py -m yolov4-416
the complete log like this:
however the command yolo-onnx runs well, the log like this:
to find where the error from, I try the onnx (got form the jetpack4.5.1,which once turned out good) to exucte the command
python3 onnx_to_tensorrt.py -m yolov4-416
but i got the same error that i don't how to deal with it. Any advice will be appreciated!