Update README.md

README.md

@@ -14,6 +14,43 @@ tags:
 
 Models of experiment: https://github.com/deepghs/tagger_embedding_aligner
 
+```python
+import numpy as np
+
+from imgutils.tagging import get_wd14_tags, convert_wd14_emb_to_prediction, denormalize_wd14_emb
+from test.testings import get_testfile
+
+embedding, (r, g, c) = get_wd14_tags(
+    get_testfile('nian.png'),
+    fmt=('embedding', ('rating', 'general', 'character')),
+)
+# normal tag results
+print('Expected result:')
+print(r)
+print(g)
+print(c)
+
+# normalize embedding
+embedding = embedding / np.linalg.norm(embedding)
+# bad tag results
+br, bg, bc = convert_wd14_emb_to_prediction(embedding)
+print('Bad results due to the embedding normalization:')
+print(br)
+print(bg)
+print(bc)
+
+# denormalize this embedding
+output = denormalize_wd14_emb(embedding)
+print(output.shape)
+
+# should be similar to r, g, c, approx 1e-3 error
+rating, general, character = convert_wd14_emb_to_prediction(output)
+print('De-normalized result:')
+print(rating)
+print(general)
+print(character)
+```
+
 |         Name          |   Tagger    |  Embedding Width  |  Tags Count  |   FLOPS   |  Params  |  EMB Cosine  |  EMB Norm  |  Pred Loss  |  Pred MSE  |
 |:---------------------:|:-----------:|:-----------------:|:------------:|:---------:|:--------:|:------------:|:----------:|:-----------:|:----------:|
 |   ViT_v3_mnum2_all    |   ViT_v3    |        768        |    10861     | 0.000398G |  0.40M   |      1       |   0.1712   |  0.004306   | 2.116e-08  |