🍞 TextBaker

⚠️ Disclaimer: This project was developed with the assistance of AI tools (GitHub Copilot). While efforts have been made to ensure quality, it may still contain errors or bugs. Use at your own discretion and feel free to report any issues.

Synthetic Text Dataset Generator for OCR Training

TextBaker generates synthetic text images by combining character datasets with backgrounds and applying transformations. Perfect for training OCR models and data augmentation.

🖼️ Example Outputs

Basic	Transformed	Colored	Background	Texture	Full Pipeline

📖 See Examples Documentation for code samples.

✨ Features

• 🎨 GUI Application - Interactive interface for real-time text generation
• ✏️ Custom Character Drawing - Draw and save custom characters directly in the app
• ✂️ Character Segmentation - Extract characters from images using polygon selection
• 🖥️ CLI Tool - Batch processing from command line
• 📚 Python Library - Programmatic API for integration
• 🔄 Transformations - Rotation, perspective, scale, shear
• 🎭 Textures & Backgrounds - Apply overlays and composite on images
• 🔧 YAML/JSON Configs - Save and load configurations

📦 Installation

pip install textbaker

Or from source:

git clone https://github.com/q-viper/text-baker.git
cd text-baker
pip install -e .

🚀 Quick Start

GUI

textbaker

CLI

# Generate specific texts
textbaker generate "Hello" "World" -d ./dataset -o ./output

# Generate random samples with transforms
textbaker generate -n 100 --seed 42 -r "-15,15" -b ./backgrounds

Python

from textbaker import TextGenerator, GeneratorConfig

generator = TextGenerator()
result = generator.generate("Hello")
generator.save(result)

📖 See full documentation for detailed usage.

📁 Dataset Structure

dataset/
├── A/
│   ├── sample1.png
│   └── sample2.png
├── B/
│   └── ...
└── 0/
    └── ...

📖 Documentation

• Installation & Quick Start
• Examples & Code Samples
• Configuration Reference
• CLI Reference
• API Reference

🧪 Development

git clone https://github.com/q-viper/text-baker.git
cd text-baker
pip install -e ".[dev]"

# Run tests
pytest tests/ -v

# Run linting
ruff check . && ruff format .

📄 License

MIT License - see LICENSE for details.

🤝 Contributing

Contributions welcome! Please fork, create a feature branch, and submit a PR.

👤 Author

Ramkrishna Acharya (@q-viper)

Built with PySide6, OpenCV, and Typer