044daafd9d
* Add output names & dynamic axes for onnx export Add output_names and dynamic_axes names for all outputs in torch.onnx.export. The first four outputs of the model will have names output0, output1, output2, output3 * use first output only + cleanup Co-authored-by: Samridha Shrestha <samridha.shrestha@g42.ai> Co-authored-by: Glenn Jocher <glenn.jocher@ultralytics.com> |
||
|---|---|---|
| .. | ||
| hub | ||
| __init__.py | ||
| common.py | ||
| experimental.py | ||
| export.py | ||
| yolo.py | ||
| yolov5l.yaml | ||
| yolov5m.yaml | ||
| yolov5s.yaml | ||
| yolov5x.yaml | ||