CoolFace
Modelpublic

proj-airi/factorio-yolo-v0

sourceHugging Facemitupdated 7mo agoView on Hugging Face
1likes
Model Card

Factorio YOLO v0

![Open In Colab](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.

Dataset is available on HuggingFace. Training and prediction are available in airi-factorio.

Playground(WIP) is hosted on HuggingFace. 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="./valbatch0pred.jpg" alt="Results" style="width: 40%;object-fit: contain;" title="Validation batch 0 predictions"> </p>

Training and Prediction

See airi-factorio for further details.