KerasHub
basnet_duts / task.json
Divyasreepat's picture
Upload folder using huggingface_hub
2d94dbc verified
{
"module": "keras_hub.src.models.basnet.basnet",
"class_name": "BASNetImageSegmenter",
"config": {
"backbone": {
"module": "keras_hub.src.models.basnet.basnet_backbone",
"class_name": "BASNetBackbone",
"config": {
"name": "bas_net_backbone",
"trainable": true,
"image_encoder": {
"module": "keras_hub.src.models.resnet.resnet_backbone",
"class_name": "ResNetBackbone",
"config": {
"name": "res_net_backbone",
"trainable": true,
"input_conv_filters": [
64
],
"input_conv_kernel_sizes": [
7
],
"stackwise_num_filters": [
64,
128,
256,
512
],
"stackwise_num_blocks": [
3,
4,
6,
3
],
"stackwise_num_strides": [
1,
2,
2,
2
],
"block_type": "basic_block",
"use_pre_activation": false,
"image_shape": [
null,
null,
3
]
},
"registered_name": "keras_hub>ResNetBackbone"
},
"num_classes": 1,
"image_shape": [
null,
null,
3
],
"projection_filters": 64,
"prediction_heads": [
{
"module": "keras",
"class_name": "Sequential",
"config": {
"name": "sequential",
"trainable": true,
"dtype": {
"module": "keras",
"class_name": "DTypePolicy",
"config": {
"name": "float32"
},
"registered_name": null
},
"layers": [
{
"module": "keras.layers",
"class_name": "InputLayer",
"config": {
"batch_shape": [
null,
null,
null,
512
],
"dtype": "float32",
"sparse": false,
"name": "input_layer_2"
},
"registered_name": null
},
{
"module": "keras.layers",
"class_name": "Conv2D",
"config": {
"name": "conv2d",
"trainable": true,
"dtype": {
"module": "keras",
"class_name": "DTypePolicy",
"config": {
"name": "float32"
},
"registered_name": null
},
"filters": 1,
"kernel_size": [
3,
3
],
"strides": [
1,
1
],
"padding": "same",
"data_format": "channels_last",
"dilation_rate": [
1,
1
],
"groups": 1,
"activation": "linear",
"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,
null,
null,
512
]
}
}
],
"build_input_shape": [
null,
null,
null,
512
]
},
"registered_name": null,
"build_config": {
"input_shape": [
null,
null,
null,
512
]
}
},
{
"module": "keras",
"class_name": "Sequential",
"config": {
"name": "sequential_1",
"trainable": true,
"dtype": {
"module": "keras",
"class_name": "DTypePolicy",
"config": {
"name": "float32"
},
"registered_name": null
},
"layers": [
{
"module": "keras.layers",
"class_name": "InputLayer",
"config": {
"batch_shape": [
null,
null,
null,
512
],
"dtype": "float32",
"sparse": false,
"name": "input_layer_3"
},
"registered_name": null
},
{
"module": "keras.layers",
"class_name": "Conv2D",
"config": {
"name": "conv2d_1",
"trainable": true,
"dtype": {
"module": "keras",
"class_name": "DTypePolicy",
"config": {
"name": "float32"
},
"registered_name": null
},
"filters": 1,
"kernel_size": [
3,
3
],
"strides": [
1,
1
],
"padding": "same",
"data_format": "channels_last",
"dilation_rate": [
1,
1
],
"groups": 1,
"activation": "linear",
"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,
null,
null,
512
]
}
},
{
"module": "keras.layers",
"class_name": "UpSampling2D",
"config": {
"name": "up_sampling2d",
"trainable": true,
"dtype": {
"module": "keras",
"class_name": "DTypePolicy",
"config": {
"name": "float32"
},
"registered_name": null
},
"size": [
2,
2
],
"data_format": "channels_last",
"interpolation": "bilinear"
},
"registered_name": null,
"build_config": {
"input_shape": [
null,
null,
null,
1
]
}
}
],
"build_input_shape": [
null,
null,
null,
512
]
},
"registered_name": null,
"build_config": {
"input_shape": [
null,
null,
null,
512
]
}
},
{
"module": "keras",
"class_name": "Sequential",
"config": {
"name": "sequential_2",
"trainable": true,
"dtype": {
"module": "keras",
"class_name": "DTypePolicy",
"config": {
"name": "float32"
},
"registered_name": null
},
"layers": [
{
"module": "keras.layers",
"class_name": "InputLayer",
"config": {
"batch_shape": [
null,
null,
null,
512
],
"dtype": "float32",
"sparse": false,
"name": "input_layer_4"
},
"registered_name": null
},
{
"module": "keras.layers",
"class_name": "Conv2D",
"config": {
"name": "conv2d_2",
"trainable": true,
"dtype": {
"module": "keras",
"class_name": "DTypePolicy",
"config": {
"name": "float32"
},
"registered_name": null
},
"filters": 1,
"kernel_size": [
3,
3
],
"strides": [
1,
1
],
"padding": "same",
"data_format": "channels_last",
"dilation_rate": [
1,
1
],
"groups": 1,
"activation": "linear",
"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,
null,
null,
512
]
}
},
{
"module": "keras.layers",
"class_name": "UpSampling2D",
"config": {
"name": "up_sampling2d_1",
"trainable": true,
"dtype": {
"module": "keras",
"class_name": "DTypePolicy",
"config": {
"name": "float32"
},
"registered_name": null
},
"size": [
4,
4
],
"data_format": "channels_last",
"interpolation": "bilinear"
},
"registered_name": null,
"build_config": {
"input_shape": [
null,
null,
null,
1
]
}
}
],
"build_input_shape": [
null,
null,
null,
512
]
},
"registered_name": null,
"build_config": {
"input_shape": [
null,
null,
null,
512
]
}
},
{
"module": "keras",
"class_name": "Sequential",
"config": {
"name": "sequential_3",
"trainable": true,
"dtype": {
"module": "keras",
"class_name": "DTypePolicy",
"config": {
"name": "float32"
},
"registered_name": null
},
"layers": [
{
"module": "keras.layers",
"class_name": "InputLayer",
"config": {
"batch_shape": [
null,
null,
null,
512
],
"dtype": "float32",
"sparse": false,
"name": "input_layer_5"
},
"registered_name": null
},
{
"module": "keras.layers",
"class_name": "Conv2D",
"config": {
"name": "conv2d_3",
"trainable": true,
"dtype": {
"module": "keras",
"class_name": "DTypePolicy",
"config": {
"name": "float32"
},
"registered_name": null
},
"filters": 1,
"kernel_size": [
3,
3
],
"strides": [
1,
1
],
"padding": "same",
"data_format": "channels_last",
"dilation_rate": [
1,
1
],
"groups": 1,
"activation": "linear",
"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,
null,
null,
512
]
}
},
{
"module": "keras.layers",
"class_name": "UpSampling2D",
"config": {
"name": "up_sampling2d_2",
"trainable": true,
"dtype": {
"module": "keras",
"class_name": "DTypePolicy",
"config": {
"name": "float32"
},
"registered_name": null
},
"size": [
8,
8
],
"data_format": "channels_last",
"interpolation": "bilinear"
},
"registered_name": null,
"build_config": {
"input_shape": [
null,
null,
null,
1
]
}
}
],
"build_input_shape": [
null,
null,
null,
512
]
},
"registered_name": null,
"build_config": {
"input_shape": [
null,
null,
null,
512
]
}
},
{
"module": "keras",
"class_name": "Sequential",
"config": {
"name": "sequential_4",
"trainable": true,
"dtype": {
"module": "keras",
"class_name": "DTypePolicy",
"config": {
"name": "float32"
},
"registered_name": null
},
"layers": [
{
"module": "keras.layers",
"class_name": "InputLayer",
"config": {
"batch_shape": [
null,
null,
null,
512
],
"dtype": "float32",
"sparse": false,
"name": "input_layer_6"
},
"registered_name": null
},
{
"module": "keras.layers",
"class_name": "Conv2D",
"config": {
"name": "conv2d_4",
"trainable": true,
"dtype": {
"module": "keras",
"class_name": "DTypePolicy",
"config": {
"name": "float32"
},
"registered_name": null
},
"filters": 1,
"kernel_size": [
3,
3
],
"strides": [
1,
1
],
"padding": "same",
"data_format": "channels_last",
"dilation_rate": [
1,
1
],
"groups": 1,
"activation": "linear",
"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,
null,
null,
512
]
}
},
{
"module": "keras.layers",
"class_name": "UpSampling2D",
"config": {
"name": "up_sampling2d_3",
"trainable": true,
"dtype": {
"module": "keras",
"class_name": "DTypePolicy",
"config": {
"name": "float32"
},
"registered_name": null
},
"size": [
16,
16
],
"data_format": "channels_last",
"interpolation": "bilinear"
},
"registered_name": null,
"build_config": {
"input_shape": [
null,
null,
null,
1
]
}
}
],
"build_input_shape": [
null,
null,
null,
512
]
},
"registered_name": null,
"build_config": {
"input_shape": [
null,
null,
null,
512
]
}
},
{
"module": "keras",
"class_name": "Sequential",
"config": {
"name": "sequential_5",
"trainable": true,
"dtype": {
"module": "keras",
"class_name": "DTypePolicy",
"config": {
"name": "float32"
},
"registered_name": null
},
"layers": [
{
"module": "keras.layers",
"class_name": "InputLayer",
"config": {
"batch_shape": [
null,
null,
null,
512
],
"dtype": "float32",
"sparse": false,
"name": "input_layer_7"
},
"registered_name": null
},
{
"module": "keras.layers",
"class_name": "Conv2D",
"config": {
"name": "conv2d_5",
"trainable": true,
"dtype": {
"module": "keras",
"class_name": "DTypePolicy",
"config": {
"name": "float32"
},
"registered_name": null
},
"filters": 1,
"kernel_size": [
3,
3
],
"strides": [
1,
1
],
"padding": "same",
"data_format": "channels_last",
"dilation_rate": [
1,
1
],
"groups": 1,
"activation": "linear",
"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,
null,
null,
512
]
}
},
{
"module": "keras.layers",
"class_name": "UpSampling2D",
"config": {
"name": "up_sampling2d_4",
"trainable": true,
"dtype": {
"module": "keras",
"class_name": "DTypePolicy",
"config": {
"name": "float32"
},
"registered_name": null
},
"size": [
32,
32
],
"data_format": "channels_last",
"interpolation": "bilinear"
},
"registered_name": null,
"build_config": {
"input_shape": [
null,
null,
null,
1
]
}
}
],
"build_input_shape": [
null,
null,
null,
512
]
},
"registered_name": null,
"build_config": {
"input_shape": [
null,
null,
null,
512
]
}
},
{
"module": "keras",
"class_name": "Sequential",
"config": {
"name": "sequential_6",
"trainable": true,
"dtype": {
"module": "keras",
"class_name": "DTypePolicy",
"config": {
"name": "float32"
},
"registered_name": null
},
"layers": [
{
"module": "keras.layers",
"class_name": "InputLayer",
"config": {
"batch_shape": [
null,
null,
null,
512
],
"dtype": "float32",
"sparse": false,
"name": "input_layer_8"
},
"registered_name": null
},
{
"module": "keras.layers",
"class_name": "Conv2D",
"config": {
"name": "conv2d_6",
"trainable": true,
"dtype": {
"module": "keras",
"class_name": "DTypePolicy",
"config": {
"name": "float32"
},
"registered_name": null
},
"filters": 1,
"kernel_size": [
3,
3
],
"strides": [
1,
1
],
"padding": "same",
"data_format": "channels_last",
"dilation_rate": [
1,
1
],
"groups": 1,
"activation": "linear",
"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,
null,
null,
512
]
}
},
{
"module": "keras.layers",
"class_name": "UpSampling2D",
"config": {
"name": "up_sampling2d_5",
"trainable": true,
"dtype": {
"module": "keras",
"class_name": "DTypePolicy",
"config": {
"name": "float32"
},
"registered_name": null
},
"size": [
32,
32
],
"data_format": "channels_last",
"interpolation": "bilinear"
},
"registered_name": null,
"build_config": {
"input_shape": [
null,
null,
null,
1
]
}
}
],
"build_input_shape": [
null,
null,
null,
512
]
},
"registered_name": null,
"build_config": {
"input_shape": [
null,
null,
null,
512
]
}
}
],
"refinement_head": {
"module": "keras",
"class_name": "Sequential",
"config": {
"name": "sequential_7",
"trainable": true,
"dtype": {
"module": "keras",
"class_name": "DTypePolicy",
"config": {
"name": "float32"
},
"registered_name": null
},
"layers": [
{
"module": "keras.layers",
"class_name": "InputLayer",
"config": {
"batch_shape": [
null,
null,
null,
64
],
"dtype": "float32",
"sparse": false,
"name": "input_layer_9"
},
"registered_name": null
},
{
"module": "keras.layers",
"class_name": "Conv2D",
"config": {
"name": "conv2d_7",
"trainable": true,
"dtype": {
"module": "keras",
"class_name": "DTypePolicy",
"config": {
"name": "float32"
},
"registered_name": null
},
"filters": 1,
"kernel_size": [
3,
3
],
"strides": [
1,
1
],
"padding": "same",
"data_format": "channels_last",
"dilation_rate": [
1,
1
],
"groups": 1,
"activation": "linear",
"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,
null,
null,
64
]
}
}
],
"build_input_shape": [
null,
null,
null,
64
]
},
"registered_name": null
}
},
"registered_name": "keras_hub>BASNetBackbone"
},
"preprocessor": {
"module": "keras_hub.src.models.basnet.basnet_preprocessor",
"class_name": "BASNetPreprocessor",
"config": {
"name": "bas_net_preprocessor",
"trainable": true,
"dtype": {
"module": "keras",
"class_name": "DTypePolicy",
"config": {
"name": "float32"
},
"registered_name": null
},
"config_file": "preprocessor.json"
},
"registered_name": "keras_hub>BASNetPreprocessor"
},
"name": "bas_net_image_segmenter"
},
"registered_name": "keras_hub>BASNetImageSegmenter"
}