Page History
...
Please run yolov5_cut.py with the following command. If executed successfully, yolov5s_v5_wo_post.onnx will be generated.
$ python3 yolov5_cut.py
The above output names, 418, 378, and 398, are for YOLOv5s. Please note that different names must be specified for YOLOv5l and YOLOv5m. As shown in the figure below, please find and specify the output name of Conv before Shape and ReShape of the three outputs of YOLOv5.
5. Translate ONNX
...
Overview
Content Tools