Migrasi otomatis
Browse filesThis view is limited to 50 files because it contains too many changes. See raw diff
- README.md +23 -0
- caformer_s36/meta.json +7 -0
- caformer_s36/metrics.json +7 -0
- caformer_s36/model.ckpt +3 -0
- caformer_s36/model.onnx +3 -0
- caformer_s36/plot_confusion.png +0 -0
- caformer_s36/plot_f1_curve.png +0 -0
- caformer_s36/plot_p_curve.png +0 -0
- caformer_s36/plot_pr_curve.png +0 -0
- caformer_s36/plot_r_curve.png +0 -0
- caformer_s36/plot_roc_curve.png +0 -0
- caformer_s36_plus/meta.json +9 -0
- caformer_s36_plus/metrics.json +7 -0
- caformer_s36_plus/model.ckpt +3 -0
- caformer_s36_plus/model.onnx +3 -0
- caformer_s36_plus/plot_confusion.png +0 -0
- caformer_s36_plus/plot_f1_curve.png +0 -0
- caformer_s36_plus/plot_p_curve.png +0 -0
- caformer_s36_plus/plot_pr_curve.png +0 -0
- caformer_s36_plus/plot_r_curve.png +0 -0
- caformer_s36_plus/plot_roc_curve.png +0 -0
- caformer_s36_plus_safe2/meta.json +9 -0
- caformer_s36_plus_safe2/metrics.json +7 -0
- caformer_s36_plus_safe2/model.ckpt +3 -0
- caformer_s36_plus_safe2/model.onnx +3 -0
- caformer_s36_plus_safe2/plot_confusion.png +0 -0
- caformer_s36_plus_safe2/plot_f1_curve.png +0 -0
- caformer_s36_plus_safe2/plot_p_curve.png +0 -0
- caformer_s36_plus_safe2/plot_pr_curve.png +0 -0
- caformer_s36_plus_safe2/plot_r_curve.png +0 -0
- caformer_s36_plus_safe2/plot_roc_curve.png +0 -0
- caformer_s36_safe2/meta.json +7 -0
- caformer_s36_safe2/metrics.json +7 -0
- caformer_s36_safe2/model.ckpt +3 -0
- caformer_s36_safe2/model.onnx +3 -0
- caformer_s36_safe2/plot_confusion.png +0 -0
- caformer_s36_safe2/plot_f1_curve.png +0 -0
- caformer_s36_safe2/plot_p_curve.png +0 -0
- caformer_s36_safe2/plot_pr_curve.png +0 -0
- caformer_s36_safe2/plot_r_curve.png +0 -0
- caformer_s36_safe2/plot_roc_curve.png +0 -0
- mobilenetv3_large_100/meta.json +7 -0
- mobilenetv3_large_100/metrics.json +7 -0
- mobilenetv3_large_100/model.ckpt +3 -0
- mobilenetv3_large_100/model.onnx +3 -0
- mobilenetv3_large_100/plot_confusion.png +0 -0
- mobilenetv3_large_100/plot_f1_curve.png +0 -0
- mobilenetv3_large_100/plot_p_curve.png +0 -0
- mobilenetv3_large_100/plot_pr_curve.png +0 -0
- mobilenetv3_large_100/plot_r_curve.png +0 -0
README.md
ADDED
|
@@ -0,0 +1,23 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
---
|
| 2 |
+
license: mit
|
| 3 |
+
datasets:
|
| 4 |
+
- deepghs/monochrome_danbooru
|
| 5 |
+
metrics:
|
| 6 |
+
- accuracy
|
| 7 |
+
pipeline_tag: image-classification
|
| 8 |
+
tags:
|
| 9 |
+
- art
|
| 10 |
+
---
|
| 11 |
+
|
| 12 |
+
The models used for determining whether an anime image is monochrome have a training size of 384.
|
| 13 |
+
|
| 14 |
+
| Model | FLOPs | Accuracy | Confusion Matrix | Description |
|
| 15 |
+
|:--------------------------------:|:------:|:--------:|:----------------------------------------------------------------------------------------------------------------------------------:|--------------------------------------------------------------------------------------------------------------------------------------------------------|
|
| 16 |
+
| caformer_s36 | 22.10G | 95.63% | [Confusion Matrix](https://huggingface.co/deepghs/monochrome_detect/blob/main/caformer_s36/plot_confusion.png) | Model: caformer_s36 from timm |
|
| 17 |
+
| caformer_s36_safe2 | 22.10G | 95.52% | [Confusion Matrix](https://huggingface.co/deepghs/monochrome_detect/blob/main/caformer_s36_safe2/plot_confusion.png) | Model: caformer_s36 from timm, which have better precision and lower recall than caformer_s36 |
|
| 18 |
+
| caformer_s36_plus | 22.10G | 97.31% | [Confusion Matrix](https://huggingface.co/deepghs/monochrome_detect/blob/main/caformer_s36_plus/plot_confusion.png) | Model: caformer_s36.sail_in22k_ft_in1k_384 pratrained from timm |
|
| 19 |
+
| caformer_s36_plus_safe2 | 22.10G | 97.09% | [Confusion Matrix](https://huggingface.co/deepghs/monochrome_detect/blob/main/caformer_s36_plus_safe2/plot_confusion.png) | Model: caformer_s36.sail_in22k_ft_in1k_384 pratrained from timm, which have better precision and lower recall than caformer_s36.sail_in22k_ft_in1k_384 |
|
| 20 |
+
| mobilenetv3_large_100 | 0.63G | 95.40% | [Confusion Matrix](https://huggingface.co/deepghs/monochrome_detect/blob/main/mobilenetv3_large_100/plot_confusion.png) | Model: mobilenetv3_large_100 from timm |
|
| 21 |
+
| mobilenetv3_large_100_dist | 0.63G | 96.30% | [Confusion Matrix](https://huggingface.co/deepghs/monochrome_detect/blob/main/mobilenetv3_large_100_dist/plot_confusion.png) | Distillated from caformer_s36_plus, using mobilenetv3_large_100 |
|
| 22 |
+
| mobilenetv3_large_100_safe2 | 0.63G | 94.62% | [Confusion Matrix](https://huggingface.co/deepghs/monochrome_detect/blob/main/mobilenetv3_large_100_safe2/plot_confusion.png) | Model: mobilenetv3_large_100 from timm, which have better precision and lower recall than mobilenetv3_large_100 |
|
| 23 |
+
| mobilenetv3_large_100_dist_safe2 | 0.63G | 95.85% | [Confusion Matrix](https://huggingface.co/deepghs/monochrome_detect/blob/main/mobilenetv3_large_100_dist_safe2/plot_confusion.png) | Distillated from caformer_s36_plus_safe2, using mobilenetv3_large_100 |
|
caformer_s36/meta.json
ADDED
|
@@ -0,0 +1,7 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"labels": [
|
| 3 |
+
"monochrome",
|
| 4 |
+
"normal"
|
| 5 |
+
],
|
| 6 |
+
"name": "caformer_s36"
|
| 7 |
+
}
|
caformer_s36/metrics.json
ADDED
|
@@ -0,0 +1,7 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"AUC": 0.9907084785133566,
|
| 3 |
+
"accuracy": 0.9562780269058296,
|
| 4 |
+
"loss": 0.042353093779683444,
|
| 5 |
+
"mAP": 0.9907146862735166,
|
| 6 |
+
"step": 150
|
| 7 |
+
}
|
caformer_s36/model.ckpt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:26d810bf024e45930b9241cbd575c204c3ed9c703dabad21acac0468397f5d37
|
| 3 |
+
size 162139119
|
caformer_s36/model.onnx
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:ef04c4bb554d372722bebb2a02e5c2ea1f58b2491eff87d2d2e4f6d108b76279
|
| 3 |
+
size 149574704
|
caformer_s36/plot_confusion.png
ADDED
|
caformer_s36/plot_f1_curve.png
ADDED
|
caformer_s36/plot_p_curve.png
ADDED
|
caformer_s36/plot_pr_curve.png
ADDED
|
caformer_s36/plot_r_curve.png
ADDED
|
caformer_s36/plot_roc_curve.png
ADDED
|
caformer_s36_plus/meta.json
ADDED
|
@@ -0,0 +1,9 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"drop_path_rate": 0.4,
|
| 3 |
+
"labels": [
|
| 4 |
+
"monochrome",
|
| 5 |
+
"normal"
|
| 6 |
+
],
|
| 7 |
+
"name": "caformer_s36.sail_in22k_ft_in1k_384",
|
| 8 |
+
"pretrained": true
|
| 9 |
+
}
|
caformer_s36_plus/metrics.json
ADDED
|
@@ -0,0 +1,7 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"AUC": 0.9966514321915019,
|
| 3 |
+
"accuracy": 0.9730941704035875,
|
| 4 |
+
"loss": 0.021936069169309357,
|
| 5 |
+
"mAP": 0.9967353129101008,
|
| 6 |
+
"step": 20
|
| 7 |
+
}
|
caformer_s36_plus/model.ckpt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:306336a8d87a2c926b086f79b2bbb1ee502942a22384d77b19fb76bb71e63628
|
| 3 |
+
size 162143087
|
caformer_s36_plus/model.onnx
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:7c71d42cbd99dc77cc5daf147418aaf73c1a291d703119ce23e23b8f2d26d0a4
|
| 3 |
+
size 149574704
|
caformer_s36_plus/plot_confusion.png
ADDED
|
caformer_s36_plus/plot_f1_curve.png
ADDED
|
caformer_s36_plus/plot_p_curve.png
ADDED
|
caformer_s36_plus/plot_pr_curve.png
ADDED
|
caformer_s36_plus/plot_r_curve.png
ADDED
|
caformer_s36_plus/plot_roc_curve.png
ADDED
|
caformer_s36_plus_safe2/meta.json
ADDED
|
@@ -0,0 +1,9 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"drop_path_rate": 0.4,
|
| 3 |
+
"labels": [
|
| 4 |
+
"monochrome",
|
| 5 |
+
"normal"
|
| 6 |
+
],
|
| 7 |
+
"name": "caformer_s36.sail_in22k_ft_in1k_384",
|
| 8 |
+
"pretrained": true
|
| 9 |
+
}
|
caformer_s36_plus_safe2/metrics.json
ADDED
|
@@ -0,0 +1,7 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"AUC": 0.9955302150424101,
|
| 3 |
+
"accuracy": 0.9708520179372198,
|
| 4 |
+
"loss": 0.028756862397946736,
|
| 5 |
+
"mAP": 0.9956484992485766,
|
| 6 |
+
"step": 30
|
| 7 |
+
}
|
caformer_s36_plus_safe2/model.ckpt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:864c728956337e0fc393c0467971205e5bac1e587929cf1d60d62ff2d6c0a534
|
| 3 |
+
size 162142191
|
caformer_s36_plus_safe2/model.onnx
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:3a9d36b66e6b8fd812e0253d81b6f0ef758814248928de48f0a890bc85d5620c
|
| 3 |
+
size 149574704
|
caformer_s36_plus_safe2/plot_confusion.png
ADDED
|
caformer_s36_plus_safe2/plot_f1_curve.png
ADDED
|
caformer_s36_plus_safe2/plot_p_curve.png
ADDED
|
caformer_s36_plus_safe2/plot_pr_curve.png
ADDED
|
caformer_s36_plus_safe2/plot_r_curve.png
ADDED
|
caformer_s36_plus_safe2/plot_roc_curve.png
ADDED
|
caformer_s36_safe2/meta.json
ADDED
|
@@ -0,0 +1,7 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"labels": [
|
| 3 |
+
"monochrome",
|
| 4 |
+
"normal"
|
| 5 |
+
],
|
| 6 |
+
"name": "caformer_s36"
|
| 7 |
+
}
|
caformer_s36_safe2/metrics.json
ADDED
|
@@ -0,0 +1,7 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"AUC": 0.9869627082170636,
|
| 3 |
+
"accuracy": 0.9551569506726457,
|
| 4 |
+
"loss": 0.06188738217507907,
|
| 5 |
+
"mAP": 0.9868226986519757,
|
| 6 |
+
"step": 225
|
| 7 |
+
}
|
caformer_s36_safe2/model.ckpt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:d3eb5cdc657d9bb4f7280640d5d50a602d41938784fe8f94209dd82f36a59824
|
| 3 |
+
size 162137007
|
caformer_s36_safe2/model.onnx
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:4c42d263d142b29effc9710ae53c1af1eda18b646d786c9e9c10b3da22cd6a1a
|
| 3 |
+
size 149574704
|
caformer_s36_safe2/plot_confusion.png
ADDED
|
caformer_s36_safe2/plot_f1_curve.png
ADDED
|
caformer_s36_safe2/plot_p_curve.png
ADDED
|
caformer_s36_safe2/plot_pr_curve.png
ADDED
|
caformer_s36_safe2/plot_r_curve.png
ADDED
|
caformer_s36_safe2/plot_roc_curve.png
ADDED
|
mobilenetv3_large_100/meta.json
ADDED
|
@@ -0,0 +1,7 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"labels": [
|
| 3 |
+
"monochrome",
|
| 4 |
+
"normal"
|
| 5 |
+
],
|
| 6 |
+
"name": "mobilenetv3_large_100"
|
| 7 |
+
}
|
mobilenetv3_large_100/metrics.json
ADDED
|
@@ -0,0 +1,7 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"AUC": 0.9904420008949627,
|
| 3 |
+
"accuracy": 0.9540358744394619,
|
| 4 |
+
"loss": 0.033743727342497074,
|
| 5 |
+
"mAP": 0.9705153386460841,
|
| 6 |
+
"step": 245
|
| 7 |
+
}
|
mobilenetv3_large_100/model.ckpt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:8460fa84beadf26c556ab5125ffc54ace0adad3b4d0c583a0547b2fdbbd7d927
|
| 3 |
+
size 29771275
|
mobilenetv3_large_100/model.onnx
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:a80173590e933d8043a56de7b9627978dde4fc0ab578f726464fd04ae6b3adb1
|
| 3 |
+
size 16822434
|
mobilenetv3_large_100/plot_confusion.png
ADDED
|
mobilenetv3_large_100/plot_f1_curve.png
ADDED
|
mobilenetv3_large_100/plot_p_curve.png
ADDED
|
mobilenetv3_large_100/plot_pr_curve.png
ADDED
|
mobilenetv3_large_100/plot_r_curve.png
ADDED
|