mirror of
https://github.com/deepinsight/insightface.git
synced 2025-12-30 08:02:27 +00:00
266 lines
5.6 KiB
JSON
266 lines
5.6 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": "10",
|
|
"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": "(2,2)",
|
|
"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": "16",
|
|
"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": "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": "(3,3)",
|
|
"no_bias": "False",
|
|
"num_filter": "32",
|
|
"num_group": "1",
|
|
"pad": "(0,0)",
|
|
"stride": "(1,1)",
|
|
"workspace": "1024"
|
|
},
|
|
"name": "conv3",
|
|
"inputs": [[11, 0], [12, 0], [13, 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": [[14, 0], [15, 0]],
|
|
"backward_source_id": -1
|
|
},
|
|
{
|
|
"op": "null",
|
|
"param": {},
|
|
"name": "conv4_2_weight",
|
|
"inputs": [],
|
|
"backward_source_id": -1
|
|
},
|
|
{
|
|
"op": "null",
|
|
"param": {},
|
|
"name": "conv4_2_bias",
|
|
"inputs": [],
|
|
"backward_source_id": -1
|
|
},
|
|
{
|
|
"op": "Convolution",
|
|
"param": {
|
|
"cudnn_off": "False",
|
|
"cudnn_tune": "off",
|
|
"dilate": "(1,1)",
|
|
"kernel": "(1,1)",
|
|
"no_bias": "False",
|
|
"num_filter": "4",
|
|
"num_group": "1",
|
|
"pad": "(0,0)",
|
|
"stride": "(1,1)",
|
|
"workspace": "1024"
|
|
},
|
|
"name": "conv4_2",
|
|
"inputs": [[16, 0], [17, 0], [18, 0]],
|
|
"backward_source_id": -1
|
|
},
|
|
{
|
|
"op": "null",
|
|
"param": {},
|
|
"name": "conv4_1_weight",
|
|
"inputs": [],
|
|
"backward_source_id": -1
|
|
},
|
|
{
|
|
"op": "null",
|
|
"param": {},
|
|
"name": "conv4_1_bias",
|
|
"inputs": [],
|
|
"backward_source_id": -1
|
|
},
|
|
{
|
|
"op": "Convolution",
|
|
"param": {
|
|
"cudnn_off": "False",
|
|
"cudnn_tune": "off",
|
|
"dilate": "(1,1)",
|
|
"kernel": "(1,1)",
|
|
"no_bias": "False",
|
|
"num_filter": "2",
|
|
"num_group": "1",
|
|
"pad": "(0,0)",
|
|
"stride": "(1,1)",
|
|
"workspace": "1024"
|
|
},
|
|
"name": "conv4_1",
|
|
"inputs": [[16, 0], [20, 0], [21, 0]],
|
|
"backward_source_id": -1
|
|
},
|
|
{
|
|
"op": "SoftmaxActivation",
|
|
"param": {"mode": "channel"},
|
|
"name": "prob1",
|
|
"inputs": [[22, 0]],
|
|
"backward_source_id": -1
|
|
}
|
|
],
|
|
"arg_nodes": [
|
|
0,
|
|
1,
|
|
2,
|
|
4,
|
|
7,
|
|
8,
|
|
10,
|
|
12,
|
|
13,
|
|
15,
|
|
17,
|
|
18,
|
|
20,
|
|
21
|
|
],
|
|
"heads": [[19, 0], [23, 0]]
|
|
} |