lucasjinreal/yolov7_d2

Error exporting to onnx

nagulsr opened this issue · 3 comments

i'm getting this issue on exporting to onnx

[08/25 11:30:02 detectron2]: input shape: torch.Size([1, 640, 640, 3])
Traceback (most recent call last):
File "C:\Users\nagulehambharam.r\OneDrive - Wabtec Corporation\Desktop\yolov7_d2-main\export.py", line 284, in
input_names, output_names, dynamic_axes = get_model_infos(args.config_file)
ValueError: not enough values to unpack (expected 3, got 1)

Any answer for it?

Hi,
What is the solution to this issue? Thank you in advance.

the export works only for dection models.