search

ttanzhiqiang / onnx_tensorrt_project

Support Yolov5(4.0)/Yolov5(5.0)/YoloR/YoloX/Yolov4/Yolov3/CenterNet/CenterFace/RetinaFace/Classify/Unet. use darknet/libtorch/pytorch/mxnet to onnx to tensorrt
210 stars 43 forks source link

KeyError: 'down_stride' #8

Closed swithmn1 closed 3 years ago

swithmn1 commented 3 years ago

image

ogoun commented 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'
ttanzhiqiang commented 3 years ago

you can look my yolov3.cfg. like https://github.com/ttanzhiqiang/onnx_tensorrt_project/blob/main/model/darknet_onnx_tensorrt_yolo/yolov3.cfg image

you add down_stride = 8 in yolo Plugin.

swithmn1 commented 3 years ago

@ttanzhiqiang ok, i try to use your yolov3.cfg, it works!