import onnx
input_path = '/home/sunyingl1/yolov5/runs/train/exp_yolov5m_0913_6/weights/best_640_640.onnx'
output_path = '/home/sunyingl1/yolov5/runs/train/exp_yolov5m_0913_6/weights/best_640_640_cut.onnx'
input_names = ["images"]
output_names = ["onnx::Reshape_367","onnx::Reshape_406","onnx::Reshape_445"]
onnx.utils.extract_model(input_path, output_path, input_names, output_names)
yolov5 端侧部署,模型裁剪
最新推荐文章于 2024-09-30 14:16:32 发布