Merge pull request #123 from deepghs/dev/tconv

dev(narugo): add convert function from embeddings to understandable result of taggers
deepghs · Nov 8, 2024 · c1bca12 · c1bca12
2 parents 0627839 + 555b541
commit c1bca12
Show file tree

Hide file tree

Showing 10 changed files with 343 additions and 51 deletions.
diff --git a/docs/source/api_doc/tagging/wd14.rst b/docs/source/api_doc/tagging/wd14.rst
@@ -12,3 +12,11 @@ get_wd14_tags
 .. autofunction:: get_wd14_tags
 
 
+
+convert_wd14_emb_to_prediction
+----------------------------------------------
+
+.. autofunction:: convert_wd14_emb_to_prediction
+
+
+
diff --git a/docs/source/api_doc/utils/func.rst b/docs/source/api_doc/utils/func.rst
@@ -0,0 +1,14 @@
+imgutils.utils.func
+====================================
+
+.. currentmodule:: imgutils.utils.func
+
+.. automodule:: imgutils.utils.func
+
+
+sigmoid
+-------------------------
+
+.. autofunction:: sigmoid
+
+
diff --git a/docs/source/api_doc/utils/index.rst b/docs/source/api_doc/utils/index.rst
@@ -9,4 +9,5 @@ imgutils.utils
 .. toctree::
     :maxdepth: 3
 
+    func
     onnxruntime
diff --git a/imgutils/tagging/__init__.py b/imgutils/tagging/__init__.py
@@ -16,4 +16,4 @@
 from .mldanbooru import get_mldanbooru_tags
 from .order import sort_tags
 from .overlap import drop_overlap_tags
-from .wd14 import get_wd14_tags
+from .wd14 import get_wd14_tags, convert_wd14_emb_to_prediction