|
yiyidz
| 来自山东
在执行tools/optimize/yolov5s-opt.py时报错
---- Tengine YOLOv5 Optimize Tool ----
Input model : yolov5s.v4.onnx
Output model : yolov5s.v4.opt.onnx
Input tensor : 167
Output tensor : 381,420,459
[Quant Tools Info]: Step 0, load original onnx model from yolov5s.v4.onnx.
256
[Quant Tools Info]: Step 1, Remove the focus and postprocess nodes.
Traceback (most recent call last):
File "yolov5s-opt.py", line 222, in
main()
File "yolov5s-opt.py", line 184, in main
new_nodes = cut_focus_output(old_node, in_tensor, out_tensor)
File "yolov5s-opt.py", line 82, in cut_focus_output
output_pass[node_dict[out_name]] = 2
KeyError: '381'
是因为模型更新了吗,大佬? |
|