Recommended way to export Pre/Post-Processing with YOLOv5

[RobinFrcd]

Hi,
I have a YOLOv5 model I want to script with both pre and post-processing functions. What would be the way to go ? The YOLOv5 postprocess does not include the nms. What is the way to go in that case ?

I know there's https://github.com/zhiqwang/yolort/blob/main/notebooks/how-to-align-with-ultralytics-yolov5.ipynb but everything is not in the exported model here !

Thanks !