This project provides a GUI for viewing images, editing associated text files, and using tag autocomplete functionality.
- View images and edit associated text files
- Tag autocomplete using various tag sources
- Automatic download of missing tag files - Big thanks to DominikDoom (https://github.com/DominikDoom) the tags files are from a1111-sd-webui-tagcomplete, which is licensed under the MIT License.
- Python 3.7+
- Pillow (PIL)
- tkinter
- requests
-
Clone this repository:
git clone https://github.com/yourusername/your-repo-name.git cd your-repo-name
-
(Optional) Create and activate a virtual environment:
Using
venv
:python -m venv env source env/bin/activate # On Windows use `env\Scripts\activate`
Using
conda
:conda create --name dataset-editor-env python=3.7 conda activate dataset-editor-env
-
Install the required packages:
pip install -r requirements.txt
-
Run the script:
python edit-dataset.py
-
On the first run, the script will automatically download necessary tag files.
-
Use the GUI to navigate through images, edit text, and use tag autocomplete.
- The tag files are not included in this repository and will be downloaded automatically when needed.
- Big thanks to DominikDoom (https://github.com/DominikDoom) the tags files are from a1111-sd-webui-tagcomplete, which is licensed under the MIT License.
- If you want to use your own tag files, place them in a
tags
folder in the project directory.
Your project should be organized as follows:
your-project-folder/
│
├── image_tagger.py # Your main script
├── .gitignore
├── config.json # This file will be created by the script
├── README.md
├── requirements.txt
└── tags/ # This folder will be created by the script