Models
Orion provides YOLO12 models fine-tuned on a custom dataset of military vehicles with 4 classes.
| Model | size (pixels) |
params (M) |
|---|---|---|
| orion12n | 640 | 2.6 |
| orion12s | 640 | 9.3 |
| orion12m | 640 | 20.2 |
| orion12l | 640 | 26.4 |
Training results
Orion12n
Orion12s
Orion12m
Orion12l
Usage
To use one of Orion's models, download it from the links above and then use one of Orion's CLI commands:
Detect military vehicles in images
The predict command will use the model to detect military vehicles in images.
orion predict --help
Usage: orion predict [OPTIONS] MODEL_PATH DATA
Run predictions on a set of images using the given model.
╭─ Arguments ─────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────╮
│ * model_path PATH model path. [required] │
│ * data PATH data to make predictions on. [required] │
╰─────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────╯
╭─ Options ───────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────╮
│ --save -s save annotated images. │
│ --save-txt --no-save-txt save detection results in a txt file. [default: save-txt] │
│ --save-conf --no-save-conf save confidence score for each detection. [default: save-conf] │
│ --output -o DIRECTORY save directory. [default: Path.cwd() /runs/predict] │
│ --help Show this message and exit. │
╰─────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────╯
For example, if you downloaded the orion12m.pt model file into your current directory, and want to use it to detect military vehicles in an image, run
Tip
The predict command with the -s option will save the annotated image.
Track military vehicles in videos
The track command will use the model to track military vehicles in videos.
orion track --help
Usage: orion track [OPTIONS] MODEL_PATH DATA
Track tanks in a video using a YOLO model and specified tracker.
╭─ Arguments ─────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────╮
│ * model_path PATH model path. [required] │
│ * data PATH input video. [required] │
╰─────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────╯
╭─ Options ───────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────╮
│ --conf -c FLOAT confidence threshold for detections. [default: 0.5] │
│ --tracker -t TEXT tracker configuration file. [default: botsort.yaml] │
│ --output -o DIRECTORY save directory. [default: Path.cwd() /runs/track] │
│ --help Show this message and exit. │
╰─────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────────╯
For example, if you downloaded the orion12m.pt model file into your current directory, and want to use it to track military vehicles in a video, run
