AldyRevigustian commited on
Commit ·
27840cb
0
Parent(s):
Upload model and supporting files
Browse files- .gitattributes +1 -0
- Model-0.4/config.json +38 -0
- Model-0.4/face_model.pkl +3 -0
- Model-0.4/member.json +26 -0
- Model-0.5/config.json +38 -0
- Model-0.5/face_model.pkl +3 -0
- Model-0.5/member.json +26 -0
- Model-0.6/config.json +38 -0
- Model-0.6/face_model.pkl +3 -0
- Model-0.6/member.json +26 -0
- Model-base/config.json +37 -0
- Model-base/member.json +26 -0
- Model-base/svm_face_model.pkl +3 -0
- README.md +72 -0
.gitattributes
ADDED
|
@@ -0,0 +1 @@
|
|
|
|
|
|
|
| 1 |
+
*.pkl filter=lfs diff=lfs merge=lfs -text
|
Model-0.4/config.json
ADDED
|
@@ -0,0 +1,38 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"model_name": "tripleS-Member-Recognition 0.4",
|
| 3 |
+
"description": "Fine-tuned classifier model with confidence threshold of 0.4. Higher recall, lower precision.",
|
| 4 |
+
"threshold": 0.4,
|
| 5 |
+
"fine_tuned": true,
|
| 6 |
+
"model_file": "face_model.pkl",
|
| 7 |
+
"embedding_model": "buffalo_l",
|
| 8 |
+
"format": "pkl",
|
| 9 |
+
"total_persons": 24,
|
| 10 |
+
"total_samples": 13951,
|
| 11 |
+
"num_classes": 24,
|
| 12 |
+
"person_names": [
|
| 13 |
+
"ChaeWon",
|
| 14 |
+
"ChaeYeon",
|
| 15 |
+
"DaHyun",
|
| 16 |
+
"HaYeon",
|
| 17 |
+
"HyeRin",
|
| 18 |
+
"JiWoo",
|
| 19 |
+
"JiYeon",
|
| 20 |
+
"JooBin",
|
| 21 |
+
"Kaede",
|
| 22 |
+
"Kotone",
|
| 23 |
+
"Lynn",
|
| 24 |
+
"Mayu",
|
| 25 |
+
"NaKyoung",
|
| 26 |
+
"Nien",
|
| 27 |
+
"SeoAh",
|
| 28 |
+
"SeoYeon",
|
| 29 |
+
"ShiOn",
|
| 30 |
+
"SoHyun",
|
| 31 |
+
"SooMin",
|
| 32 |
+
"Sullin",
|
| 33 |
+
"Xinyu",
|
| 34 |
+
"YeonJi",
|
| 35 |
+
"YooYeon",
|
| 36 |
+
"YuBin"
|
| 37 |
+
]
|
| 38 |
+
}
|
Model-0.4/face_model.pkl
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:dc2ad2a7d4c07ba199ada263a6948f01599a35509b55bb55999e13d70f8ae9a3
|
| 3 |
+
size 16295426
|
Model-0.4/member.json
ADDED
|
@@ -0,0 +1,26 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
[
|
| 2 |
+
"ChaeWon",
|
| 3 |
+
"ChaeYeon",
|
| 4 |
+
"DaHyun",
|
| 5 |
+
"HaYeon",
|
| 6 |
+
"HyeRin",
|
| 7 |
+
"JiWoo",
|
| 8 |
+
"JiYeon",
|
| 9 |
+
"JooBin",
|
| 10 |
+
"Kaede",
|
| 11 |
+
"Kotone",
|
| 12 |
+
"Lynn",
|
| 13 |
+
"Mayu",
|
| 14 |
+
"NaKyoung",
|
| 15 |
+
"Nien",
|
| 16 |
+
"SeoAh",
|
| 17 |
+
"SeoYeon",
|
| 18 |
+
"ShiOn",
|
| 19 |
+
"SoHyun",
|
| 20 |
+
"SooMin",
|
| 21 |
+
"Sullin",
|
| 22 |
+
"Xinyu",
|
| 23 |
+
"YeonJi",
|
| 24 |
+
"YooYeon",
|
| 25 |
+
"YuBin"
|
| 26 |
+
]
|
Model-0.5/config.json
ADDED
|
@@ -0,0 +1,38 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"model_name": "tripleS-Member-Recognition 0.5",
|
| 3 |
+
"description": "Fine-tuned classifier model with confidence threshold of 0.5. Balanced recall and precision.",
|
| 4 |
+
"threshold": 0.5,
|
| 5 |
+
"fine_tuned": true,
|
| 6 |
+
"model_file": "face_model.pkl",
|
| 7 |
+
"embedding_model": "buffalo_l",
|
| 8 |
+
"format": "pkl",
|
| 9 |
+
"total_persons": 24,
|
| 10 |
+
"total_samples": 13707,
|
| 11 |
+
"num_classes": 24,
|
| 12 |
+
"person_names": [
|
| 13 |
+
"ChaeWon",
|
| 14 |
+
"ChaeYeon",
|
| 15 |
+
"DaHyun",
|
| 16 |
+
"HaYeon",
|
| 17 |
+
"HyeRin",
|
| 18 |
+
"JiWoo",
|
| 19 |
+
"JiYeon",
|
| 20 |
+
"JooBin",
|
| 21 |
+
"Kaede",
|
| 22 |
+
"Kotone",
|
| 23 |
+
"Lynn",
|
| 24 |
+
"Mayu",
|
| 25 |
+
"NaKyoung",
|
| 26 |
+
"Nien",
|
| 27 |
+
"SeoAh",
|
| 28 |
+
"SeoYeon",
|
| 29 |
+
"ShiOn",
|
| 30 |
+
"SoHyun",
|
| 31 |
+
"SooMin",
|
| 32 |
+
"Sullin",
|
| 33 |
+
"Xinyu",
|
| 34 |
+
"YeonJi",
|
| 35 |
+
"YooYeon",
|
| 36 |
+
"YuBin"
|
| 37 |
+
]
|
| 38 |
+
}
|
Model-0.5/face_model.pkl
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:18d7861c9b0335568e089eb72f5092d1002b596f4848a08b9e9d52ad9793493a
|
| 3 |
+
size 16889682
|
Model-0.5/member.json
ADDED
|
@@ -0,0 +1,26 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
[
|
| 2 |
+
"ChaeWon",
|
| 3 |
+
"ChaeYeon",
|
| 4 |
+
"DaHyun",
|
| 5 |
+
"HaYeon",
|
| 6 |
+
"HyeRin",
|
| 7 |
+
"JiWoo",
|
| 8 |
+
"JiYeon",
|
| 9 |
+
"JooBin",
|
| 10 |
+
"Kaede",
|
| 11 |
+
"Kotone",
|
| 12 |
+
"Lynn",
|
| 13 |
+
"Mayu",
|
| 14 |
+
"NaKyoung",
|
| 15 |
+
"Nien",
|
| 16 |
+
"SeoAh",
|
| 17 |
+
"SeoYeon",
|
| 18 |
+
"ShiOn",
|
| 19 |
+
"SoHyun",
|
| 20 |
+
"SooMin",
|
| 21 |
+
"Sullin",
|
| 22 |
+
"Xinyu",
|
| 23 |
+
"YeonJi",
|
| 24 |
+
"YooYeon",
|
| 25 |
+
"YuBin"
|
| 26 |
+
]
|
Model-0.6/config.json
ADDED
|
@@ -0,0 +1,38 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"model_name": "tripleS-Member-Recognition 0.6",
|
| 3 |
+
"description": "Fine-tuned classifier model with confidence threshold of 0.6. Higher precision, lower recall.",
|
| 4 |
+
"threshold": 0.6,
|
| 5 |
+
"fine_tuned": true,
|
| 6 |
+
"model_file": "face_model.pkl",
|
| 7 |
+
"embedding_model": "buffalo_l",
|
| 8 |
+
"format": "pkl",
|
| 9 |
+
"total_persons": 24,
|
| 10 |
+
"total_samples": 13358,
|
| 11 |
+
"num_classes": 24,
|
| 12 |
+
"person_names": [
|
| 13 |
+
"ChaeWon",
|
| 14 |
+
"ChaeYeon",
|
| 15 |
+
"DaHyun",
|
| 16 |
+
"HaYeon",
|
| 17 |
+
"HyeRin",
|
| 18 |
+
"JiWoo",
|
| 19 |
+
"JiYeon",
|
| 20 |
+
"JooBin",
|
| 21 |
+
"Kaede",
|
| 22 |
+
"Kotone",
|
| 23 |
+
"Lynn",
|
| 24 |
+
"Mayu",
|
| 25 |
+
"NaKyoung",
|
| 26 |
+
"Nien",
|
| 27 |
+
"SeoAh",
|
| 28 |
+
"SeoYeon",
|
| 29 |
+
"ShiOn",
|
| 30 |
+
"SoHyun",
|
| 31 |
+
"SooMin",
|
| 32 |
+
"Sullin",
|
| 33 |
+
"Xinyu",
|
| 34 |
+
"YeonJi",
|
| 35 |
+
"YooYeon",
|
| 36 |
+
"YuBin"
|
| 37 |
+
]
|
| 38 |
+
}
|
Model-0.6/face_model.pkl
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:063f8579d8f7d58723f27474f5fe7e691b7a2a98d062751ffffa66c309350da3
|
| 3 |
+
size 17332002
|
Model-0.6/member.json
ADDED
|
@@ -0,0 +1,26 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
[
|
| 2 |
+
"ChaeWon",
|
| 3 |
+
"ChaeYeon",
|
| 4 |
+
"DaHyun",
|
| 5 |
+
"HaYeon",
|
| 6 |
+
"HyeRin",
|
| 7 |
+
"JiWoo",
|
| 8 |
+
"JiYeon",
|
| 9 |
+
"JooBin",
|
| 10 |
+
"Kaede",
|
| 11 |
+
"Kotone",
|
| 12 |
+
"Lynn",
|
| 13 |
+
"Mayu",
|
| 14 |
+
"NaKyoung",
|
| 15 |
+
"Nien",
|
| 16 |
+
"SeoAh",
|
| 17 |
+
"SeoYeon",
|
| 18 |
+
"ShiOn",
|
| 19 |
+
"SoHyun",
|
| 20 |
+
"SooMin",
|
| 21 |
+
"Sullin",
|
| 22 |
+
"Xinyu",
|
| 23 |
+
"YeonJi",
|
| 24 |
+
"YooYeon",
|
| 25 |
+
"YuBin"
|
| 26 |
+
]
|
Model-base/config.json
ADDED
|
@@ -0,0 +1,37 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"model_name": "tripleS-Member-Recognition base",
|
| 3 |
+
"description": "Base classifier model trained only on the tripleS Member Objekt dataset without any fine-tuning or external data.",
|
| 4 |
+
"fine_tuned": false,
|
| 5 |
+
"model_file": "face_model.pkl",
|
| 6 |
+
"embedding_model": "buffalo_l",
|
| 7 |
+
"format": "pkl",
|
| 8 |
+
"total_persons": 24,
|
| 9 |
+
"total_samples": 3480,
|
| 10 |
+
"num_classes": 24,
|
| 11 |
+
"person_names": [
|
| 12 |
+
"ChaeWon",
|
| 13 |
+
"ChaeYeon",
|
| 14 |
+
"DaHyun",
|
| 15 |
+
"HaYeon",
|
| 16 |
+
"HyeRin",
|
| 17 |
+
"JiWoo",
|
| 18 |
+
"JiYeon",
|
| 19 |
+
"JooBin",
|
| 20 |
+
"Kaede",
|
| 21 |
+
"Kotone",
|
| 22 |
+
"Lynn",
|
| 23 |
+
"Mayu",
|
| 24 |
+
"NaKyoung",
|
| 25 |
+
"Nien",
|
| 26 |
+
"SeoAh",
|
| 27 |
+
"SeoYeon",
|
| 28 |
+
"ShiOn",
|
| 29 |
+
"SoHyun",
|
| 30 |
+
"SooMin",
|
| 31 |
+
"Sullin",
|
| 32 |
+
"Xinyu",
|
| 33 |
+
"YeonJi",
|
| 34 |
+
"YooYeon",
|
| 35 |
+
"YuBin"
|
| 36 |
+
]
|
| 37 |
+
}
|
Model-base/member.json
ADDED
|
@@ -0,0 +1,26 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
[
|
| 2 |
+
"ChaeWon",
|
| 3 |
+
"ChaeYeon",
|
| 4 |
+
"DaHyun",
|
| 5 |
+
"HaYeon",
|
| 6 |
+
"HyeRin",
|
| 7 |
+
"JiWoo",
|
| 8 |
+
"JiYeon",
|
| 9 |
+
"JooBin",
|
| 10 |
+
"Kaede",
|
| 11 |
+
"Kotone",
|
| 12 |
+
"Lynn",
|
| 13 |
+
"Mayu",
|
| 14 |
+
"NaKyoung",
|
| 15 |
+
"Nien",
|
| 16 |
+
"SeoAh",
|
| 17 |
+
"SeoYeon",
|
| 18 |
+
"ShiOn",
|
| 19 |
+
"SoHyun",
|
| 20 |
+
"SooMin",
|
| 21 |
+
"Sullin",
|
| 22 |
+
"Xinyu",
|
| 23 |
+
"YeonJi",
|
| 24 |
+
"YooYeon",
|
| 25 |
+
"YuBin"
|
| 26 |
+
]
|
Model-base/svm_face_model.pkl
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:cf1c26d2704b8a3ec0b5b4df372d095bbfffc66fdf1add284de5349573e711a1
|
| 3 |
+
size 7690050
|
README.md
ADDED
|
@@ -0,0 +1,72 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
---
|
| 2 |
+
license: apache-2.0
|
| 3 |
+
language:
|
| 4 |
+
- en
|
| 5 |
+
pipeline_tag: image-classification
|
| 6 |
+
tags:
|
| 7 |
+
- face_recognition
|
| 8 |
+
- tripleS
|
| 9 |
+
- k-pop
|
| 10 |
+
- computer-vision
|
| 11 |
+
- face-classification
|
| 12 |
+
---
|
| 13 |
+
|
| 14 |
+
# tripleS Member Recognition Model
|
| 15 |
+
|
| 16 |
+
|
| 17 |
+
This repository contains face recognition models specifically trained to identify members of the K-pop girl group **tripleS**. The models are built using deep learning techniques with InsightFace embeddings and can classify faces of all 24 tripleS members with varying levels of precision and recall depending on the model version.
|
| 18 |
+
|
| 19 |
+
## Model Versions
|
| 20 |
+
|
| 21 |
+
This repository includes multiple model versions, each trained with different confidence thresholds to provide various trade-offs between precision and recall:
|
| 22 |
+
|
| 23 |
+
### Base Model (`Model-base/`)
|
| 24 |
+
- **Description**: Base classifier model trained only on the tripleS Member Objekt dataset without any fine-tuning or external data
|
| 25 |
+
- **Training samples**: 3,480 samples
|
| 26 |
+
- **Fine-tuned**: No
|
| 27 |
+
- **Use case**: Baseline performance reference
|
| 28 |
+
|
| 29 |
+
### Version 0.4 (`Model-0.4/`)
|
| 30 |
+
- **Confidence threshold**: ≥ 0.4
|
| 31 |
+
- **Description**: Trained using more inclusive data, resulting in a model that may generalize better but risks misclassification
|
| 32 |
+
- **Training samples**: 13,951 samples
|
| 33 |
+
- **Characteristics**: Higher recall, lower precision
|
| 34 |
+
- **Use case**: When you want to catch as many faces as possible, even with some false positives
|
| 35 |
+
|
| 36 |
+
### Version 0.5 (`Model-0.5/`)
|
| 37 |
+
- **Confidence threshold**: ≥ 0.5
|
| 38 |
+
- **Description**: Balanced training set offering a trade-off between recall and precision
|
| 39 |
+
- **Training samples**: 13,707 samples
|
| 40 |
+
- **Characteristics**: Balanced recall and precision
|
| 41 |
+
- **Use case**: General-purpose recognition with balanced performance
|
| 42 |
+
|
| 43 |
+
### Version 0.6 (`Model-0.6/`)
|
| 44 |
+
- **Confidence threshold**: ≥ 0.6
|
| 45 |
+
- **Description**: Trained only on high-confidence samples, producing a stricter classifier with higher precision
|
| 46 |
+
- **Training samples**: 13,358 samples
|
| 47 |
+
- **Characteristics**: Higher precision, lower recall
|
| 48 |
+
- **Use case**: When accuracy is critical and you want to minimize false positives
|
| 49 |
+
|
| 50 |
+
## Recognized Members
|
| 51 |
+
|
| 52 |
+
The model can identify all 24 tripleS members:
|
| 53 |
+
|
| 54 |
+
**Korean Members**: ChaeWon, ChaeYeon, DaHyun, HaYeon, HyeRin, JiWoo, JiYeon, JooBin, NaKyoung, SeoAh, SeoYeon, ShiOn, SoHyun, SooMin, YeonJi, YooYeon, YuBin
|
| 55 |
+
|
| 56 |
+
**Japanese Members**: Kaede, Kotone, Lynn, Mayu
|
| 57 |
+
|
| 58 |
+
**Chinese Members**: Nien, Sullin, Xinyu
|
| 59 |
+
|
| 60 |
+
## Model Files Structure
|
| 61 |
+
|
| 62 |
+
Each model version contains:
|
| 63 |
+
- `config.json`: Model configuration and metadata
|
| 64 |
+
- `face_model.pkl`: Trained model weights
|
| 65 |
+
- `face_model_names.json`: Label mapping (versions 0.4-0.6 only)
|
| 66 |
+
|
| 67 |
+
## Model Selection Guide
|
| 68 |
+
|
| 69 |
+
- **Choose Base Model** if you want the original performance without fine-tuning
|
| 70 |
+
- **Choose Version 0.4** if you prioritize detecting all possible members (high recall)
|
| 71 |
+
- **Choose Version 0.5** if you want balanced performance for general use cases
|
| 72 |
+
- **Choose Version 0.6** if you prioritize accuracy and want to minimize false identifications
|