Home

Awesome

Dataset Tag Editor

Stand alone version is here: This may be better to avoid some known bugs.

Due to gradio update on webUI, the latest version don't support old webUI.
Please see Releases page and check the compatibility with webUI you are using.

日本語 Readme

This is an extension to edit captions in training dataset for Stable Diffusion web UI by AUTOMATIC1111.

It works well with text captions in comma-separated style (such as the tags generated by DeepBooru interrogator).

Caption in the filenames of images can be loaded, but edited captions can only be saved in the form of text files.

Installation

Extensions tab on WebUI

Copy https://github.com/toshiaki1729/stable-diffusion-webui-dataset-tag-editor.git into "Install from URL" tab.

Also, if you see this extension listed, you can install from "Available" tab with a single click.

Please note that if you update this extension from "Extensions" tab, you will need to restart web UI to reload completely.

Install Manually

To install, clone the repository into the extensions directory and restart the web UI.

On the web UI directory, run the following command to install:

git clone https://github.com/toshiaki1729/stable-diffusion-webui-dataset-tag-editor.git extensions/dataset-tag-editor

Features

Note. "tag" means each blocks of caption separated by commas.

Usage

  1. Make dataset using web UI
    • better to use already cropped images
  2. Load them
    • use interrogator if needed
  3. Edit their captions
    • filter images you want to edit by tags in "Filter by Tags" tab
    • filter images manually in "Filter by Selection" tab
    • replace/remove tags or append new tags in "Batch Edit Captions" tab
    • edit captions individually in "Edit Caption of Selected Image" tab
      • you also can use interrogator here
    • move/delete files in "Move or Delete Files" tab if needed
  4. Click "Save all changes" button

By the way, how can I edit tags quickly?

Basic workflow is as follows:

  1. Filter images
  2. Batch edit

Please note that all batch editing will be applyed only to displayed images (=filtered images).

1. Which filter is appropriate?

2. How can I edit as I want?

Trouble shooting

Cannot see any image in dataset and saying "All files must contained within the Gradio python app working directory…"

Set folder to store temporaly image in the "Settings" tab. Input path in "Directory to save temporary files" and check "Force using temporary file…"

So laggy when opening many images or extremely large image

Check "Force image gallery to use temporary files" and input number in "Maximum resolution of ..." in the "Settings" tab.
It may not work with dataset with millions of images.
If it doesn't work, please consider using stand alone version.

Description of Display

Moved to here