| { | |
| "module": "keras_hub.src.models.cspnet.cspnet_image_classifier", | |
| "class_name": "CSPNetImageClassifier", | |
| "config": { | |
| "backbone": { | |
| "module": "keras_hub.src.models.cspnet.cspnet_backbone", | |
| "class_name": "CSPNetBackbone", | |
| "config": { | |
| "name": "csp_net_backbone", | |
| "trainable": true, | |
| "stem_filters": 32, | |
| "stem_kernel_size": 3, | |
| "stem_strides": 1, | |
| "stackwise_depth": [ | |
| 1, | |
| 2, | |
| 8, | |
| 8, | |
| 4 | |
| ], | |
| "stackwise_strides": 2, | |
| "stackwise_num_filters": [ | |
| 64, | |
| 128, | |
| 256, | |
| 512, | |
| 1024 | |
| ], | |
| "stage_type": "dark", | |
| "block_type": "dark_block", | |
| "output_strides": 32, | |
| "groups": 1, | |
| "activation": "leaky_relu", | |
| "bottle_ratio": 0.5, | |
| "block_ratio": 1.0, | |
| "expand_ratio": null, | |
| "stem_padding": "same", | |
| "stem_pooling": null, | |
| "avg_down": null, | |
| "down_growth": null, | |
| "cross_linear": null, | |
| "image_shape": [ | |
| null, | |
| null, | |
| 3 | |
| ], | |
| "data_format": "channels_last" | |
| }, | |
| "registered_name": "keras_hub>CSPNetBackbone" | |
| }, | |
| "preprocessor": { | |
| "module": "keras_hub.src.models.cspnet.cspnet_image_classifier_preprocessor", | |
| "class_name": "CSPNetImageClassifierPreprocessor", | |
| "config": { | |
| "name": "csp_net_image_classifier_preprocessor", | |
| "trainable": true, | |
| "dtype": { | |
| "module": "keras", | |
| "class_name": "DTypePolicy", | |
| "config": { | |
| "name": "float32" | |
| }, | |
| "registered_name": null | |
| }, | |
| "image_converter": { | |
| "module": "keras_hub.src.models.cspnet.cspnet_image_converter", | |
| "class_name": "CSPNetImageConverter", | |
| "config": { | |
| "name": "csp_net_image_converter", | |
| "trainable": true, | |
| "dtype": { | |
| "module": "keras", | |
| "class_name": "DTypePolicy", | |
| "config": { | |
| "name": "float32" | |
| }, | |
| "registered_name": null | |
| }, | |
| "image_size": [ | |
| 256, | |
| 256 | |
| ], | |
| "scale": [ | |
| 0.017124753831663668, | |
| 0.01750700280112045, | |
| 0.017429193899782133 | |
| ], | |
| "offset": [ | |
| -2.1179039301310043, | |
| -2.0357142857142856, | |
| -1.8044444444444445 | |
| ], | |
| "interpolation": "bicubic", | |
| "antialias": false, | |
| "crop_to_aspect_ratio": true, | |
| "pad_to_aspect_ratio": false, | |
| "bounding_box_format": "yxyx" | |
| }, | |
| "registered_name": "keras_hub>CSPNetImageConverter" | |
| }, | |
| "config_file": "preprocessor.json" | |
| }, | |
| "registered_name": "keras_hub>CSPNetImageClassifierPreprocessor" | |
| }, | |
| "name": "csp_net_image_classifier", | |
| "num_classes": 1000, | |
| "pooling": "avg", | |
| "activation": "softmax", | |
| "dropout": 0.0 | |
| }, | |
| "registered_name": "keras_hub>CSPNetImageClassifier" | |
| } |