Error when reading Yolo5 as onnx using cv2

Python
, ,
3

I used Google Colab for everything

  1. !git clone GitHub - ultralytics/yolov5: YOLOv5 🚀 in PyTorch > ONNX > CoreML > TFLite - Cloned the GitHub repo
  2. !python train.py --data data.yaml --cfg yolov5s.yaml --batch-size 8 --name Model --epochs 50 - Initiated the training process for Yolo5
  3. !python export.py --weights runs/train/Model/weights/best.pt --include onnx --simplify - Converted the PyTorch model(yolo) to onnx

If you need more details on each thing I am glad to provide any information!

1 Like