Does a tutorial exist for converting a YOLOv5-Seg model to a Hailo HEF file with NMS post-processing included in the model script, running on the host CPU?
Or link to the existing hef file… I found yolov5m-seg.hef, with nms on cpu…
I solved it and I will put later the complete script and explanation how to search for the correct nodes.
I’m having the same issue. Could you please provide the complete script? Thank you!
Here’s a guide on how to add the nms-postprocess to the hef:
This guide walks you through the process of taking a YOLOv5-Seg model, adding an NMS post-processing layer, and compiling it into an HEF file.
Create model script
Post-processing must be added to the network using a model script.
This means adding the nms_postprocess command to the script, which will later be used with the DFC or ModelZoo tools.
Example Model script for YOLOv5-Seg models:
normalization1 = normalization([0.0, 0.0, 0.0], [255.0, 255.0, 255.0]) nms_postprocess(meta_arch=yolov5…