/comfyui_tag_fillter

ComfyUI custom node for filtering tags based on categories such as pose, gesture, action, emotion, expression, camera, angle, sensitive, and liquid.

Primary LanguagePythonMIT LicenseMIT

comfyui_tag_filter

WD14Tagger などが出力したタグを、特定のカテゴリに分けて、カテゴリごとにフィルタリングして返す ComfyUI のカスタムノードです。

TagFilter

特定のカテゴリに属するタグだけを抽出します。 smile や grin などは、expression というカテゴリです。 色のついているものは color カテゴリです。

image

include_categories は、含めたいカテゴリ名をカンマ区切りで指定します。

include_categories に expression と入力すると、上のチェックボックスをクリックしたのと同じ結果になります。

exclude_categories は逆に取り除きたいカテゴリを指定できます。服のタグだけ取りたいが色の指定は不要な場合に、include_categories に cloth と入れて、exclude_categories に color と打ちます。

何がどのカテゴリに分類されているかは、tag_category.json を見て確認してください。

tag_category.json は AI により自動仕分けされた、タグのカテゴリ分けファイルです。誤った内容が含まれる場合もあります(細々と手動で調整を進めています)

TagRemover

プロンプトの中から指定されたタグを削除したい時に使います。 TagFilter と合わせて使うと、表情のタグだけ消す、色(color)タグだけ消す、といった事が出来ます。

image

TagReplace (動作検証中、うまく動かないかも)

同じカテゴリのタグを置き換えます。

例えば long hair というタグが含まれるプロンプトを入力し、置き換えるタグに twintails を指定すると、long hair を twintails に置き換える事が出来ます。

どれぐらい近いタグを置き換えるかを、0.0 ~ 1.0 で指定します。1.0 を指定するとカテゴリとして完全一致した場合にのみタグが置換されます。0.3 ぐらいがちょうどいいです。うまく置換されない場合は値を下げてください。

image