wdsort

wdsort is a command-line interface (CLI) tool designed to help users sort images into folders using the WaifuDiffusion model.

This program utilizes the model from Hugging Face - SmilingWolf/wd-v1-4-swinv2-tagger-v2 to categorize images based on their content.

Features

• Image Sorting: Automatically sort images into designated folders based on their content.
• Easy to Use: Simple CLI commands for easy access to users of any skill level.

Prerequisites

Before using wdsort, ensure you have the following:

• Python 3.7 or higher
• Required Python libraries (listed in requirements.txt)

Installation

1. Clone the repository:

   git clone https://github.com/yourusername/wdsort.git

2. Navigate to project directory:

   cd wdsort

3. Create virtual environment and activate:

   python -m venv .venv && .venv\Scripts\Activate

4. Install requirements

   pip install -r requirements.txt

5. Download model.onnx, selected_tags.csv, and config.json from Hugging Face.s

6. Place these files in model directory

Using the Program

1. To Scan by Folder

   With a bias towards the first character identified. Otherwise, the first tag found.

   python wdsort.py --folder "path/to/folder"

2. To Assess Tags of Single Image

   Output all tags for a single image to console.

   python wdsort.py --scan "path/to/image"

3. Sort By Specific Tag

   Will sort by specified "tag" and any combination where that tag is used (e.g., eat and eating)

   python wdsort.py --bytag "tag"

Note

This is not the perfect solution for organizing images. I am new to programming in general, so there is still a lot of work to do. Primary credit goes to SmilingWolf for offering the model on HugggingFace.

If this program provides any legal issues to anybody, please let me know and I will take it down promptly.

To Do

[] Add more customized sorting criteria