Allow passing onnx_file_name

Tianxiaomo / pytorch-YOLOv4

PyTorch ,ONNX and TensorRT implementation of YOLOv4

Apache License 2.0

4.46k stars 1.49k forks source link

Allow passing onnx_file_name #485

Closed dumbPy closed 2 years ago

dumbPy commented 2 years ago

Allow passing onnx_file_name to the transform_to_onnx so it can be used in any script and output file can be created in a pre-determined location.