PixelGen2

PixelGen2 от DJLougen — крошечный PyTorch nn.Module для text-to-image: превращает промпт вида «red potion» в 16×16 спрайт в палитре PICO-8 с апскейлом до 128×128. В модели 6 747 параметров, и все они заданы вручную; обучение не использовалось. В репозитории также лежит v1 PixelGen на 4 227 параметров.

✨ Ключевые особенности

• ✨ 6 747 параметров и 0 шагов обучения — веса прописаны прямо в __init__.
• 🖼 Промпт-грамматика: [small] [colour] <body> [with [colour] <accessory>], 21 тело и 7 аксессуаров.
• 🌀 21 866 уникальных спрайтов при seed 0 из 24 192 промптов.
• 💻 Выход 16×16 → 128×128 RGB, каждый пиксель из палитры PICO-8.
• 📚 v2 добавляет аксессуар, второй цвет, структурное освещение и размер; v1 — минимальная версия.

💻 Пример

from pixelgen.compose import PixelGen2

model = PixelGen2()
img = model.sprite("purple ghost with horns", seed=4, scale=8)
img.save("ghost.png")

Подробнее

Параметров: 6 747 (v2), буферов: 7 333, всего ≈56,3 KB fp32.
Обучение: 0 шагов, 0 оптимизаторов и backward passes.
Ограничения: только 21 тело и 7 аксессуаров; все формы горизонтально симметричны; один аксессуар на спрайт; seed меняет лишь несколько пикселей.
Качество — programmer art; структурное освещение помогает округлым формам и вредит тонким.