Stephenfang51 / tracklite

Using Tensort to speed up yolov3 with deepsort for MOT
157 stars 48 forks source link

python3 onnx_to_tensorrt.py failed #22

Open Myron1996 opened 3 years ago

Myron1996 commented 3 years ago
tdc@tdc:~/Downloads/tracklite$ python3 onnx_to_tensorrt.py --onnx ./weights/yolov3_416.onnx --output_engine ./weights/yolov3_416.engine
Loading ONNX file from path ./weights/yolov3_416.onnx...
Beginning ONNX file parsing
Completed parsing of ONNX file
Building an engine from file ./weights/yolov3_416.onnx; this may take a while...
[TensorRT] ERROR: Network must have at least one output
[TensorRT] ERROR: Network validation failed.
Completed creating Engine
Traceback (most recent call last):
  File "onnx_to_tensorrt.py", line 56, in <module>
    main()
  File "onnx_to_tensorrt.py", line 52, in main
    get_engine(args.onnx, args.output_engine)
  File "onnx_to_tensorrt.py", line 45, in get_engine
    f.write(engine.serialize())
AttributeError: 'NoneType' object has no attribute 'serialize'

Can you help please @Stephenfang51 . Thank you

byq-luo commented 3 years ago

我也是这个问题,求指导