[TensorRT] ERROR: Network must have at least one output
yolov3 转 tensorrt,运行onnx转tensorrt 有时会遇到上述错误。
onnx转tensorrt.py文件中,在parser.parse下with循环外添加两行代码
last_layer = network.get_layer(network.num_layers - 1)
network.mark_output(last_layer.get_output(0))