diff --git a/README.md b/README.md new file mode 100644 index 0000000000000000000000000000000000000000..70daef3132bea7a8b18886a70a09a99f4f4dbbcc --- /dev/null +++ b/README.md @@ -0,0 +1,18 @@ +--- +license: mit +datasets: +- deepghs/anime_ai_check +metrics: +- accuracy +pipeline_tag: image-classification +tags: +- art +--- + +The model used to predict whether an image is generated by AI techniques, mainly diffusion models. Please note that this model is trained on crawled data, and **its predictions should be used for reference purposes only**. It should not be used to determine the source or origin of an image. + +| Model | FLOPs | Accuracy | Confusion Matrix | Description | +|:---------------------:|:------:|:--------:|:--------------------------------------------------------------------------------------------------------------------:|---------------------------------------------------------------------| +| caformer_s36_plus_sce | 22.10G | 97.54% | [Confusion Matrix](https://huggingface.co/deepghs/anime_ai_check/blob/main/caformer_s36_plus_sce/plot_confusion.png) | Model: caformer_s36.sail_in22k_ft_in1k_384 pratrained from timm | +| mobilenetv3_sce | 0.63G | 93.20% | [Confusion Matrix](https://huggingface.co/deepghs/anime_ai_check/blob/main/mobilenetv3_sce/plot_confusion.png) | Model: mobilenetv3_large_100 from timm | +| mobilenetv3_sce_dist | 0.63G | 95.65% | [Confusion Matrix](https://huggingface.co/deepghs/anime_ai_check/blob/main/mobilenetv3_sce_dist/plot_confusion.png) | Distillated from caformer_s36_plus_sce, using mobilenetv3_large_100 | diff --git a/caformer_s36_plus_sce/meta.json b/caformer_s36_plus_sce/meta.json new file mode 100644 index 0000000000000000000000000000000000000000..0c40001a8d2c8466ddbfe6b09a1c0439b2ac3ccb --- /dev/null +++ b/caformer_s36_plus_sce/meta.json @@ -0,0 +1,9 @@ +{ + "drop_path_rate": 0.4, + "labels": [ + "ai", + "human" + ], + "name": "caformer_s36.sail_in22k_ft_in1k_384", + "pretrained": true +} \ No newline at end of file diff --git a/caformer_s36_plus_sce/metrics.json b/caformer_s36_plus_sce/metrics.json new file mode 100644 index 0000000000000000000000000000000000000000..6d76069a717da6f0c35041c9abd59c96c61b79a5 --- /dev/null +++ b/caformer_s36_plus_sce/metrics.json @@ -0,0 +1,7 @@ +{ + "AUC": 0.9873855979080761, + "accuracy": 0.9754385964912281, + "loss": 0.25407937819952564, + "mAP": 0.9871511707341074, + "step": 21 +} \ No newline at end of file diff --git a/caformer_s36_plus_sce/model.ckpt b/caformer_s36_plus_sce/model.ckpt new file mode 100644 index 0000000000000000000000000000000000000000..e13de20f61a6691eed9fecd0c9b883c1d991ba7a --- /dev/null +++ b/caformer_s36_plus_sce/model.ckpt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:27331a1e9e030f5785f45895cd2f5a51be629a4aefd11eb8de9bfe61e9549a2c +size 162179823 diff --git a/caformer_s36_plus_sce/model.onnx b/caformer_s36_plus_sce/model.onnx new file mode 100644 index 0000000000000000000000000000000000000000..22413df029fcefd95dc0e5304f450de83f8ad794 --- /dev/null +++ b/caformer_s36_plus_sce/model.onnx @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:0c2120ec61d5adb9c87048d8dbd206c003de10d22029b745c91f3a2fdb053b5f +size 149574704 diff --git a/caformer_s36_plus_sce/plot_confusion.png b/caformer_s36_plus_sce/plot_confusion.png new file mode 100644 index 0000000000000000000000000000000000000000..74e15ba7873c2aa6015e43720407e1ad864132e1 Binary files /dev/null and b/caformer_s36_plus_sce/plot_confusion.png differ diff --git a/caformer_s36_plus_sce/plot_f1_curve.png b/caformer_s36_plus_sce/plot_f1_curve.png new file mode 100644 index 0000000000000000000000000000000000000000..168cae6d353c32d137ae2a368e23d8723e3b43e8 Binary files /dev/null and b/caformer_s36_plus_sce/plot_f1_curve.png differ diff --git a/caformer_s36_plus_sce/plot_p_curve.png b/caformer_s36_plus_sce/plot_p_curve.png new file mode 100644 index 0000000000000000000000000000000000000000..6effb57a0c73aa7182130c26a0c1aaee608b8e7e Binary files /dev/null and b/caformer_s36_plus_sce/plot_p_curve.png differ diff --git a/caformer_s36_plus_sce/plot_pr_curve.png b/caformer_s36_plus_sce/plot_pr_curve.png new file mode 100644 index 0000000000000000000000000000000000000000..8246b8b47d7fabc3c2359ff09a53d2b6255216f1 Binary files /dev/null and b/caformer_s36_plus_sce/plot_pr_curve.png differ diff --git a/caformer_s36_plus_sce/plot_r_curve.png b/caformer_s36_plus_sce/plot_r_curve.png new file mode 100644 index 0000000000000000000000000000000000000000..e97933e947c8ebe52de8d88e1f13aaa29296fe1d Binary files /dev/null and b/caformer_s36_plus_sce/plot_r_curve.png differ diff --git a/caformer_s36_plus_sce/plot_roc_curve.png b/caformer_s36_plus_sce/plot_roc_curve.png new file mode 100644 index 0000000000000000000000000000000000000000..6ee8e2ceeffd88b3743ba770d030a593d2f62e1e Binary files /dev/null and b/caformer_s36_plus_sce/plot_roc_curve.png differ diff --git a/caformer_s36_sce_v1/meta.json b/caformer_s36_sce_v1/meta.json new file mode 100644 index 0000000000000000000000000000000000000000..0c40001a8d2c8466ddbfe6b09a1c0439b2ac3ccb --- /dev/null +++ b/caformer_s36_sce_v1/meta.json @@ -0,0 +1,9 @@ +{ + "drop_path_rate": 0.4, + "labels": [ + "ai", + "human" + ], + "name": "caformer_s36.sail_in22k_ft_in1k_384", + "pretrained": true +} \ No newline at end of file diff --git a/caformer_s36_sce_v1/metrics.json b/caformer_s36_sce_v1/metrics.json new file mode 100644 index 0000000000000000000000000000000000000000..4c8e50bdb9f7969c2599865a9d86b5321a0a79ac --- /dev/null +++ b/caformer_s36_sce_v1/metrics.json @@ -0,0 +1,7 @@ +{ + "AUC": 0.9909439909256175, + "accuracy": 0.9610443431413178, + "loss": 0.3885019321138965, + "mAP": 0.9910901017282925, + "step": 40 +} \ No newline at end of file diff --git a/caformer_s36_sce_v1/model.ckpt b/caformer_s36_sce_v1/model.ckpt new file mode 100644 index 0000000000000000000000000000000000000000..beeb7bbeb0c3a6785c82d3372dd2d6879d754c8a --- /dev/null +++ b/caformer_s36_sce_v1/model.ckpt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:555ef075329f8fd7887f7c2d7af2460ecab4b77ea33a4fda4de6beb6506c68b1 +size 162181295 diff --git a/caformer_s36_sce_v1/model.onnx b/caformer_s36_sce_v1/model.onnx new file mode 100644 index 0000000000000000000000000000000000000000..7364096b98a6918eb4918de7f367ac7b3a52fc3d --- /dev/null +++ b/caformer_s36_sce_v1/model.onnx @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:6795bfa68cbba4b6ffe977339331207327c31db60679c6ac6785417199be0d07 +size 149574704 diff --git a/caformer_s36_sce_v1/plot_confusion.png b/caformer_s36_sce_v1/plot_confusion.png new file mode 100644 index 0000000000000000000000000000000000000000..8ba5fd969fd53d841c21a9765042b2cbe7039814 Binary files /dev/null and b/caformer_s36_sce_v1/plot_confusion.png differ diff --git a/caformer_s36_sce_v1/plot_f1_curve.png b/caformer_s36_sce_v1/plot_f1_curve.png new file mode 100644 index 0000000000000000000000000000000000000000..98824ab1d51b651cff25f9717a8843f6ccbde6a0 Binary files /dev/null and b/caformer_s36_sce_v1/plot_f1_curve.png differ diff --git a/caformer_s36_sce_v1/plot_p_curve.png b/caformer_s36_sce_v1/plot_p_curve.png new file mode 100644 index 0000000000000000000000000000000000000000..bbe642b1872627fc267456e7ad056c78810808ea Binary files /dev/null and b/caformer_s36_sce_v1/plot_p_curve.png differ diff --git a/caformer_s36_sce_v1/plot_pr_curve.png b/caformer_s36_sce_v1/plot_pr_curve.png new file mode 100644 index 0000000000000000000000000000000000000000..ac343d84dfd810750bfa67c83b9271982e03aa6d Binary files /dev/null and b/caformer_s36_sce_v1/plot_pr_curve.png differ diff --git a/caformer_s36_sce_v1/plot_r_curve.png b/caformer_s36_sce_v1/plot_r_curve.png new file mode 100644 index 0000000000000000000000000000000000000000..484839548d760640e35d25b1f9d81572737c9d0e Binary files /dev/null and b/caformer_s36_sce_v1/plot_r_curve.png differ diff --git a/caformer_s36_sce_v1/plot_roc_curve.png b/caformer_s36_sce_v1/plot_roc_curve.png new file mode 100644 index 0000000000000000000000000000000000000000..767c35d8ead81ef214e04944d8b409ee1745c4ed Binary files /dev/null and b/caformer_s36_sce_v1/plot_roc_curve.png differ diff --git a/mobilenetv3_sce/meta.json b/mobilenetv3_sce/meta.json new file mode 100644 index 0000000000000000000000000000000000000000..a343a34035763d684fb9916dd582bd75605984ed --- /dev/null +++ b/mobilenetv3_sce/meta.json @@ -0,0 +1,8 @@ +{ + "labels": [ + "ai", + "human" + ], + "name": "mobilenetv3_large_100.miil_in21k_ft_in1k", + "pretrained": true +} \ No newline at end of file diff --git a/mobilenetv3_sce/metrics.json b/mobilenetv3_sce/metrics.json new file mode 100644 index 0000000000000000000000000000000000000000..4d72d5b519835e93012fbf393acf7d07445b5dae --- /dev/null +++ b/mobilenetv3_sce/metrics.json @@ -0,0 +1,7 @@ +{ + "AUC": 0.9689821985316303, + "accuracy": 0.9319298245614035, + "loss": 0.7638901418277844, + "mAP": 0.9697108619511428, + "step": 17 +} \ No newline at end of file diff --git a/mobilenetv3_sce/model.ckpt b/mobilenetv3_sce/model.ckpt new file mode 100644 index 0000000000000000000000000000000000000000..32f70690dac1097709103d544522a7ebd48adbc2 --- /dev/null +++ b/mobilenetv3_sce/model.ckpt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:e55541e0660261920b171b1451a9670d1625c7998c5b6b8306d680b7d61b1ccb +size 29811851 diff --git a/mobilenetv3_sce/model.onnx b/mobilenetv3_sce/model.onnx new file mode 100644 index 0000000000000000000000000000000000000000..c6095d474bbbeaa9e8de0cb7691cd3be00625434 --- /dev/null +++ b/mobilenetv3_sce/model.onnx @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:94bacb5fc7e2eeae3ae10d4d1ff9c83bcc50b9f140740e30b9f4a5fe77282df5 +size 16822434 diff --git a/mobilenetv3_sce/plot_confusion.png b/mobilenetv3_sce/plot_confusion.png new file mode 100644 index 0000000000000000000000000000000000000000..e781f25415ce51c2a40605a94c3672ffe3716ec6 Binary files /dev/null and b/mobilenetv3_sce/plot_confusion.png differ diff --git a/mobilenetv3_sce/plot_f1_curve.png b/mobilenetv3_sce/plot_f1_curve.png new file mode 100644 index 0000000000000000000000000000000000000000..a013fafb5eab3098e823671d2bf11e0232ef2b82 Binary files /dev/null and b/mobilenetv3_sce/plot_f1_curve.png differ diff --git a/mobilenetv3_sce/plot_p_curve.png b/mobilenetv3_sce/plot_p_curve.png new file mode 100644 index 0000000000000000000000000000000000000000..d9ad96ee3f58488e11ad383a2b453bcd8fa18264 Binary files /dev/null and b/mobilenetv3_sce/plot_p_curve.png differ diff --git a/mobilenetv3_sce/plot_pr_curve.png b/mobilenetv3_sce/plot_pr_curve.png new file mode 100644 index 0000000000000000000000000000000000000000..230ff250448f0f1fb9e65a23b8d242f7f0122b85 Binary files /dev/null and b/mobilenetv3_sce/plot_pr_curve.png differ diff --git a/mobilenetv3_sce/plot_r_curve.png b/mobilenetv3_sce/plot_r_curve.png new file mode 100644 index 0000000000000000000000000000000000000000..1ad37e5982f2f21bf7556acb324e8552d657ec7d Binary files /dev/null and b/mobilenetv3_sce/plot_r_curve.png differ diff --git a/mobilenetv3_sce/plot_roc_curve.png b/mobilenetv3_sce/plot_roc_curve.png new file mode 100644 index 0000000000000000000000000000000000000000..a58686e4329e8bd7aadce78a8dbcd22b43f02ffe Binary files /dev/null and b/mobilenetv3_sce/plot_roc_curve.png differ diff --git a/mobilenetv3_sce_dist/meta.json b/mobilenetv3_sce_dist/meta.json new file mode 100644 index 0000000000000000000000000000000000000000..15bef1f9b226f85f8625c471eab10a09e7b47b7e --- /dev/null +++ b/mobilenetv3_sce_dist/meta.json @@ -0,0 +1,8 @@ +{ + "labels": [ + "ai", + "human" + ], + "name": "mobilenetv3_large_100", + "pretrained": true +} \ No newline at end of file diff --git a/mobilenetv3_sce_dist/metrics.json b/mobilenetv3_sce_dist/metrics.json new file mode 100644 index 0000000000000000000000000000000000000000..6fb3136cd50daba3a40e3d7e15728995181dd55d --- /dev/null +++ b/mobilenetv3_sce_dist/metrics.json @@ -0,0 +1,7 @@ +{ + "AUC": 0.9862194508699589, + "accuracy": 0.9564912280701754, + "cls_loss": 0.4791290839545821, + "mAP": 0.985848883761399, + "step": 21 +} \ No newline at end of file diff --git a/mobilenetv3_sce_dist/model.ckpt b/mobilenetv3_sce_dist/model.ckpt new file mode 100644 index 0000000000000000000000000000000000000000..8fd45067ade93559f9e1c5e2245f267732ab75d2 --- /dev/null +++ b/mobilenetv3_sce_dist/model.ckpt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:0c90349ce6efce2fdd2ad735f34d6e5e7959cfa2e41e7015561e0005dd78c8c6 +size 29809675 diff --git a/mobilenetv3_sce_dist/model.onnx b/mobilenetv3_sce_dist/model.onnx new file mode 100644 index 0000000000000000000000000000000000000000..48dae6f54c5dc88af6c829412362ad422d3f8ef4 --- /dev/null +++ b/mobilenetv3_sce_dist/model.onnx @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:0215a215470145a676d7496d138cac3129d93067893573f0c57a7c9ecf7d0dee +size 16822434 diff --git a/mobilenetv3_sce_dist/plot_confusion.png b/mobilenetv3_sce_dist/plot_confusion.png new file mode 100644 index 0000000000000000000000000000000000000000..3ffd7952408f02fa548bcea69dd6a514dfcd42ef Binary files /dev/null and b/mobilenetv3_sce_dist/plot_confusion.png differ diff --git a/mobilenetv3_sce_dist/plot_f1_curve.png b/mobilenetv3_sce_dist/plot_f1_curve.png new file mode 100644 index 0000000000000000000000000000000000000000..649e9a43dbc2d882839a6f4a165a8450088a4c29 Binary files /dev/null and b/mobilenetv3_sce_dist/plot_f1_curve.png differ diff --git a/mobilenetv3_sce_dist/plot_p_curve.png b/mobilenetv3_sce_dist/plot_p_curve.png new file mode 100644 index 0000000000000000000000000000000000000000..7737bf3fd56ce3faa39e4a00be08ecf5177a8b73 Binary files /dev/null and b/mobilenetv3_sce_dist/plot_p_curve.png differ diff --git a/mobilenetv3_sce_dist/plot_pr_curve.png b/mobilenetv3_sce_dist/plot_pr_curve.png new file mode 100644 index 0000000000000000000000000000000000000000..a1591c719db7342d170d482f3d7e0ffe9f1c0dd6 Binary files /dev/null and b/mobilenetv3_sce_dist/plot_pr_curve.png differ diff --git a/mobilenetv3_sce_dist/plot_r_curve.png b/mobilenetv3_sce_dist/plot_r_curve.png new file mode 100644 index 0000000000000000000000000000000000000000..fe62d491b0ee5e238964b7134c5ce3df79e8f8a1 Binary files /dev/null and b/mobilenetv3_sce_dist/plot_r_curve.png differ diff --git a/mobilenetv3_sce_dist/plot_roc_curve.png b/mobilenetv3_sce_dist/plot_roc_curve.png new file mode 100644 index 0000000000000000000000000000000000000000..bce798ecafe134209037c3bb7f59ba3560772c19 Binary files /dev/null and b/mobilenetv3_sce_dist/plot_roc_curve.png differ diff --git a/mobilenetv3_sce_dist_v1/meta.json b/mobilenetv3_sce_dist_v1/meta.json new file mode 100644 index 0000000000000000000000000000000000000000..15bef1f9b226f85f8625c471eab10a09e7b47b7e --- /dev/null +++ b/mobilenetv3_sce_dist_v1/meta.json @@ -0,0 +1,8 @@ +{ + "labels": [ + "ai", + "human" + ], + "name": "mobilenetv3_large_100", + "pretrained": true +} \ No newline at end of file diff --git a/mobilenetv3_sce_dist_v1/metrics.json b/mobilenetv3_sce_dist_v1/metrics.json new file mode 100644 index 0000000000000000000000000000000000000000..316f2c5da2b81fb8e87db3ca4731e8894251cbfa --- /dev/null +++ b/mobilenetv3_sce_dist_v1/metrics.json @@ -0,0 +1,7 @@ +{ + "AUC": 0.9905496560359759, + "accuracy": 0.9639452963116453, + "cls_loss": 0.37494325319342114, + "mAP": 0.990459750684058, + "step": 96 +} \ No newline at end of file diff --git a/mobilenetv3_sce_dist_v1/model.ckpt b/mobilenetv3_sce_dist_v1/model.ckpt new file mode 100644 index 0000000000000000000000000000000000000000..a381954fa8e99a2ba3ba237c87b7819e90b068da --- /dev/null +++ b/mobilenetv3_sce_dist_v1/model.ckpt @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:71cffa9d46442df267e4505b37bdfbfe674cf64e63c8dead71fc8b31de6ae473 +size 29811723 diff --git a/mobilenetv3_sce_dist_v1/model.onnx b/mobilenetv3_sce_dist_v1/model.onnx new file mode 100644 index 0000000000000000000000000000000000000000..415b5458d2d1a76acd54cae1c650aa652b70ed09 --- /dev/null +++ b/mobilenetv3_sce_dist_v1/model.onnx @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:c6bf6201d99b86cd77821790c4403c81630d51918be5d3debe23e07a29f95d2a +size 16822434 diff --git a/mobilenetv3_sce_dist_v1/plot_confusion.png b/mobilenetv3_sce_dist_v1/plot_confusion.png new file mode 100644 index 0000000000000000000000000000000000000000..50ace9aeef8eee623da7564e0128f575ed0fe162 Binary files /dev/null and b/mobilenetv3_sce_dist_v1/plot_confusion.png differ diff --git a/mobilenetv3_sce_dist_v1/plot_f1_curve.png b/mobilenetv3_sce_dist_v1/plot_f1_curve.png new file mode 100644 index 0000000000000000000000000000000000000000..9c2efdb5794ca5edbba9449c1f27ac99643e56a8 Binary files /dev/null and b/mobilenetv3_sce_dist_v1/plot_f1_curve.png differ diff --git a/mobilenetv3_sce_dist_v1/plot_p_curve.png b/mobilenetv3_sce_dist_v1/plot_p_curve.png new file mode 100644 index 0000000000000000000000000000000000000000..91aeb1d26b58bc188f919e902664aaaf228a4071 Binary files /dev/null and b/mobilenetv3_sce_dist_v1/plot_p_curve.png differ diff --git a/mobilenetv3_sce_dist_v1/plot_pr_curve.png b/mobilenetv3_sce_dist_v1/plot_pr_curve.png new file mode 100644 index 0000000000000000000000000000000000000000..f275896724ceb39523087965fb92e77f3415fb3d Binary files /dev/null and b/mobilenetv3_sce_dist_v1/plot_pr_curve.png differ diff --git a/mobilenetv3_sce_dist_v1/plot_r_curve.png b/mobilenetv3_sce_dist_v1/plot_r_curve.png new file mode 100644 index 0000000000000000000000000000000000000000..eae32f2bca7137cf841ebd85aa238cda86a85ed4 Binary files /dev/null and b/mobilenetv3_sce_dist_v1/plot_r_curve.png differ diff --git a/mobilenetv3_sce_dist_v1/plot_roc_curve.png b/mobilenetv3_sce_dist_v1/plot_roc_curve.png new file mode 100644 index 0000000000000000000000000000000000000000..df8329b9df20eabefd182d86d99dc6c08b1c05b3 Binary files /dev/null and b/mobilenetv3_sce_dist_v1/plot_roc_curve.png differ