rohit-gupta / diffusion-gen

Use stable diffusion to generate images guided by text

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

diffusion-gen

Use stable diffusion to generate synthetic images to train a drone object detector

INSTALLATION

git clone https://github.com/CompVis/stable-diffusion
cd stable-diffusion
conda env create -f environment.yaml
conda activate ldm

conda install -c conda-forge diffusers

USAGE

python main.py [filename.jpg] --prompt "prompt text here"

NOTE: do not change the order of the arguments

You can choose to skip the prompt, in which case it will just use "a high resolution photograph"

About

Use stable diffusion to generate images guided by text

License:MIT License


Languages

Language:Python 100.0%