DreamBooth model for for anatomical drawings

This is a Stable Diffusion model fine-tuned on the sbtstyle concept with DreamBooth. It can be used by modifying the instance_prompt: an anatomical drawing in the style of sbtstyle

This model was created as part of the DreamBooth Hackathon 🔥. Visit the organisation page for instructions on how to take part!

Examples and data

Some more examples generated by others:

Dataset

The full dataset used to train on can be found here: Sanderbaduk/sobotta-anatomical-dataset. These images and others like it can be found on wikimedia commons.

If you use this directly you will get an error as the raw images have different numbers of channels. Include a grayscale step to fix this and ensure that classic feel.

self.transforms = transforms.Compose(
    [
        transforms.Resize(size),
        transforms.Grayscale(num_output_channels=3), 
        transforms.CenterCrop(size),
        transforms.ToTensor(),
        transforms.Normalize([0.5], [0.5]),
    ]
)

Usage

from diffusers import StableDiffusionPipeline

pipeline = StableDiffusionPipeline.from_pretrained('Sanderbaduk/science-sobotta-anatomical-drawings')
image = pipeline().images[0]
image