{
    "model_type": "keras",
    "framework": "tensorflow",
    "format": "h5",
    "task": "image-segmentation",
    "architecture": "unet",
    "input_shape": [128, 128, 3],
    "loss": {
        "name": "SparseCategoricalCrossentropy",
        "from_logits": true
    },
    "preprocessing": {
        "image_resize": [128, 128],
        "mask_resize": [128, 128],
        "mask_resize_method": "nearest",
        "image_normalization": "x/255.0",
        "mask_label_offset": -1
    },
    "dataset": {
        "name": "oxford_iiit_pet",
        "version": "3.."
    },
    "num_classes": 3,
    "id2label": {
        "0": "background",
        "1": "pet",
        "2": "outline"
    },
    "label2id": {
        "background": 0,
        "pet": 1,
        "outline": 2
    },
    "model_file": "segmentation_pets.h5"
}