BLIP VS Dreambooth-Stable-Diffusion

I suggest trying BLIP for this. I've had really good results from that.

https://github.com/salesforce/BLIP

I built a tiny Python CLI wrapper for it to make it easier to try: https://github.com/simonw/blip-caption

You can download the source and run it yourself from here: https://github.com/salesforce/BLIP

Then there's also BLIP (Bootstrapping Language-Image Pre-training).

Or BLIP

GitHub - salesforce/BLIP: PyTorch code for BLIP: Bootstrapping Language-Image Pre-training for Unified Vision-Language Understanding and Generation

From what I understand, what's most important to you is having a model that's already trained on something, rather than the architecture. Yolo is probably fine, as would be some of the older ones. You should be able to find a model that's been pretrained on COCO - you can look at see what classes are included. I don't know if there are other broadly trained models available that will serve your purpose. What I'd do is just run your picture through a COCO trained object detection model and see if the annotations do what you want.

Though backing up a bit, there are also image captioning models that may better do what you want to do for organizing your photos. I'm not really familiar with any - though I did come across BLIP the other day but I haven't used it: https://github.com/salesforce/BLIP

This may be a better way to get at what you want

git clone https://github.com/salesforce/BLIP.git repositories/BLIP

# Copy models as described in README cp ~/Downloads/model.ckpt . cp ~/Downloads/GFPGANv1.3.pth . # Clone other repos as mentioned in README mkdir repositories git clone https://github.com/CompVis/stable-diffusion.git repositories/stable-diffusion git clone https://github.com/CompVis/taming-transformers.git repositories/taming-transformers git clone https://github.com/sczhou/CodeFormer.git repositories/CodeFormer git clone https://github.com/salesforce/BLIP.git repositories/BLIP export NIXPKGS_ALLOW_UNFREE=1 nix-shell default.nix pip install torch --extra-index-url https://download.pytorch.org/whl/cu113 # Also from linux instructions. Can probably be added to default.nix python webui.py

Also check img2text (basically to prompt): https://github.com/salesforce/BLIP

Tons of stuff here, no? https://github.com/JoePenna/Dreambooth-Stable-Diffusion/

Joe Penna's Dreambooth (Tutorial|24GB) Most popular DB repo with great results.

1) The https://github.com/JoePenna/Dreambooth-Stable-Diffusion instructions say to use the 1.5 checkpoints - is that the latest? Can I use the 2+ models or?

I quickly turned to the second library, https://github.com/JoePenna/Dreambooth-Stable-Diffusion, because its readme was very encouraging, and its results were the best. Unfortunately, to use it on Colab, you need to sign up for Colab Pro to use advanced GPUs (at least 24GB of VRAM), and training a model requires at least 14 compute units. As a poor Chinese person, I could only buy Colab Pro from a proxy. The results from JoePenna/Dreambooth-Stable-Diffusion were fantastic, and the preparation was straightforward, requiring only <=20 512*512 photos without writing captions. I used it to create many beautiful photos.

I've found EveryDream to be brilliant and have switched from JoePenna's Dreambooth because I've found I get better results so long as I provide good captions for all the images, even if preparing the dataset takes 3x as long (took me 2 hours to crop and label the 54 images).

From what I know you can train with whatever size you want. But you need software that will support it. For example, ShivamShrirao/diffusers repo seems to allow a change of dimension. Also, you need HW that would support the training, because bigger images need more VRAM, for example,Joe Penna repo is using ~23GB with 512x512px so probably it's not a valid option. But the ShivamShrirao repo has optimizations that allow to run it with less VRAM.

This is a good starting place: https://github.com/JoePenna/Dreambooth-Stable-Diffusion