AldyRevigustian commited on
Commit
27840cb
·
0 Parent(s):

Upload model and supporting files

Browse files
.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