返回模型
说明文档
imgutils-models
本仓库包含 deepghs/imgutils 中的所有模型。
LPIPS
该模型用于聚类动漫图片（中文称为差分），基于 richzhang/PerceptualSimilarity，使用数据集 deepghs/chafen_arknights(private) 训练。
当阈值为 0.45 时，调整兰德系数 可达到 0.995。
文件列表：
lpips_diff.onnx，特征差异。lpips_feature.onnx，特征提取。
单色图像
该模型用于单色图像分类，基于 CNN 和 Transformer，使用数据集 deepghs/monochrome_danbooru(private) 训练。
以下是已正式投入使用的检查点，均基于 Caformer 架构：
| 检查点 | 算法 | 安全等级 | 准确率 | 假阴性 | 假阳性 |
|---|---|---|---|---|---|
| monochrome-caformer-40 | caformer | 0 | 96.41% | 2.69% | 0.89% |
| monochrome-caformer-110 | caformer | 0 | 96.97% | 1.57% | 1.46% |
| monochrome-caformer_safe2-80 | caformer | 2 | 94.84% | 1.12% | 4.03% |
| monochrome-caformer_safe4-70 | caformer | 4 | 94.28% | 0.67% | 5.04% |
monochrome-caformer-110 在其中具有最佳的整体准确率，但考虑到该模型常用于筛选单色图像，而我们希望尽可能不遗漏地筛选出所有单色图像，因此我们还引入了加权模型（safe2 和 safe4）。虽然它们的整体准确率略有下降，但假阴性（将单色图像误判为彩色图像）的概率更低，更适合批量筛选。
Deepdanbooru
deepdanbooru 是一个用于给动漫图像打标签的模型。这里我们提供了一个标签分类表 deepdanbooru_tags.csv，以及一个 ONNX 模型（来自 chinoll/deepdanbooru）。
值得注意的是，由于 deepdanbooru 模型本身质量较差且数据集相对较旧，它仅用于测试目的，不建议作为主要的分类模型。我们推荐使用 wd14 模型，参见：
- https://huggingface.co/spaces/SmilingWolf/wd-v1-4-tags
deepghs/imgutils-models
作者 deepghs
↓ 0
♥ 9
创建时间: 2023-03-11 08:37:38+00:00
更新时间: 2023-07-03 04:12:18+00:00
在 Hugging Face 上查看文件 (84)
.gitattributes
README.md
anime_cls/anime_cls_mobilenetv3_large_100.ckpt
anime_cls/anime_cls_mobilenetv3_large_100.onnx
ONNX
anime_cls/anime_cls_mobilenetv3_large_100_cm.png
anime_cls/anime_cls_mobilenetv3_large_100_f1.png
anime_cls/anime_cls_mobilenetv3_large_100_pr.png
anime_cls/anime_cls_mobilevitv2_150.ckpt
anime_cls/anime_cls_mobilevitv2_150.onnx
ONNX
anime_cls/anime_cls_mobilevitv2_150_cm.png
anime_cls/anime_cls_mobilevitv2_150_f1.png
anime_cls/anime_cls_mobilevitv2_150_pr.png
ccip/ccip-caformer-23_randaug_fp32_feat.onnx
ONNX
ccip/ccip-caformer-23_randaug_fp32_metrics.onnx
ONNX
ccip/ccip-caformer-23_randaug_fp32_safe_metrics.onnx
ONNX
ccip/ccip-caformer-2_fp32_feat.onnx
ONNX
ccip/ccip-caformer-2_fp32_metrics.onnx
ONNX
ccip/ccip-caformer-2_fp32_safe_metrics.onnx
ONNX
ccip/ccip-caformer-4_fp32_feat.onnx
ONNX
ccip/ccip-caformer-4_fp32_metrics.onnx
ONNX
ccip/ccip-caformer-4_fp32_safe_metrics.onnx
ONNX
ccip/ccip-caformer-5_fp32_feat.onnx
ONNX
ccip/ccip-caformer-5_fp32_metrics.onnx
ONNX
ccip/ccip-caformer-5_fp32_safe_metrics.onnx
ONNX
deepdanbooru/deepdanbooru.onnx
ONNX
deepdanbooru/deepdanbooru_tags.csv
face_detect/face_detect_best_n.onnx
ONNX
face_detect/face_detect_best_n.pt
face_detect/face_detect_best_s.onnx
ONNX
face_detect/face_detect_best_s.pt
face_detect/face_detect_v1_best_n.onnx
ONNX
face_detect/face_detect_v1_best_n.pt
face_detect/face_detect_v1_best_s.onnx
ONNX
face_detect/face_detect_v1_best_s.pt
head_detect/head_detect_best_n.onnx
ONNX
head_detect/head_detect_best_n.pt
head_detect/head_detect_best_s.onnx
ONNX
head_detect/head_detect_best_s.pt
lineart/hed.onnx
ONNX
lineart/lineart.onnx
ONNX
lineart/lineart_anime.onnx
ONNX
lineart/lineart_coarse.onnx
ONNX
lineart/pidi.onnx
ONNX
lpips/lpips_diff.onnx
ONNX
lpips/lpips_feature.onnx
ONNX
manbits_detect/manbits_detect_best_m.onnx
ONNX
manbits_detect/manbits_detect_best_m.pt
mldanbooru/mldanbooru_tags.csv
monochrome/monochrome-alexnet-480.ckpt
monochrome/monochrome-alexnet-480.onnx
ONNX
monochrome/monochrome-caformer-110.ckpt
monochrome/monochrome-caformer-110.onnx
ONNX
monochrome/monochrome-caformer-40.ckpt
monochrome/monochrome-caformer-40.onnx
ONNX
monochrome/monochrome-caformer_safe2-80.ckpt
monochrome/monochrome-caformer_safe2-80.onnx
ONNX
monochrome/monochrome-caformer_safe4-70.ckpt
monochrome/monochrome-caformer_safe4-70.onnx
ONNX
monochrome/monochrome-levit_d0.2-500.ckpt
monochrome/monochrome-levit_d0.2-500.onnx
ONNX
monochrome/monochrome-resnet18-480.ckpt
monochrome/monochrome-resnet18-480.onnx
ONNX
monochrome/monochrome-resnet18-safe2-450.ckpt
monochrome/monochrome-resnet18-safe2-450.onnx
ONNX
monochrome/monochrome-transformer-480.ckpt
monochrome/monochrome-transformer-480.onnx
ONNX
nsfw/inception_v3.onnx
ONNX
nsfw/nsfwjs.onnx
ONNX
person_detect/person_detect_best_m.onnx
ONNX
person_detect/person_detect_best_m.pt
person_detect/person_detect_best_s.onnx
ONNX
person_detect/person_detect_best_s.pt
person_detect/person_detect_best_x.onnx
ONNX
person_detect/person_detect_best_x.pt
person_detect/person_detect_plus_best_m.onnx
ONNX
person_detect/person_detect_plus_best_m.pt
person_detect/person_detect_plus_v1.1_best_m.onnx
ONNX
person_detect/person_detect_plus_v1.1_best_m.pt
person_detect/person_detect_plus_v1.1_best_n.onnx
ONNX
person_detect/person_detect_plus_v1.1_best_n.pt
person_detect/person_detect_plus_v1.1_best_s.onnx
ONNX
person_detect/person_detect_plus_v1.1_best_s.pt
real_esrgan/RealESRGAN_x4plus_anime_4B32F.onnx
ONNX
real_esrgan/RealESRGAN_x4plus_anime_6B.onnx
ONNX