ONNX
narugo commited on
Commit
923738b
1 Parent(s): 25ec5a2

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +24 -0
README.md CHANGED
@@ -1,3 +1,27 @@
1
  ---
2
  license: mit
 
 
 
 
 
3
  ---
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
  ---
2
  license: mit
3
+ datasets:
4
+ - deepghs/chafen_arknights
5
+ - deepghs/monochrome_danbooru
6
+ metrics:
7
+ - accuracy
8
  ---
9
+
10
+ # imgutils-models
11
+
12
+ This repository includes all the models in [deepghs/imgutils](https://github.com/deepghs/imgutils).
13
+
14
+ ## LPIPS
15
+
16
+ This model is used for clustering anime images (named `差分` in Chinese), based on [richzhang/PerceptualSimilarity](https://github.com/richzhang/PerceptualSimilarity), trained with dataset [deepghs/chafen_arknights(private)](https://huggingface.co/datasets/deepghs/chafen_arknights).
17
+
18
+ When threshold is `0.45`, the [adjusted rand score](https://scikit-learn.org/stable/modules/generated/sklearn.metrics.adjusted_rand_score.html) can reach `0.995`.
19
+
20
+ File lists:
21
+ * `lpips_diff.onnx`, feature difference.
22
+ * `lpips_feature.onnx`, feature extracting.
23
+
24
+ ## Monochrome
25
+
26
+ These model is used for monochrome image classification, based on CNNs and Transformers, trained with dataset [deepghs/monochrome_danbooru(private)](https://huggingface.co/datasets/deepghs/monochrome_danbooru).
27
+