narugo1992
commited on
Commit
·
923738b
1
Parent(s):
25ec5a2
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 |
+
|