| <h1 align="center">OmniGen: Unified Image Generation</h1> | |
| <p align="center"> | |
| <a href=""> | |
| <img alt="Build" src="https://img.shields.io/badge/Project%20Page-OmniGen-yellow"> | |
| </a> | |
| <a href="https://arxiv.org/abs/2409.11340"> | |
| <img alt="Build" src="https://img.shields.io/badge/arXiv%20paper-2409.11340-b31b1b.svg"> | |
| </a> | |
| <a href="https://huggingface.co/spaces/Shitao/OmniGen"> | |
| <img alt="License" src="https://img.shields.io/badge/HF%20Demo-🤗-lightblue"> | |
| </a> | |
| <a href="https://huggingface.co/Shitao/OmniGen-v1"> | |
| <img alt="Build" src="https://img.shields.io/badge/HF%20Model-🤗-yellow"> | |
| </a> | |
| </p> | |
| <h4 align="center"> | |
| <p> | |
| <a href=#2-news>Credits for Quantized version</a> | | |
| <a href=#3-methodology>Methodology</a> | | |
| <a href=#4-what-can-omnigen-do>Capabilities</a> | | |
| <a href="#license">License</a> | | |
| <a href="#citation">Citation</a> | |
| <p> | |
| </h4> | |
| ## 1. Overview | |
| OmniGen is a unified image generation model that can generate a wide range of images from multi-modal prompts. It is designed to be simple, flexible and easy to use. We provide [inference code](#5-quick-start) so that everyone can explore more functionalities of OmniGen. | |
| Existing image generation models often require loading several additional network modules (such as ControlNet, IP-Adapter, Reference-Net, etc.) and performing extra preprocessing steps (e.g., face detection, pose estimation, cropping, etc.) to generate a satisfactory image. However, **we believe that the future image generation paradigm should be more simple and flexible, that is, generating various images directly through arbitrarily multi-modal instructions without the need for additional plugins and operations, similar to how GPT works in language generation.** | |
| Due to the limited resources, OmniGen still has room for improvement. We will continue to optimize it, and hope it inspire more universal image generation models. You can also easily fine-tune OmniGen without worrying about designing networks for specific tasks; you just need to prepare the corresponding data, and then run the [script](#6-finetune). Imagination is no longer limited; everyone can construct any image generation task, and perhaps we can achieve very interesting, wonderful and creative things. | |
| If you have any questions, ideas or interesting tasks you want OmniGen to accomplish, feel free to discuss with us: [email protected], [email protected], [email protected]. We welcome any feedback to help us improve the model. | |
| ## 2. Credits for Quantized version | |
| - https://github.com/Manni1000 | |
| ## 3. Methodology | |
| You can see details in our [paper](https://arxiv.org/abs/2409.11340). | |
| ## 4. What Can OmniGen do? | |
| OmniGen is a unified image generation model that you can use to perform various tasks, including but not limited to text-to-image generation, subject-driven generation, Identity-Preserving Generation, image editing, and image-conditioned generation. **OmniGen don't need additional plugins or operations, it can automatically identify the features (e.g., required object, human pose, depth mapping) in input images according the text prompt.** | |
| We showcase some examples in [inference.ipynb](inference.ipynb). And in [inference_demo.ipynb](inference_demo.ipynb), we show an interesting pipeline to generate and modify a image. | |
| Here is the illustration of OmniGen's capabilities: | |
| - You can control the image generation flexibly via OmniGen | |
|  | |
| - Referring Expression Generation: You can generate images by simply referring to objects, and OmniGen will automatically recognize the required objects in the image. | |
|  | |
| If you are not entirely satisfied with certain functionalities or wish to add new capabilities, you can try [fine-tuning OmniGen](#6-finetune). | |
| ## 5. Quick Start | |
| ### Please refer youtube video for installation | |
| https://www.youtube.com/watch?v=9ZXmXA2AJZ4 | |
| ## License | |
| This repo is licensed under the [MIT License](LICENSE). | |
| ## Citation | |
| If you find this repository useful, please consider giving a star ⭐ and citation | |
| ``` | |
| @article{xiao2024omnigen, | |
| title={Omnigen: Unified image generation}, | |
| author={Xiao, Shitao and Wang, Yueze and Zhou, Junjie and Yuan, Huaying and Xing, Xingrun and Yan, Ruiran and Wang, Shuting and Huang, Tiejun and Liu, Zheng}, | |
| journal={arXiv preprint arXiv:2409.11340}, | |
| year={2024} | |
| } | |
| ``` | |