how to export trt model？

TrojanXu / yolov5-tensorrt

A tensorrt implementation of yolov5: https://github.com/ultralytics/yolov5

Apache License 2.0

190 stars 46 forks source link

Closed Wulingtian closed 4 years ago

TrojanXu commented 4 years ago

with open(engine_file_path, "wb") as f: f.write(engine.serialize())