airi-factorio/models/factorio-yolo-v0

Factorio YOLO v0

A model for detecting items in Factorio based on YOLOv11n. It can detect 6 items, see names.

Playground is hosted on HuggingFace.

Available formats:

• ONNX
• PyTorch

Results

Validation batch 0 labels and predictions:

Dataset

The dataset is a collection of images of items in Factorio, collector code here.

It contains 20 images for each item, 17 for training and 2 for validation and 1 for testing.

Training

pixi train

Predicting

It will show a window with the predicted bounding boxes.

pixi predict