search

Tianxiaomo / pytorch-YOLOv4

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

I met some problem when img width is different with height. how to configure? #544

Open iamecong opened 1 year ago

iamecong commented 1 year ago

RuntimeError: shape '[2, RuntimeError: shape '[2, 3, 7, 56, 56]' is invalid for input of size 2257923, 7, 56, 56]' is invalid for input of size 225792RuntimeError: shape '[2, 3, 7, 56, 56]' is invalid for input of size 225792