candy machine

糖果機 - 手動圖像標記器

基於Web的手動圖像標記器用於訓練自定義穩定擴散Loras＆Dreambooth模型

Candy Machine是一個新生的圖像標記器，用於用.txt字幕文件手動標記小數據集（<1k images）：

• 可自定義的標籤佈局，用於一致的標籤

• 佔位符標籤模板：ie {type} clothes ，在添加標籤時可以指定{type}

• 內部形成的圖像編輯（作物，旋轉和翻轉水平）

• 創建新項目時使用wd-v1-4-convnext-tagger.v3的標籤建議

• 自動文件從.webp，.avif，.gif等轉換為.png

• 進度％餅圖- 在您想知道“我已經做好了嗎？！”的時候

• 鍵盤快捷鍵- 按'？ '對於列表

• 還有更多即將到來！

糖果機完全在您的本地機器上以“單人玩家”模式運行。沒有信息傳輸到任何第三方系統。

內置圖像編輯器

為什麼？

我受到有關如何最好地標記小型圖像數據集的Reddit帖子的啟發，主要關註一致性。

我也從不喜歡構建數據集的“貼心”部分 - 轉換圖像，重命名，裁剪等。我想快速，簡單，簡單，誰知道，甚至很有趣！

H/T到Binaryalley的原型也是如此。

要求：

• git
• Python 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/

使用GPU進行圖像標記

需要Cuda 12.2和Cudnn8.x。

1. 激活Python Venv。 （查看Inside 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字幕文件。）

功能管道

• 刪除標籤
• 按使用情況排序標籤
• 添加更多圖像！
• 編輯觸發單詞 /同義詞
• 圖像變焦
• 彩色選擇器 - >顏色名稱
• 查找並替換標籤（整個項目）
• 導出嚮導準備數據並為Kohya SS生成建議的設置
• 從導入分析自定義標籤佈局
• 保存的預設用於標籤佈局（即攝影，動漫，風格洛拉斯等）
• 去除水印（塗料？）
• 從數據集中檢測和刪除仿生的圖像等（以尊重版權持有人的意願並保護您的數據集）。
• 也許將來：自動化圖像搜索以查找更多圖像？

想貢獻嗎？

技術堆棧：

• 基本Python HTTP服務器 - 使用Virtual Env
• 建造JS和CSS捆綁包的包裹
• 打字稿
• Flowbite對UI組件的反應
• 用於造型的尾風CSS

在花費任何時間編寫代碼之前，請與您的提案打開一個問題，以便我們討論。謝謝！

虛擬設施設置（Windows）：

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

運行UI：

先決條件：節點和紗線。

 cd ui
yarn start

運行服務器：

 cd server
python main.py 

問題，評論，反饋？

請給我發送消息或打開問題。謝謝！