Update README.md
Browse files
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 |
+
|