Closed swithmn1 closed 3 years ago
The same error, i try convert Yolov4 and Yolov4-p5 darknet models. And got error:
File "yolo_to_onnx.py", line 750, in <module>
main()
File "yolo_to_onnx.py", line 742, in main
model_def = builder.build_onnx_graph(
File "yolo_to_onnx.py", line 324, in build_onnx_graph
major_node_specs = self._make_onnx_node(layer_name, layer_dict)
File "yolo_to_onnx.py", line 397, in _make_onnx_node
node_creators[layer_type](layer_name, layer_dict)
File "yolo_to_onnx.py", line 700, in _make_yolo_node
down_stride = int(layer_dict['down_stride'])
KeyError: 'down_stride'
you can look my yolov3.cfg. like https://github.com/ttanzhiqiang/onnx_tensorrt_project/blob/main/model/darknet_onnx_tensorrt_yolo/yolov3.cfg
you add down_stride = 8 in yolo Plugin.
@ttanzhiqiang ok, i try to use your yolov3.cfg, it works!