search

CaoWGG / TensorRT-YOLOv4

tensorrt5, yolov4, yolov3,yolov3-tniy,yolov3-tniy-prn
264 stars 76 forks source link

KeyError: 'stride' #29

Closed Stephenfang51 closed 4 years ago

Stephenfang51 commented 4 years ago

Hi Sir

run yolo_to_onnx.py

Traceback (most recent call last):
  File "yolo_to_onnx.py", line 746, in <module>
    main()
  File "yolo_to_onnx.py", line 741, in main
    verbose=True)
  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 454, in _make_conv_node
    stride = layer_dict['stride']
KeyError: 'stride'

why ? Thanks so much !