mirror of
https://github.com/deepinsight/insightface.git
synced 2025-12-30 08:02:27 +00:00
324 lines
6.7 KiB
JSON
324 lines
6.7 KiB
JSON
{
|
|
"nodes": [
|
|
{
|
|
"op": "null",
|
|
"param": {},
|
|
"name": "data",
|
|
"inputs": [],
|
|
"backward_source_id": -1
|
|
},
|
|
{
|
|
"op": "null",
|
|
"param": {},
|
|
"name": "conv1_weight",
|
|
"inputs": [],
|
|
"backward_source_id": -1
|
|
},
|
|
{
|
|
"op": "null",
|
|
"param": {},
|
|
"name": "conv1_bias",
|
|
"inputs": [],
|
|
"backward_source_id": -1
|
|
},
|
|
{
|
|
"op": "Convolution",
|
|
"param": {
|
|
"cudnn_off": "False",
|
|
"cudnn_tune": "off",
|
|
"dilate": "(1,1)",
|
|
"kernel": "(3,3)",
|
|
"no_bias": "False",
|
|
"num_filter": "28",
|
|
"num_group": "1",
|
|
"pad": "(0,0)",
|
|
"stride": "(1,1)",
|
|
"workspace": "1024"
|
|
},
|
|
"name": "conv1",
|
|
"inputs": [[0, 0], [1, 0], [2, 0]],
|
|
"backward_source_id": -1
|
|
},
|
|
{
|
|
"op": "null",
|
|
"param": {},
|
|
"name": "prelu1_gamma",
|
|
"inputs": [],
|
|
"backward_source_id": -1
|
|
},
|
|
{
|
|
"op": "LeakyReLU",
|
|
"param": {
|
|
"act_type": "prelu",
|
|
"lower_bound": "0.125",
|
|
"slope": "0.25",
|
|
"upper_bound": "0.334"
|
|
},
|
|
"name": "prelu1",
|
|
"inputs": [[3, 0], [4, 0]],
|
|
"backward_source_id": -1
|
|
},
|
|
{
|
|
"op": "Pooling",
|
|
"param": {
|
|
"global_pool": "False",
|
|
"kernel": "(3,3)",
|
|
"pad": "(0,0)",
|
|
"pool_type": "max",
|
|
"pooling_convention": "full",
|
|
"stride": "(2,2)"
|
|
},
|
|
"name": "pool1",
|
|
"inputs": [[5, 0]],
|
|
"backward_source_id": -1
|
|
},
|
|
{
|
|
"op": "null",
|
|
"param": {},
|
|
"name": "conv2_weight",
|
|
"inputs": [],
|
|
"backward_source_id": -1
|
|
},
|
|
{
|
|
"op": "null",
|
|
"param": {},
|
|
"name": "conv2_bias",
|
|
"inputs": [],
|
|
"backward_source_id": -1
|
|
},
|
|
{
|
|
"op": "Convolution",
|
|
"param": {
|
|
"cudnn_off": "False",
|
|
"cudnn_tune": "off",
|
|
"dilate": "(1,1)",
|
|
"kernel": "(3,3)",
|
|
"no_bias": "False",
|
|
"num_filter": "48",
|
|
"num_group": "1",
|
|
"pad": "(0,0)",
|
|
"stride": "(1,1)",
|
|
"workspace": "1024"
|
|
},
|
|
"name": "conv2",
|
|
"inputs": [[6, 0], [7, 0], [8, 0]],
|
|
"backward_source_id": -1
|
|
},
|
|
{
|
|
"op": "null",
|
|
"param": {},
|
|
"name": "prelu2_gamma",
|
|
"inputs": [],
|
|
"backward_source_id": -1
|
|
},
|
|
{
|
|
"op": "LeakyReLU",
|
|
"param": {
|
|
"act_type": "prelu",
|
|
"lower_bound": "0.125",
|
|
"slope": "0.25",
|
|
"upper_bound": "0.334"
|
|
},
|
|
"name": "prelu2",
|
|
"inputs": [[9, 0], [10, 0]],
|
|
"backward_source_id": -1
|
|
},
|
|
{
|
|
"op": "Pooling",
|
|
"param": {
|
|
"global_pool": "False",
|
|
"kernel": "(3,3)",
|
|
"pad": "(0,0)",
|
|
"pool_type": "max",
|
|
"pooling_convention": "full",
|
|
"stride": "(2,2)"
|
|
},
|
|
"name": "pool2",
|
|
"inputs": [[11, 0]],
|
|
"backward_source_id": -1
|
|
},
|
|
{
|
|
"op": "null",
|
|
"param": {},
|
|
"name": "conv3_weight",
|
|
"inputs": [],
|
|
"backward_source_id": -1
|
|
},
|
|
{
|
|
"op": "null",
|
|
"param": {},
|
|
"name": "conv3_bias",
|
|
"inputs": [],
|
|
"backward_source_id": -1
|
|
},
|
|
{
|
|
"op": "Convolution",
|
|
"param": {
|
|
"cudnn_off": "False",
|
|
"cudnn_tune": "off",
|
|
"dilate": "(1,1)",
|
|
"kernel": "(2,2)",
|
|
"no_bias": "False",
|
|
"num_filter": "64",
|
|
"num_group": "1",
|
|
"pad": "(0,0)",
|
|
"stride": "(1,1)",
|
|
"workspace": "1024"
|
|
},
|
|
"name": "conv3",
|
|
"inputs": [[12, 0], [13, 0], [14, 0]],
|
|
"backward_source_id": -1
|
|
},
|
|
{
|
|
"op": "null",
|
|
"param": {},
|
|
"name": "prelu3_gamma",
|
|
"inputs": [],
|
|
"backward_source_id": -1
|
|
},
|
|
{
|
|
"op": "LeakyReLU",
|
|
"param": {
|
|
"act_type": "prelu",
|
|
"lower_bound": "0.125",
|
|
"slope": "0.25",
|
|
"upper_bound": "0.334"
|
|
},
|
|
"name": "prelu3",
|
|
"inputs": [[15, 0], [16, 0]],
|
|
"backward_source_id": -1
|
|
},
|
|
{
|
|
"op": "null",
|
|
"param": {},
|
|
"name": "conv4_weight",
|
|
"inputs": [],
|
|
"backward_source_id": -1
|
|
},
|
|
{
|
|
"op": "null",
|
|
"param": {},
|
|
"name": "conv4_bias",
|
|
"inputs": [],
|
|
"backward_source_id": -1
|
|
},
|
|
{
|
|
"op": "FullyConnected",
|
|
"param": {
|
|
"no_bias": "False",
|
|
"num_hidden": "128"
|
|
},
|
|
"name": "conv4",
|
|
"inputs": [[17, 0], [18, 0], [19, 0]],
|
|
"backward_source_id": -1
|
|
},
|
|
{
|
|
"op": "null",
|
|
"param": {},
|
|
"name": "prelu4_gamma",
|
|
"inputs": [],
|
|
"backward_source_id": -1
|
|
},
|
|
{
|
|
"op": "LeakyReLU",
|
|
"param": {
|
|
"act_type": "prelu",
|
|
"lower_bound": "0.125",
|
|
"slope": "0.25",
|
|
"upper_bound": "0.334"
|
|
},
|
|
"name": "prelu4",
|
|
"inputs": [[20, 0], [21, 0]],
|
|
"backward_source_id": -1
|
|
},
|
|
{
|
|
"op": "null",
|
|
"param": {},
|
|
"name": "conv5_2_weight",
|
|
"inputs": [],
|
|
"backward_source_id": -1
|
|
},
|
|
{
|
|
"op": "null",
|
|
"param": {},
|
|
"name": "conv5_2_bias",
|
|
"inputs": [],
|
|
"backward_source_id": -1
|
|
},
|
|
{
|
|
"op": "FullyConnected",
|
|
"param": {
|
|
"no_bias": "False",
|
|
"num_hidden": "4"
|
|
},
|
|
"name": "conv5_2",
|
|
"inputs": [[22, 0], [23, 0], [24, 0]],
|
|
"backward_source_id": -1
|
|
},
|
|
{
|
|
"op": "null",
|
|
"param": {},
|
|
"name": "conv5_1_weight",
|
|
"inputs": [],
|
|
"backward_source_id": -1
|
|
},
|
|
{
|
|
"op": "null",
|
|
"param": {},
|
|
"name": "conv5_1_bias",
|
|
"inputs": [],
|
|
"backward_source_id": -1
|
|
},
|
|
{
|
|
"op": "FullyConnected",
|
|
"param": {
|
|
"no_bias": "False",
|
|
"num_hidden": "2"
|
|
},
|
|
"name": "conv5_1",
|
|
"inputs": [[22, 0], [26, 0], [27, 0]],
|
|
"backward_source_id": -1
|
|
},
|
|
{
|
|
"op": "null",
|
|
"param": {},
|
|
"name": "prob1_label",
|
|
"inputs": [],
|
|
"backward_source_id": -1
|
|
},
|
|
{
|
|
"op": "SoftmaxOutput",
|
|
"param": {
|
|
"grad_scale": "1",
|
|
"ignore_label": "-1",
|
|
"multi_output": "False",
|
|
"normalization": "null",
|
|
"use_ignore": "False"
|
|
},
|
|
"name": "prob1",
|
|
"inputs": [[28, 0], [29, 0]],
|
|
"backward_source_id": -1
|
|
}
|
|
],
|
|
"arg_nodes": [
|
|
0,
|
|
1,
|
|
2,
|
|
4,
|
|
7,
|
|
8,
|
|
10,
|
|
13,
|
|
14,
|
|
16,
|
|
18,
|
|
19,
|
|
21,
|
|
23,
|
|
24,
|
|
26,
|
|
27,
|
|
29
|
|
],
|
|
"heads": [[25, 0], [30, 0]]
|
|
} |