search

Tianxiaomo / pytorch-YOLOv4

PyTorch ,ONNX and TensorRT implementation of YOLOv4
Apache License 2.0
4.46k stars 1.49k forks source link

support for yolo-fastest #506

Open bieganski opened 2 years ago

bieganski commented 2 years ago

is it possible for demo_darknet2onnx.py script to add support for Yolo-Fastest (https://github.com/dog-qiuqiu/Yolo-Fastest/) configs?

python3 demo_darknet2onnx.py  yolo-fastest.cfg  80classes.txt yolo-fastest.weights image.jpg  0

produces errors:

unknown type dropout
unknown type dropout
unknown type dropout
unknown type dropout
unknown type dropout
...
    print_cfg(self.blocks)
  File "/home/mateusz/github/Yolo-Fastest/Yolo-Fastest/COCO/pytorch-YOLOv4/tool/config.py", line 170, in print_cfg
    prev_width = out_widths[from_id]
IndexError: list index out of range