candy machine

Candy Machine - Manual Image Tagger

В веб-ручном тегатере для обучения пользовательской стабильной диффузии Loras & Dreambooth модели

Candy Machine - это зарождающийся теггер изображения для вручную, помечая небольшие наборы данных (<1K изображения) с файлами .txt .

• Настраиваемый макет тегов для последовательного тега

• Шаблоны тегов заполнителей : т.е. {type} clothes , где {type} может быть указан при добавлении тега

• Встроенное редактирование изображений (обрезка, вращение и перевернуть горизонтальный)

• Предложения тегов с использованием wd-v1-4-convnext-tagger.v3 при создании нового проекта

• Автоматическое преобразование файлов в .png от .webp, .avif, .gif и т. Д.

• Progress % Pie Dicart - в те времена, когда вам интересно : «Я сделал еще?!»

• Комплект клавиатуры - нажмите '?' для списка

• И скоро появится еще!

Candy Machine полностью работает на вашей локальной машине в режиме «одиночный игрок». Никакой информации не передается ни одной сторонней системе.

Встроенный редактор изображений

Почему?

Я был вдохновлен этим постом Reddit о том, как наилучшим образом пометить небольшие наборы данных, сосредоточившись в основном на последовательности.

Я также никогда не любил «скромные» части создания набора данных - преобразование изображений, переименование их, обрезка и т. Д. Я хочу сделать наборы данных быстрого, легкого и, кто знает, может быть, даже весело!

H/T к Binaryalley также для их прототипа.

Требования:

• Git
• Питон 3

Бесплатно для личного, некоммерческого использования.

Это программное обеспечение полностью бесплатно для личного, некоммерческого использования. Если вы используете его в бизнес -контексте или для коммерческого использования (т.е. для заработка на нем), пожалуйста, свяжитесь со мной, чтобы договориться о платной лицензии. Это поможет поддержать мои затраты на разработку.

Как запустить на Windows

 git clone [email protected]:mikeknapp/candy-machine.git
cd candy-machine
run

Окно браузера должно автоматически открыть: http://127.0.0.1:5000/

Как бежать на macOS/Unix

(Обратите внимание, я еще не проверял это, могут быть ошибки!)

 git clone [email protected]:mikeknapp/candy-machine.git
cd candy-machine
chmod +x run.sh
./run.sh

Окно браузера должно автоматически открыть: http://127.0.0.1:5000/

Использование графического процессора для тега изображения

Требуется CUDA 12.2 и Cudnn8.x.

1. Активируйте Python Venv. (Загляните внутрь run.bat для подсказков о том, как это сделать.)

2. Установите onnxruntime-gpu

 pip install onnxruntime-gpu --extra-index-url 
    https://aiinfra.pkgs.visualstudio.com/PublicPackages/_packaging/onnxruntime-cuda-12/pypi/simple/ 

Известные проблемы

• Не могу редактировать макет тега проекта или макет тега по умолчанию внутри приложения. (Обходной путь: редактировать default_categories.json в каталоге сервера или categories.json проекта.

• Не могу добавить больше изображений в проект! (Обходной путь: Создайте новый проект. Я знаю, не идеален.)

• Не могу редактировать триггерное слово / синонимы. (Обходной путь: отредактируйте проект config.json и все существующие файлы подписи .txt.)

Функция трубопровода

• Удалить теги
• Сортировать теги при использовании
• Добавьте больше изображений!
• Редактировать Trigger Word / Synonyms
• Изображение Zoom
• Цветовой сборщик -> название цвета
• Найдите и замените теги (через проект)
• Мастер экспорта для подготовки данных и создания рекомендуемых настроек для Kohya SS
• Настроить макет тега из анализа импорта
• Сохраненные предустановки для макетов тегов (т.е. Фотография, аниме, стиль Лорас и т. Д.)
• Удаление водяного знака (внедрение?)
• Обнаружение и удаление изображений, выпущенных паслей, и т. Д. Из набора данных (для уважения желаний держателей авторских прав и защиты вашего набора данных).
• Может быть, в будущем: автоматический поиск изображений, чтобы найти больше изображений?

Хотите внести свой вклад?

Технический стек:

• Basic Python Http Server - с помощью Virtual Env
• Посылка для строительства пакетов JS и CSS
• Машинопись
• Реакция Flowbite для компонентов пользовательского интерфейса
• Tailwind CSS для стиля

Прежде чем потратить в любое время на написание кода, откройте проблему с вашим предложением, чтобы мы могли обсудить. Спасибо!

Virtual Env Setup (Windows):

 python -m venv venv
call .venvScriptsactivate
pip install -r requirements.txt

Запустите пользовательский интерфейс:

Предварительные условия: узел и пряжа.

 cd ui
yarn start

Запустите сервер:

 cd server
python main.py 

Вопросы, комментарии, обратная связь?

Пожалуйста, пришлите мне сообщение или откройте проблему. Спасибо!