justin-onda commited on
Commit
3662de4
·
1 Parent(s): 898e848

누락된 model_info.json 파일 다시 추가

Browse files
Files changed (1) hide show
  1. model_info.json +112 -0
model_info.json ADDED
@@ -0,0 +1,112 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "model_architecture": {
3
+ "backbone": "vit_large_patch14_dinov2.lvd142m",
4
+ "feature_dim": 1024,
5
+ "total_parameters": 303252502,
6
+ "trainable_parameters": 24598,
7
+ "freeze_backbone": true
8
+ },
9
+ "input_specification": {
10
+ "image_size": [
11
+ 224,
12
+ 224
13
+ ],
14
+ "channels": 3,
15
+ "pixel_range": [
16
+ 0.0,
17
+ 1.0
18
+ ],
19
+ "normalization": {
20
+ "mean": [
21
+ 0.485,
22
+ 0.456,
23
+ 0.406
24
+ ],
25
+ "std": [
26
+ 0.229,
27
+ 0.224,
28
+ 0.225
29
+ ],
30
+ "description": "ImageNet normalization for DINOv2"
31
+ },
32
+ "input_format": "RGB",
33
+ "tensor_layout": "NCHW"
34
+ },
35
+ "output_specification": {
36
+ "heads": {
37
+ "scene": {
38
+ "num_classes": 6,
39
+ "output_type": "logits",
40
+ "activation": "softmax",
41
+ "classes": [
42
+ 16000001,
43
+ 16000002,
44
+ 16000006,
45
+ 16000008,
46
+ 16000009,
47
+ 16000011
48
+ ]
49
+ },
50
+ "concept": {
51
+ "num_classes": 3,
52
+ "output_type": "logits",
53
+ "activation": "softmax",
54
+ "classes": [
55
+ 17000001,
56
+ 17000002,
57
+ 17000003
58
+ ]
59
+ },
60
+ "object": {
61
+ "num_classes": 13,
62
+ "output_type": "logits",
63
+ "activation": "softmax",
64
+ "classes": [
65
+ 18000001,
66
+ 18000002,
67
+ 18000004,
68
+ 18000005,
69
+ 18000006,
70
+ 18000007,
71
+ 18000008,
72
+ 18000009,
73
+ 18000010,
74
+ 18000012,
75
+ 18000014,
76
+ 18000016,
77
+ "unclassified"
78
+ ]
79
+ }
80
+ }
81
+ },
82
+ "class_mappings": {
83
+ "scene": {
84
+ "0": 16000001,
85
+ "1": 16000002,
86
+ "2": 16000006,
87
+ "3": 16000008,
88
+ "4": 16000009,
89
+ "5": 16000011
90
+ },
91
+ "concept": {
92
+ "0": 17000001,
93
+ "1": 17000002,
94
+ "2": 17000003
95
+ },
96
+ "object": {
97
+ "0": 18000001,
98
+ "1": 18000002,
99
+ "2": 18000004,
100
+ "3": 18000005,
101
+ "4": 18000006,
102
+ "5": 18000007,
103
+ "6": 18000008,
104
+ "7": 18000009,
105
+ "8": 18000010,
106
+ "9": 18000012,
107
+ "10": 18000014,
108
+ "11": 18000016,
109
+ "12": "unclassified"
110
+ }
111
+ }
112
+ }