ONNX
imgutils-models / README.md
narugo's picture
Update README.md
923738b
|
raw
history blame
1.02 kB
metadata
license: mit
datasets:
  - deepghs/chafen_arknights
  - deepghs/monochrome_danbooru
metrics:
  - accuracy

imgutils-models

This repository includes all the models in deepghs/imgutils.

LPIPS

This model is used for clustering anime images (named 差分 in Chinese), based on richzhang/PerceptualSimilarity, trained with dataset deepghs/chafen_arknights(private).

When threshold is 0.45, the adjusted rand score 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).