DiffusionToolkit

DiffusionToolkit

Metadata-indexer and Viewer for AI-generated images

Stars: 728

Visit
 screenshot

Diffusion Toolkit is an image metadata-indexer and viewer for AI-generated images. It helps you organize, search, and sort your ever-growing collection. Key features include: - Scanning images and storing prompts and other metadata (PNGInfo) - Searching for images using simple queries or filters - Viewing images and metadata easily - Tagging images with favorites, ratings, and NSFW flags - Sorting images by date created, aesthetic score, or rating - Auto-tagging NSFW images by keywords - Blurring images tagged as NSFW - Creating and managing albums - Viewing and searching prompts - Drag-and-drop functionality Diffusion Toolkit supports various image formats, including JPG/JPEG, PNG, WebP, and TXT metadata. It also supports metadata formats from popular AI image generators like AUTOMATIC1111, InvokeAI, NovelAI, Stable Diffusion, and more. You can use Diffusion Toolkit even on images without metadata and still enjoy features like rating and album management.

README:

Diffusion Toolkit

Diffusion Toolkit is an image metadata-indexer and viewer for AI-generated images. It aims to help you organize, search and sort your ever-growing collection.

Usage

Usage should be pretty straightforward, but there are a lot of tips and tricks and shortcuts you can learn. See the documentation for Getting Started

Thanks to Bill Meeks for putting together a demonstration video.

Organize your AI Images

Installation

  • Currently runs on Windows only
  • Download the latest release
    • Look for > Assets under the latest release, expand it, then grab the zip file Diffusion.Toolkit.v1.x.zip.
  • Unzip all the files to a folder
  • You may need to install the .NET 6 Desktop Runtime if you haven't already

Build from source

Prerequisites

  • Requires Visual Studio 2022
  • .NET 6 SDK (includes the desktop runtime)

Building

  • Clone this repository
  • Run publish.cmd

A folder named build will be created, containing all the necessary files.

Features

  • Scan images, store and index prompts and other metadata (PNGInfo)
  • Search for your images
    • Using a simple query
    • Using the filter
  • View images and the metadata easily
    • Toggle PNGInfo
  • Tag your images
    • Favorite
    • Rating (1-10)
    • NSFW
  • Sort images
    • by Date Created
    • by Aesthetic Score
    • by Rating
  • Auto tag NFSW by keywords
  • Blur images tagged as NSFW
    • NSFW
  • Albums
    • Select images, right-click > Add to Album
    • Drag and drop images to albums
  • Folder View
  • View and search prompts
    • List Prompts and usage
    • List Negative Prompts and usage
    • List images associated with prompts
  • Drag and Drop
    • Drag and drop images to another folder to move (CTRL-drag to copy)

Supported formats

  • JPG/JPEG + EXIF
  • PNG
  • WebP
  • .TXT metadata

Supported Metadata formats

  • AUTOMATIC1111 and A1111-compatible metadata such as
    • Tensor.Art
    • SDNext
  • InvokeAI (Dream/sd-metadata/invokeai_metadata)
  • NovelAI
  • Stable Diffusion
  • EasyDiffusion
  • RuinedFooocus
  • Fooocus
  • FooocusMRE
  • Stable Swarm

You can even use it on images without metadata and still use the other features such as rating and albums!

Screenshots

Screenshot 2024-02-09 183808

Screenshot 2024-02-09 183625

Keyboard Shortcuts

Make sure the thumbnail view or the preview pane has the focus if the shortcuts don't work.

Shortcut Action Notes
1..9, 0 Rate 1 - 10 In thumbnail or preview
I Show/Hide (PNG)Info In thumbnail or preview
F Tag Favorite In thumbnail or preview
N Tag NSFW In thumbnail or preview
B Toggle Blur NSFW In thumbnail or preview
X/Del Tag for Deletion In thumbnail or preview
Ctrl+X/Del Remove Entry Remove image from database, but do not delete
Ctrl+C Copy File to Clipboard
F3 Show/Hide Navigation Pane
F4 Show/Hide Preview
Ctrl+Shift+F Toggle Fit to Preview
Ctrl+1 Folders View
Ctrl+2 Album View
Ctrl+3 Diffusions View
Ctrl+4 Favorites Page
Ctrl+5 Recycle Bin
Ctrl+6 Prompts View
Ctrl+0 Reset Zoom in Preview Pane or Image Viewer
Ctrl+R Scan folders for new images
Alt+Home First Page
Alt+PageUp Previous Page
Alt+PageDown Next Page
Alt+End Last Page
F6 Set focus on search bar

FAQ

How do I view my image's metadata (PNGInfo)?

With the Preview Pane visible, press I in the thumbnail view or with the Preview Pane in focus to show or hide the metadata. You can also click the eye icon at the botton right of the Preview Pane.

What is Rebuild Metadata and when should I use it?

Rebuild Metadata will rescan all your images and update the database with any new or updated metadata found. It doesn't affect your custom tags (rating, favorite, nsfw).

You only need to Rebuild Metadata if a new version of Diffusion Toolkit comes out with support for metadata that exists in your existing images.

Can I move my images to a different folder?

I you want to move your images to a different folder, but still within a Diffusion folder, you should use the right-click > Move command. This allows Diffusion Toolkit to handle the moving of images, and know to keep all the Diffusion Toolkit metadata (Favorites, Rating, NSFW) intact while moving.

If you use Explorer or some other application to move the files, but still under the Diffusion folders, when you Rescan Folders or Rebuild Images Diffusion Toolkit will detect that the images have been removed, then will detect new files added. You will lose any Favorites, Ratings or other Toolkit-specific information.

Buy me a coffee

I've been told that people should be allowed to buy me a coffee

Beer works too (PayPal)

For Tasks:

Click tags to check more tools for each tasks

For Jobs:

Alternative AI tools for DiffusionToolkit

Similar Open Source Tools

For similar tasks

For similar jobs