|
--- |
|
license: mit |
|
datasets: |
|
- deepghs/chafen_arknights |
|
- deepghs/monochrome_danbooru |
|
metrics: |
|
- accuracy |
|
--- |
|
|
|
# imgutils-models |
|
|
|
This repository includes all the models in [deepghs/imgutils](https://github.com/deepghs/imgutils). |
|
|
|
## LPIPS |
|
|
|
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). |
|
|
|
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`. |
|
|
|
File lists: |
|
* `lpips_diff.onnx`, feature difference. |
|
* `lpips_feature.onnx`, feature extracting. |
|
|
|
## Monochrome |
|
|
|
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). |
|
|
|
|
