Easy YOLO ONNX

A simple CLI utility to download Ultralytics YOLO models (v5, v8, v11) and export them to ONNX format.

Features

• Auto-download: Fetches .pt models usage the ultralytics library.
• Custom Resolution: Choose between 640, 960, and 1280 input sizes.
• Auto-class Export: Generates a _classes.txt file identifying all object classes (ID: Name).
• Minimal Dependencies: Only requires ultralytics and onnx.

Usage

1. Install generic requirements (script will attempt auto-install):

   pip install -r requirements.txt

2. Run the script:

   python export_model.py

3. Follow the interactive prompts:

   • Select YOLO Version (v5, v8, v11)
   • Select Model Size (Nano to Extra Large)
   • Select Resolution

Output

The script generates two files:

• <model>.onnx: The generic ONNX model ready for inference.
• <model>_info.txt: A text file containing model metadata and class list.