---
license: mit
pipeline_tag: object-detection
tags:
- game
- airi
- object-detection
- pytorch
- onnx
- yolo
base_model:
- Ultralytics/YOLO11
---
# Factorio YOLO v0

[![Open In Colab](https://colab.research.google.com/assets/colab-badge.svg)](https://colab.research.google.com/github/proj-airi/factorio-yolo-v0/blob/main/notebook.ipynb)

A model for detecting items in Factorio based on YOLOv11n for [AIRI](https://github.com/moeru-ai/airi).

Dataset is available on [HuggingFace](https://huggingface.co/datasets/proj-airi/factorio-yolo-dataset-v0).
Training and prediction are available in [airi-factorio](https://github.com/moeru-ai/airi-factorio/tree/main/models/factorio-yolo-v0).

Playground(WIP) is hosted on [HuggingFace](https://huggingface.co/spaces/proj-airi/factorio-yolo-v0-playground).
Playground repo: https://github.com/proj-airi/game-playing-ai-playground-2d

Available formats:
- ONNX
- PyTorch

## Results

<p align="center">
  <img src="./results.png" alt="Results" style="width: 100%;object-fit: contain;" title="Results">
</p>

Validation batch 0 labels and predictions:

<p align="center">
  <img src="./val_batch0_pred.jpg" alt="Results" style="width: 40%;object-fit: contain;" title="Validation batch 0 predictions">
</p>

## Training and Prediction

See [airi-factorio](https://github.com/moeru-ai/airi-factorio/tree/main/models/factorio-yolo-v0) for further details.