Skip to content

An advanced Jupyter Notebook for creating precise datasets tailored to stable Diffusion LoRa training. Automate face detection, similarity analysis, and curation, with streamlined exporting, utilizing cutting-edge models and functions.

License

Notifications You must be signed in to change notification settings

Maximax67/LoRA-Dataset-Automaker

Repository files navigation

LoRA Dataset Automaker

License Open in Colab

Welcome to the Dataset Automaker Git repository housing an innovative and reliable Jupyter Notebook designed to facilitate the creation of datasets for training Stable Diffusion LoRAs.

image.svg

🔥 Key Features

Key Features:

  • Automated Anime Screencap Collection: Easily search and download anime screencaps from fancaps.net.

  • Advanced Data Refinement: Utilize the FiftyOne app and clip-vit-torch model to meticulously filter and curate your dataset for optimal quality.

  • Precise Face Detection: Leverage face detection models for accurate anime face identification (zymk9/yolov5_anime, ultralytics/yolov5).

  • Character Similarity Analysis: Calculate pairwise similarity distances between original and example faces, enabling targeted dataset curation with user-defined thresholds.

  • User-Guided Curation: Utilize the intuitive FiftyOne app for manual dataset refinement, ensuring precision.

  • Intelligent Tagging: Effortlessly tag results using code by kohya-colab and sd-scripts, enhancing organization and analysis.

  • Seamless Export: Conveniently zip and download your curated dataset.

🛠️ Getting Started

Local:

  1. Clone or download the repository to your local machine.
  2. Launch Jupyter Notebook and open the Dataset_Automaker.ipynb notebook.
  3. Follow the comprehensive guide within the notebook to harness the power of automated data generation.

Google colab:

  1. Open Dataset_Automaker.ipynb in Google colab.
  2. Follow the guide in the notebook

📧 Contact me

Civitai: Maximax67
Telegram: @Maximax67
Github: Maximax67
Gmail: maximax6767@gmail.com

About

An advanced Jupyter Notebook for creating precise datasets tailored to stable Diffusion LoRa training. Automate face detection, similarity analysis, and curation, with streamlined exporting, utilizing cutting-edge models and functions.

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published