AI & ML interests

Computer Vision Technology and Data Collection for Anime Waifu

Organization Card
About org cards

Who We Are

We are a team focused on automating the training process of stable diffusion models (LoRA, Lycoris, Pivotal Tuning, etc.).

Our goal is to automate the entire pipeline, including data acquisition, data filtering, training, step selection, and platform deployment, saving manpower and optimizing model training quality to the maximum.

Our team comprises a Ph.D. in Software Engineering, a Ph.D. candidate in Computer Vision, professionals in art and design, and several AI waifu enthusiasts.

We are a purely non-profit team, and all our work is completely open, without any form of charge.

Our Achievements

We have conducted multiple iterations on Pivotal Tuning training technology (training with a LoRA and one or more pt files). The progress is documented in the following technical blogs:

Models we have trained or hosted can be found at:

Quantitative analysis from the blog "Result Analysis of v1.4 Training Automation" shows that v1.4 training automation has achieved a quite impressive level in both quality and quantity, but there's still room for further improvement. This is an ongoing effort for v1.5 and future versions.

Our Technical Outputs


Project Link:

Project Documentation:

This is a library for various common operations on anime images, including but not limited to:

  • Tachie (Difference) Detection and Clustering
  • Contrastive Character Image Pretraining
  • Object Detection
  • Edge Detection / Lineart Generation
  • Monochrome Image Detection
  • Image Rating
  • Truncated Image Check
  • Image Tagging
  • Character Extraction

Check out the documentation for more features.


Project Link:

A data pipeline framework based on dghs-imgutils, supporting:

  • Fast data retrieval (local disk, danbooru, pixiv, zerochan, etc.)
  • Swift data filtering (comic exclusion, monochrome image exclusion, multi-character image exclusion, irrelevant character exclusion, etc.)
  • Rapid data saving (local, cloud; with metadata, saved in stable diffusion dataset format, etc.)
  • Quick building of processing pipelines (connecting multiple aforementioned stages)

Note: This tool is currently a work in progress, although it's in use. It hasn't been released on PyPI and lacks comprehensive documentation. These aspects will be addressed soon.

Model Zoo

We manage our models and datasets on Huggingface:

Anything More?

In fact, our plans go beyond what's mentioned here. Other tools are continuously improving and will soon be released. Stay tuned!