dragon-notdragon / config.json
hadilq's picture
add training script and prediction
fb5d392 unverified
{
"name": "sequential_2",
"id2label": {
"0": "dragon",
"1": "not dragon"
},
"layers": [
{
"module": "keras.layers",
"class_name": "InputLayer",
"config": {
"batch_input_shape": [
null,
224,
224,
3
],
"dtype": "float32",
"sparse": false,
"ragged": false,
"name": "input_3"
},
"registered_name": null
},
{
"module": "keras.layers",
"class_name": "Conv2D",
"config": {
"name": "block1_conv1",
"trainable": true,
"dtype": "float32",
"filters": 64,
"kernel_size": [
3,
3
],
"strides": [
1,
1
],
"padding": "same",
"data_format": "channels_last",
"dilation_rate": [
1,
1
],
"groups": 1,
"activation": "relu",
"use_bias": true,
"kernel_initializer": {
"module": "keras.initializers",
"class_name": "GlorotUniform",
"config": {
"seed": null
},
"registered_name": null
},
"bias_initializer": {
"module": "keras.initializers",
"class_name": "Zeros",
"config": {},
"registered_name": null
},
"kernel_regularizer": null,
"bias_regularizer": null,
"activity_regularizer": null,
"kernel_constraint": null,
"bias_constraint": null
},
"registered_name": null,
"build_config": {
"input_shape": [
null,
224,
224,
3
]
}
},
{
"module": "keras.layers",
"class_name": "Conv2D",
"config": {
"name": "block1_conv2",
"trainable": true,
"dtype": "float32",
"filters": 64,
"kernel_size": [
3,
3
],
"strides": [
1,
1
],
"padding": "same",
"data_format": "channels_last",
"dilation_rate": [
1,
1
],
"groups": 1,
"activation": "relu",
"use_bias": true,
"kernel_initializer": {
"module": "keras.initializers",
"class_name": "GlorotUniform",
"config": {
"seed": null
},
"registered_name": null
},
"bias_initializer": {
"module": "keras.initializers",
"class_name": "Zeros",
"config": {},
"registered_name": null
},
"kernel_regularizer": null,
"bias_regularizer": null,
"activity_regularizer": null,
"kernel_constraint": null,
"bias_constraint": null
},
"registered_name": null,
"build_config": {
"input_shape": [
null,
224,
224,
64
]
}
},
{
"module": "keras.layers",
"class_name": "MaxPooling2D",
"config": {
"name": "block1_pool",
"trainable": true,
"dtype": "float32",
"pool_size": [
2,
2
],
"padding": "valid",
"strides": [
2,
2
],
"data_format": "channels_last"
},
"registered_name": null
},
{
"module": "keras.layers",
"class_name": "Conv2D",
"config": {
"name": "block2_conv1",
"trainable": true,
"dtype": "float32",
"filters": 128,
"kernel_size": [
3,
3
],
"strides": [
1,
1
],
"padding": "same",
"data_format": "channels_last",
"dilation_rate": [
1,
1
],
"groups": 1,
"activation": "relu",
"use_bias": true,
"kernel_initializer": {
"module": "keras.initializers",
"class_name": "GlorotUniform",
"config": {
"seed": null
},
"registered_name": null
},
"bias_initializer": {
"module": "keras.initializers",
"class_name": "Zeros",
"config": {},
"registered_name": null
},
"kernel_regularizer": null,
"bias_regularizer": null,
"activity_regularizer": null,
"kernel_constraint": null,
"bias_constraint": null
},
"registered_name": null,
"build_config": {
"input_shape": [
null,
112,
112,
64
]
}
},
{
"module": "keras.layers",
"class_name": "Conv2D",
"config": {
"name": "block2_conv2",
"trainable": true,
"dtype": "float32",
"filters": 128,
"kernel_size": [
3,
3
],
"strides": [
1,
1
],
"padding": "same",
"data_format": "channels_last",
"dilation_rate": [
1,
1
],
"groups": 1,
"activation": "relu",
"use_bias": true,
"kernel_initializer": {
"module": "keras.initializers",
"class_name": "GlorotUniform",
"config": {
"seed": null
},
"registered_name": null
},
"bias_initializer": {
"module": "keras.initializers",
"class_name": "Zeros",
"config": {},
"registered_name": null
},
"kernel_regularizer": null,
"bias_regularizer": null,
"activity_regularizer": null,
"kernel_constraint": null,
"bias_constraint": null
},
"registered_name": null,
"build_config": {
"input_shape": [
null,
112,
112,
128
]
}
},
{
"module": "keras.layers",
"class_name": "MaxPooling2D",
"config": {
"name": "block2_pool",
"trainable": true,
"dtype": "float32",
"pool_size": [
2,
2
],
"padding": "valid",
"strides": [
2,
2
],
"data_format": "channels_last"
},
"registered_name": null
},
{
"module": "keras.layers",
"class_name": "Conv2D",
"config": {
"name": "block3_conv1",
"trainable": true,
"dtype": "float32",
"filters": 256,
"kernel_size": [
3,
3
],
"strides": [
1,
1
],
"padding": "same",
"data_format": "channels_last",
"dilation_rate": [
1,
1
],
"groups": 1,
"activation": "relu",
"use_bias": true,
"kernel_initializer": {
"module": "keras.initializers",
"class_name": "GlorotUniform",
"config": {
"seed": null
},
"registered_name": null
},
"bias_initializer": {
"module": "keras.initializers",
"class_name": "Zeros",
"config": {},
"registered_name": null
},
"kernel_regularizer": null,
"bias_regularizer": null,
"activity_regularizer": null,
"kernel_constraint": null,
"bias_constraint": null
},
"registered_name": null,
"build_config": {
"input_shape": [
null,
56,
56,
128
]
}
},
{
"module": "keras.layers",
"class_name": "Conv2D",
"config": {
"name": "block3_conv2",
"trainable": true,
"dtype": "float32",
"filters": 256,
"kernel_size": [
3,
3
],
"strides": [
1,
1
],
"padding": "same",
"data_format": "channels_last",
"dilation_rate": [
1,
1
],
"groups": 1,
"activation": "relu",
"use_bias": true,
"kernel_initializer": {
"module": "keras.initializers",
"class_name": "GlorotUniform",
"config": {
"seed": null
},
"registered_name": null
},
"bias_initializer": {
"module": "keras.initializers",
"class_name": "Zeros",
"config": {},
"registered_name": null
},
"kernel_regularizer": null,
"bias_regularizer": null,
"activity_regularizer": null,
"kernel_constraint": null,
"bias_constraint": null
},
"registered_name": null,
"build_config": {
"input_shape": [
null,
56,
56,
256
]
}
},
{
"module": "keras.layers",
"class_name": "Conv2D",
"config": {
"name": "block3_conv3",
"trainable": true,
"dtype": "float32",
"filters": 256,
"kernel_size": [
3,
3
],
"strides": [
1,
1
],
"padding": "same",
"data_format": "channels_last",
"dilation_rate": [
1,
1
],
"groups": 1,
"activation": "relu",
"use_bias": true,
"kernel_initializer": {
"module": "keras.initializers",
"class_name": "GlorotUniform",
"config": {
"seed": null
},
"registered_name": null
},
"bias_initializer": {
"module": "keras.initializers",
"class_name": "Zeros",
"config": {},
"registered_name": null
},
"kernel_regularizer": null,
"bias_regularizer": null,
"activity_regularizer": null,
"kernel_constraint": null,
"bias_constraint": null
},
"registered_name": null,
"build_config": {
"input_shape": [
null,
56,
56,
256
]
}
},
{
"module": "keras.layers",
"class_name": "MaxPooling2D",
"config": {
"name": "block3_pool",
"trainable": true,
"dtype": "float32",
"pool_size": [
2,
2
],
"padding": "valid",
"strides": [
2,
2
],
"data_format": "channels_last"
},
"registered_name": null
},
{
"module": "keras.layers",
"class_name": "Conv2D",
"config": {
"name": "block4_conv1",
"trainable": true,
"dtype": "float32",
"filters": 512,
"kernel_size": [
3,
3
],
"strides": [
1,
1
],
"padding": "same",
"data_format": "channels_last",
"dilation_rate": [
1,
1
],
"groups": 1,
"activation": "relu",
"use_bias": true,
"kernel_initializer": {
"module": "keras.initializers",
"class_name": "GlorotUniform",
"config": {
"seed": null
},
"registered_name": null
},
"bias_initializer": {
"module": "keras.initializers",
"class_name": "Zeros",
"config": {},
"registered_name": null
},
"kernel_regularizer": null,
"bias_regularizer": null,
"activity_regularizer": null,
"kernel_constraint": null,
"bias_constraint": null
},
"registered_name": null,
"build_config": {
"input_shape": [
null,
28,
28,
256
]
}
},
{
"module": "keras.layers",
"class_name": "Conv2D",
"config": {
"name": "block4_conv2",
"trainable": true,
"dtype": "float32",
"filters": 512,
"kernel_size": [
3,
3
],
"strides": [
1,
1
],
"padding": "same",
"data_format": "channels_last",
"dilation_rate": [
1,
1
],
"groups": 1,
"activation": "relu",
"use_bias": true,
"kernel_initializer": {
"module": "keras.initializers",
"class_name": "GlorotUniform",
"config": {
"seed": null
},
"registered_name": null
},
"bias_initializer": {
"module": "keras.initializers",
"class_name": "Zeros",
"config": {},
"registered_name": null
},
"kernel_regularizer": null,
"bias_regularizer": null,
"activity_regularizer": null,
"kernel_constraint": null,
"bias_constraint": null
},
"registered_name": null,
"build_config": {
"input_shape": [
null,
28,
28,
512
]
}
},
{
"module": "keras.layers",
"class_name": "Conv2D",
"config": {
"name": "block4_conv3",
"trainable": true,
"dtype": "float32",
"filters": 512,
"kernel_size": [
3,
3
],
"strides": [
1,
1
],
"padding": "same",
"data_format": "channels_last",
"dilation_rate": [
1,
1
],
"groups": 1,
"activation": "relu",
"use_bias": true,
"kernel_initializer": {
"module": "keras.initializers",
"class_name": "GlorotUniform",
"config": {
"seed": null
},
"registered_name": null
},
"bias_initializer": {
"module": "keras.initializers",
"class_name": "Zeros",
"config": {},
"registered_name": null
},
"kernel_regularizer": null,
"bias_regularizer": null,
"activity_regularizer": null,
"kernel_constraint": null,
"bias_constraint": null
},
"registered_name": null,
"build_config": {
"input_shape": [
null,
28,
28,
512
]
}
},
{
"module": "keras.layers",
"class_name": "MaxPooling2D",
"config": {
"name": "block4_pool",
"trainable": true,
"dtype": "float32",
"pool_size": [
2,
2
],
"padding": "valid",
"strides": [
2,
2
],
"data_format": "channels_last"
},
"registered_name": null
},
{
"module": "keras.layers",
"class_name": "Conv2D",
"config": {
"name": "block5_conv1",
"trainable": true,
"dtype": "float32",
"filters": 512,
"kernel_size": [
3,
3
],
"strides": [
1,
1
],
"padding": "same",
"data_format": "channels_last",
"dilation_rate": [
1,
1
],
"groups": 1,
"activation": "relu",
"use_bias": true,
"kernel_initializer": {
"module": "keras.initializers",
"class_name": "GlorotUniform",
"config": {
"seed": null
},
"registered_name": null
},
"bias_initializer": {
"module": "keras.initializers",
"class_name": "Zeros",
"config": {},
"registered_name": null
},
"kernel_regularizer": null,
"bias_regularizer": null,
"activity_regularizer": null,
"kernel_constraint": null,
"bias_constraint": null
},
"registered_name": null,
"build_config": {
"input_shape": [
null,
14,
14,
512
]
}
},
{
"module": "keras.layers",
"class_name": "Conv2D",
"config": {
"name": "block5_conv2",
"trainable": true,
"dtype": "float32",
"filters": 512,
"kernel_size": [
3,
3
],
"strides": [
1,
1
],
"padding": "same",
"data_format": "channels_last",
"dilation_rate": [
1,
1
],
"groups": 1,
"activation": "relu",
"use_bias": true,
"kernel_initializer": {
"module": "keras.initializers",
"class_name": "GlorotUniform",
"config": {
"seed": null
},
"registered_name": null
},
"bias_initializer": {
"module": "keras.initializers",
"class_name": "Zeros",
"config": {},
"registered_name": null
},
"kernel_regularizer": null,
"bias_regularizer": null,
"activity_regularizer": null,
"kernel_constraint": null,
"bias_constraint": null
},
"registered_name": null,
"build_config": {
"input_shape": [
null,
14,
14,
512
]
}
},
{
"module": "keras.layers",
"class_name": "Conv2D",
"config": {
"name": "block5_conv3",
"trainable": true,
"dtype": "float32",
"filters": 512,
"kernel_size": [
3,
3
],
"strides": [
1,
1
],
"padding": "same",
"data_format": "channels_last",
"dilation_rate": [
1,
1
],
"groups": 1,
"activation": "relu",
"use_bias": true,
"kernel_initializer": {
"module": "keras.initializers",
"class_name": "GlorotUniform",
"config": {
"seed": null
},
"registered_name": null
},
"bias_initializer": {
"module": "keras.initializers",
"class_name": "Zeros",
"config": {},
"registered_name": null
},
"kernel_regularizer": null,
"bias_regularizer": null,
"activity_regularizer": null,
"kernel_constraint": null,
"bias_constraint": null
},
"registered_name": null,
"build_config": {
"input_shape": [
null,
14,
14,
512
]
}
},
{
"module": "keras.layers",
"class_name": "MaxPooling2D",
"config": {
"name": "block5_pool",
"trainable": true,
"dtype": "float32",
"pool_size": [
2,
2
],
"padding": "valid",
"strides": [
2,
2
],
"data_format": "channels_last"
},
"registered_name": null
},
{
"module": "keras.layers",
"class_name": "Flatten",
"config": {
"name": "flatten",
"trainable": true,
"dtype": "float32",
"data_format": "channels_last"
},
"registered_name": null
},
{
"module": "keras.layers",
"class_name": "Dense",
"config": {
"name": "fc1",
"trainable": true,
"dtype": "float32",
"units": 4096,
"activation": "relu",
"use_bias": true,
"kernel_initializer": {
"module": "keras.initializers",
"class_name": "GlorotUniform",
"config": {
"seed": null
},
"registered_name": null
},
"bias_initializer": {
"module": "keras.initializers",
"class_name": "Zeros",
"config": {},
"registered_name": null
},
"kernel_regularizer": null,
"bias_regularizer": null,
"activity_regularizer": null,
"kernel_constraint": null,
"bias_constraint": null
},
"registered_name": null,
"build_config": {
"input_shape": [
null,
25088
]
}
},
{
"module": "keras.layers",
"class_name": "Dense",
"config": {
"name": "fc2",
"trainable": true,
"dtype": "float32",
"units": 4096,
"activation": "relu",
"use_bias": true,
"kernel_initializer": {
"module": "keras.initializers",
"class_name": "GlorotUniform",
"config": {
"seed": null
},
"registered_name": null
},
"bias_initializer": {
"module": "keras.initializers",
"class_name": "Zeros",
"config": {},
"registered_name": null
},
"kernel_regularizer": null,
"bias_regularizer": null,
"activity_regularizer": null,
"kernel_constraint": null,
"bias_constraint": null
},
"registered_name": null,
"build_config": {
"input_shape": [
null,
4096
]
}
},
{
"module": "keras.layers",
"class_name": "Dense",
"config": {
"name": "dense_3",
"trainable": true,
"dtype": "float32",
"units": 2,
"activation": "softmax",
"use_bias": true,
"kernel_initializer": {
"module": "keras.initializers",
"class_name": "GlorotUniform",
"config": {
"seed": null
},
"registered_name": null
},
"bias_initializer": {
"module": "keras.initializers",
"class_name": "Zeros",
"config": {},
"registered_name": null
},
"kernel_regularizer": null,
"bias_regularizer": null,
"activity_regularizer": null,
"kernel_constraint": null,
"bias_constraint": null
},
"registered_name": null,
"build_config": {
"input_shape": [
null,
4096
]
}
}
]
}