search

lucasjinreal / yolov7_d2

🔥🔥🔥🔥 (Earlier YOLOv7 not official one) YOLO with Transformers and Instance Segmentation, with TensorRT acceleration! 🔥🔥🔥
GNU General Public License v3.0
3.13k stars 483 forks source link

Error exporting to onnx #175

Closed nagulsr closed 2 years ago

nagulsr commented 2 years ago

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)

nagulsr commented 2 years ago

Any answer for it?

wirefree commented 1 year ago

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

lucasjinreal commented 1 year ago

the export works only for dection models.