brainscore-vision 2.2.4__py3-none-any.whl → 2.2.6__py3-none-any.whl

brainscore_vision/models/timm_models/model.py

@@ -0,0 +1,90 @@
+import os
+import functools
+import json
+from pathlib import Path
+import ssl
+
+import timm
+import numpy as np
+import torchvision.transforms as T
+from PIL import Image
+
+from brainscore_vision.model_helpers.activations.pytorch import PytorchWrapper
+
+# Disable SSL verification 
+ssl._create_default_https_context = ssl._create_unverified_context
+
+BIBTEX = """"""
+
+
+with open(Path(__file__).parent / "model_configs.json", "r") as f:
+    MODEL_CONFIGS = json.load(f)
+
+
+def load_image(image_filepath):
+    return Image.open(image_filepath).convert("RGB")
+
+
+def get_interpolation_mode(interpolation: str) -> int:
+    """Returns the interpolation mode for albumentations"""
+    if "linear" or "bilinear" in interpolation:
+        return 1
+    elif "cubic" or "bicubic" in interpolation:
+        return 2
+    else:
+        raise NotImplementedError(f"Interpolation mode {interpolation} not implemented")
+
+
+def custom_image_preprocess(
+    images,
+    transforms=T.Compose,
+):
+    if isinstance(transforms, T.Compose):
+        images = [transforms(image) for image in images]
+        images = [np.array(image) for image in images]
+        images = np.stack(images)
+    else:
+        raise NotImplementedError(
+            f"Transform of type {type(transforms)} is not implemented"
+        )
+
+    return images
+
+
+def load_preprocess_images_custom(
+    image_filepaths, preprocess_images=custom_image_preprocess, **kwargs
+):
+    images = [load_image(image_filepath) for image_filepath in image_filepaths]
+    images = preprocess_images(images, **kwargs)
+    return images
+
+
+def get_model(model_id:str):
+    # Unpack model config
+    config = MODEL_CONFIGS[model_id]
+    model_name = config["model_name"]
+    model_id = config["model_id"]
+    timm_model_name = config["timm_model_name"]
+    is_vit = config["is_vit"]
+    
+    # Temporary fix for vit models
+    # See https://github.com/brain-score/vision/pull/1232
+    if is_vit:
+        os.environ['RESULTCACHING_DISABLE'] = 'brainscore_vision.model_helpers.activations.core.ActivationsExtractorHelper._from_paths_stored'
+
+    
+    # Initialize model
+    model = timm.create_model(timm_model_name, pretrained=True)
+    data_config = timm.data.resolve_model_data_config(model)
+    transforms = timm.data.create_transform(**data_config, is_training=False)
+    print(f"Model {model_name} loaded")
+
+    # Wrap model
+    preprocessing = functools.partial(
+        load_preprocess_images_custom,
+        transforms=transforms
+    )
+    wrapper = PytorchWrapper(
+        identifier=model_id, model=model, preprocessing=preprocessing
+    )
+    return wrapper

brainscore_vision/models/timm_models/model_configs.json

@@ -0,0 +1,464 @@
+{
+  "convnext_base:clip_laiona_augreg_ft_in1k_384": {
+    "model_name": "convnext_base",
+    "model_id": "convnext_base:clip_laiona_augreg_ft_in1k_384",
+    "timm_model_name": "convnext_base.clip_laiona_augreg_ft_in1k_384",
+    "is_vit": false,
+    "model_commitment": {
+      "region_layer_map": {
+        "IT": "stages.2.blocks.12.conv_dw",
+        "V1": "stages.2.blocks.1",
+        "V2": "stages.2.blocks.13.conv_dw",
+        "V4": "stages.2.blocks.1"
+      },
+      "layers": [
+        "stages.2.blocks.12.conv_dw",
+        "stages.2.blocks.1",
+        "stages.2.blocks.13.conv_dw",
+        "stages.2.blocks.1",
+        "head.pre_logits"
+      ],
+      "behavioral_readout_layer": "head.pre_logits"
+    }
+  },
+  "convnext_femto_ols:d1_in1k": {
+    "model_name": "convnext_femto_ols",
+    "model_id": "convnext_femto_ols:d1_in1k",
+    "timm_model_name": "convnext_femto_ols.d1_in1k",
+    "is_vit": false,
+    "model_commitment": {
+      "region_layer_map": {
+        "IT": "stages.3.blocks.0.norm",
+        "V1": "stages.2.blocks.0.mlp.fc1",
+        "V2": "stages.2.blocks.1.conv_dw",
+        "V4": "stages.2.downsample.1"
+      },
+      "layers": [
+        "stages.3.blocks.0.norm",
+        "stages.2.blocks.0.mlp.fc1",
+        "stages.2.blocks.1.conv_dw",
+        "stages.2.downsample.1",
+        "head.pre_logits"
+      ],
+      "behavioral_readout_layer": "head.pre_logits"
+    }
+  },
+  "convnext_large:fb_in22k_ft_in1k": {
+    "model_name": "convnext_large",
+    "model_id": "convnext_large:fb_in22k_ft_in1k",
+    "timm_model_name": "convnext_large.fb_in22k_ft_in1k",
+    "is_vit": false,
+    "model_commitment": {
+      "region_layer_map": {
+        "IT": "stages.2.blocks.11.conv_dw",
+        "V1": "stages.2.blocks.2.mlp.fc1",
+        "V2": "stages.2.blocks.6.norm",
+        "V4": "stages.2.blocks.0.mlp.fc1"
+      },
+      "layers": [
+        "stages.2.blocks.11.conv_dw",
+        "stages.2.blocks.2.mlp.fc1",
+        "stages.2.blocks.6.norm",
+        "stages.2.blocks.0.mlp.fc1",
+        "head.pre_logits"
+      ],
+      "behavioral_readout_layer": "head.pre_logits"
+    }
+  },
+  "convnext_large_mlp:clip_laion2b_augreg_ft_in1k_384": {
+    "model_name": "convnext_large_mlp",
+    "model_id": "convnext_large_mlp:clip_laion2b_augreg_ft_in1k_384",
+    "timm_model_name": "convnext_large_mlp.clip_laion2b_augreg_ft_in1k_384",
+    "is_vit": false,
+    "model_commitment": {
+      "region_layer_map": {
+        "IT": "stages.2.blocks.4.conv_dw",
+        "V1": "stages.2.blocks.3",
+        "V2": "stages.2.blocks.4.conv_dw",
+        "V4": "stages.2.downsample.1"
+      },
+      "layers": [
+        "stages.2.blocks.4.conv_dw",
+        "stages.2.blocks.3",
+        "stages.2.blocks.4.conv_dw",
+        "stages.2.downsample.1",
+        "head.pre_logits"
+      ],
+      "behavioral_readout_layer": "head.pre_logits"
+    }
+  },
+  "convnext_tiny:in12k_ft_in1k": {
+    "model_name": "convnext_tiny",
+    "model_id": "convnext_tiny:in12k_ft_in1k",
+    "timm_model_name": "convnext_tiny.in12k_ft_in1k",
+    "is_vit": false,
+    "model_commitment": {
+      "region_layer_map": {
+        "IT": "stages.2.blocks.3.conv_dw",
+        "V1": "stages.3.downsample.0",
+        "V2": "stages.2.blocks.5.conv_dw",
+        "V4": "stages.2.downsample.1"
+      },
+      "layers": [
+        "stages.2.blocks.3.conv_dw",
+        "stages.3.downsample.0",
+        "stages.2.blocks.5.conv_dw",
+        "stages.2.downsample.1",
+        "head.pre_logits"
+      ],
+      "behavioral_readout_layer": "head.pre_logits"
+    }
+  },
+  "convnext_xlarge:fb_in22k_ft_in1k": {
+    "model_name": "convnext_xlarge",
+    "model_id": "convnext_xlarge:fb_in22k_ft_in1k",
+    "timm_model_name": "convnext_xlarge.fb_in22k_ft_in1k",
+    "is_vit": false,
+    "model_commitment": {
+      "region_layer_map": {
+        "IT": "stages.2.blocks.1.conv_dw",
+        "V1": "stages.2.blocks.2.mlp.act",
+        "V2": "stages.2.blocks.15.norm",
+        "V4": "stages.2.blocks.0.conv_dw"
+      },
+      "layers": [
+        "stages.2.blocks.1.conv_dw",
+        "stages.2.blocks.2.mlp.act",
+        "stages.2.blocks.15.norm",
+        "stages.2.blocks.0.conv_dw",
+        "head.pre_logits"
+      ],
+      "behavioral_readout_layer": "head.pre_logits"
+    }
+  },
+  "convnext_xxlarge:clip_laion2b_soup_ft_in1k": {
+    "model_name": "convnext_xxlarge",
+    "model_id": "convnext_xxlarge:clip_laion2b_soup_ft_in1k",
+    "timm_model_name": "convnext_xxlarge.clip_laion2b_soup_ft_in1k",
+    "is_vit": false,
+    "model_commitment": {
+      "region_layer_map": {
+        "IT": "stages.2.blocks.2.conv_dw",
+        "V1": "stages.2.blocks.1.mlp.fc2",
+        "V2": "stages.2.blocks.2",
+        "V4": "stages.2.blocks.0.mlp.fc1"
+      },
+      "layers": [
+        "stages.2.blocks.2.conv_dw",
+        "stages.2.blocks.1.mlp.fc2",
+        "stages.2.blocks.2",
+        "stages.2.blocks.0.mlp.fc1",
+        "head.pre_logits"
+      ],
+      "behavioral_readout_layer": "head.pre_logits"
+    }
+  },
+  "swin_small_patch4_window7_224:ms_in22k_ft_in1k": {
+    "model_name": "swin_small_patch4_window7_224",
+    "model_id": "swin_small_patch4_window7_224:ms_in22k_ft_in1k",
+    "timm_model_name": "swin_small_patch4_window7_224.ms_in22k_ft_in1k",
+    "is_vit": true,
+    "model_commitment": {
+      "region_layer_map": {
+        "IT": "layers.3.blocks.0.norm2",
+        "V1": "layers.2.downsample.reduction",
+        "V2": "layers.2.blocks.6",
+        "V4": "layers.2.downsample.reduction"
+      },
+      "layers": [
+        "layers.3.blocks.0.norm2",
+        "layers.2.downsample.reduction",
+        "layers.2.blocks.6",
+        "layers.2.downsample.reduction",
+        "head.global_pool"
+      ],
+      "behavioral_readout_layer": "head.global_pool"
+    }
+  },
+  "vit_base_patch16_clip_224:openai_ft_in12k_in1k": {
+    "model_name": "vit_base_patch16_clip_224",
+    "model_id": "vit_base_patch16_clip_224:openai_ft_in12k_in1k",
+    "timm_model_name": "vit_base_patch16_clip_224.openai_ft_in12k_in1k",
+    "is_vit": true,
+    "model_commitment": {
+      "region_layer_map": {
+        "IT": "blocks.6.norm2",
+        "V1": "blocks.2.attn",
+        "V2": "blocks.2.attn",
+        "V4": "blocks.2.attn"
+      },
+      "layers": [
+        "blocks.6.norm2",
+        "blocks.2.attn",
+        "blocks.2.attn",
+        "blocks.2.attn",
+        "fc_norm"
+      ],
+      "behavioral_readout_layer": "fc_norm"
+    }
+  },
+  "vit_base_patch16_clip_224:openai_ft_in1k": {
+    "model_name": "vit_base_patch16_clip_224",
+    "model_id": "vit_base_patch16_clip_224:openai_ft_in1k",
+    "timm_model_name": "vit_base_patch16_clip_224.openai_ft_in1k",
+    "is_vit": true,
+    "model_commitment": {
+      "region_layer_map": {
+        "IT": "blocks.6.attn",
+        "V1": "blocks.2.attn",
+        "V2": "blocks.8.norm2",
+        "V4": "blocks.3.attn.qkv"
+      },
+      "layers": [
+        "blocks.6.attn",
+        "blocks.2.attn",
+        "blocks.8.norm2",
+        "blocks.3.attn.qkv",
+        "fc_norm"
+      ],
+      "behavioral_readout_layer": "fc_norm"
+    }
+  },
+  "vit_huge_patch14_clip_224:laion2b_ft_in12k_in1k": {
+    "model_name": "vit_huge_patch14_clip_224",
+    "model_id": "vit_huge_patch14_clip_224:laion2b_ft_in12k_in1k",
+    "timm_model_name": "vit_huge_patch14_clip_224.laion2b_ft_in12k_in1k",
+    "is_vit": true,
+    "model_commitment": {
+      "region_layer_map": {
+        "IT": "blocks.11.attn",
+        "V1": "blocks.7.attn.qkv",
+        "V2": "blocks.7.attn",
+        "V4": "blocks.6.attn.qkv"
+      },
+      "layers": [
+        "blocks.11.attn",
+        "blocks.7.attn.qkv",
+        "blocks.7.attn",
+        "blocks.6.attn.qkv",
+        "fc_norm"
+      ],
+      "behavioral_readout_layer": "fc_norm"
+    }
+  },
+  "vit_huge_patch14_clip_336:laion2b_ft_in12k_in1k": {
+    "model_name": "vit_huge_patch14_clip_336",
+    "model_id": "vit_huge_patch14_clip_336:laion2b_ft_in12k_in1k",
+    "timm_model_name": "vit_huge_patch14_clip_336.laion2b_ft_in12k_in1k",
+    "is_vit": true,
+    "model_commitment": {
+      "region_layer_map": {
+        "IT": "blocks.11.attn",
+        "V1": "blocks.6.attn",
+        "V2": "blocks.6.attn",
+        "V4": "blocks.7.attn.qkv"
+      },
+      "layers": [
+        "blocks.11.attn",
+        "blocks.6.attn",
+        "blocks.6.attn",
+        "blocks.7.attn.qkv",
+        "fc_norm"
+      ],
+      "behavioral_readout_layer": "fc_norm"
+    }
+  },
+  "vit_large_patch14_clip_224:laion2b_ft_in12k_in1k": {
+    "model_name": "vit_large_patch14_clip_224",
+    "model_id": "vit_large_patch14_clip_224:laion2b_ft_in12k_in1k",
+    "timm_model_name": "vit_large_patch14_clip_224.laion2b_ft_in12k_in1k",
+    "is_vit": true,
+    "model_commitment": {
+      "region_layer_map": {
+        "IT": "blocks.13.norm2",
+        "V1": "blocks.4.attn.qkv",
+        "V2": "blocks.8.attn",
+        "V4": "blocks.5.attn.qkv"
+      },
+      "layers": [
+        "blocks.13.norm2",
+        "blocks.4.attn.qkv",
+        "blocks.8.attn",
+        "blocks.5.attn.qkv",
+        "fc_norm"
+      ],
+      "behavioral_readout_layer": "fc_norm"
+    }
+  },
+  "vit_large_patch14_clip_224:laion2b_ft_in1k": {
+    "model_name": "vit_large_patch14_clip_224",
+    "model_id": "vit_large_patch14_clip_224:laion2b_ft_in1k",
+    "timm_model_name": "vit_large_patch14_clip_224.laion2b_ft_in1k",
+    "is_vit": true,
+    "model_commitment": {
+      "region_layer_map": {
+        "IT": "blocks.9.attn",
+        "V1": "blocks.4.attn.qkv",
+        "V2": "blocks.8.attn",
+        "V4": "blocks.5.attn"
+      },
+      "layers": [
+        "blocks.9.attn",
+        "blocks.4.attn.qkv",
+        "blocks.8.attn",
+        "blocks.5.attn",
+        "fc_norm"
+      ],
+      "behavioral_readout_layer": "fc_norm"
+    }
+  },
+  "vit_large_patch14_clip_224:openai_ft_in12k_in1k": {
+    "model_name": "vit_large_patch14_clip_224",
+    "model_id": "vit_large_patch14_clip_224:openai_ft_in12k_in1k",
+    "timm_model_name": "vit_large_patch14_clip_224.openai_ft_in12k_in1k",
+    "is_vit": true,
+    "model_commitment": {
+      "region_layer_map": {
+        "IT": "blocks.10.attn",
+        "V1": "blocks.5",
+        "V2": "blocks.7.attn",
+        "V4": "blocks.6.attn.qkv"
+      },
+      "layers": [
+        "blocks.10.attn",
+        "blocks.5",
+        "blocks.7.attn",
+        "blocks.6.attn.qkv",
+        "fc_norm"
+      ],
+      "behavioral_readout_layer": "fc_norm"
+    }
+  },
+  "vit_large_patch14_clip_224:openai_ft_in1k": {
+    "model_name": "vit_large_patch14_clip_224",
+    "model_id": "vit_large_patch14_clip_224:openai_ft_in1k",
+    "timm_model_name": "vit_large_patch14_clip_224.openai_ft_in1k",
+    "is_vit": true,
+    "model_commitment": {
+      "region_layer_map": {
+        "IT": "blocks.10.attn",
+        "V1": "blocks.6",
+        "V2": "blocks.7.attn",
+        "V4": "blocks.6.attn"
+      },
+      "layers": [
+        "blocks.10.attn",
+        "blocks.6",
+        "blocks.7.attn",
+        "blocks.6.attn",
+        "fc_norm"
+      ],
+      "behavioral_readout_layer": "fc_norm"
+    }
+  },
+  "vit_large_patch14_clip_336:laion2b_ft_in1k": {
+    "model_name": "vit_large_patch14_clip_336",
+    "model_id": "vit_large_patch14_clip_336:laion2b_ft_in1k",
+    "timm_model_name": "vit_large_patch14_clip_336.laion2b_ft_in1k",
+    "is_vit": true,
+    "model_commitment": {
+      "region_layer_map": {
+        "IT": "blocks.11.attn.proj",
+        "V1": "blocks.6.attn",
+        "V2": "blocks.6.attn.proj",
+        "V4": "blocks.6.attn"
+      },
+      "layers": [
+        "blocks.11.attn.proj",
+        "blocks.6.attn",
+        "blocks.6.attn.proj",
+        "blocks.6.attn",
+        "fc_norm"
+      ],
+      "behavioral_readout_layer": "fc_norm"
+    }
+  },
+  "vit_large_patch14_clip_336:openai_ft_in12k_in1k": {
+    "model_name": "vit_large_patch14_clip_336",
+    "model_id": "vit_large_patch14_clip_336:openai_ft_in12k_in1k",
+    "timm_model_name": "vit_large_patch14_clip_336.openai_ft_in12k_in1k",
+    "is_vit": true,
+    "model_commitment": {
+      "region_layer_map": {
+        "IT": "blocks.10.attn.proj",
+        "V1": "blocks.4.attn",
+        "V2": "blocks.7.attn",
+        "V4": "blocks.6.attn"
+      },
+      "layers": [
+        "blocks.10.attn.proj",
+        "blocks.4.attn",
+        "blocks.7.attn",
+        "blocks.6.attn",
+        "fc_norm"
+      ],
+      "behavioral_readout_layer": "fc_norm"
+    }
+  },
+  "vit_relpos_base_patch16_clsgap_224:sw_in1k": {
+    "model_name": "vit_relpos_base_patch16_clsgap_224",
+    "model_id": "vit_relpos_base_patch16_clsgap_224:sw_in1k",
+    "timm_model_name": "vit_relpos_base_patch16_clsgap_224.sw_in1k",
+    "is_vit": true,
+    "model_commitment": {
+      "region_layer_map": {
+        "IT": "blocks.8.norm2",
+        "V1": "blocks.6.norm2",
+        "V2": "blocks.7.norm2",
+        "V4": "blocks.4.norm2"
+      },
+      "layers": [
+        "blocks.8.norm2",
+        "blocks.6.norm2",
+        "blocks.7.norm2",
+        "blocks.4.norm2",
+        "fc_norm"
+      ],
+      "behavioral_readout_layer": "fc_norm"
+    }
+  },
+  "vit_relpos_base_patch32_plus_rpn_256:sw_in1k": {
+    "model_name": "vit_relpos_base_patch32_plus_rpn_256",
+    "model_id": "vit_relpos_base_patch32_plus_rpn_256:sw_in1k",
+    "timm_model_name": "vit_relpos_base_patch32_plus_rpn_256.sw_in1k",
+    "is_vit": true,
+    "model_commitment": {
+      "region_layer_map": {
+        "IT": "blocks.8.norm1",
+        "V1": "blocks.3.norm1",
+        "V2": "blocks.6.norm1",
+        "V4": "blocks.3.norm1"
+      },
+      "layers": [
+        "blocks.8.norm1",
+        "blocks.3.norm1",
+        "blocks.6.norm1",
+        "blocks.3.norm1",
+        "fc_norm"
+      ],
+      "behavioral_readout_layer": "fc_norm"
+    }
+  },
+  "vit_tiny_r_s16_p8_384:augreg_in21k_ft_in1k": {
+    "model_name": "vit_tiny_r_s16_p8_384",
+    "model_id": "vit_tiny_r_s16_p8_384:augreg_in21k_ft_in1k",
+    "timm_model_name": "vit_tiny_r_s16_p8_384.augreg_in21k_ft_in1k",
+    "is_vit": true,
+    "model_commitment": {
+      "region_layer_map": {
+        "IT": "blocks.5",
+        "V1": "blocks.1",
+        "V2": "blocks.3",
+        "V4": "blocks.1"
+      },
+      "layers": [
+        "blocks.5",
+        "blocks.1",
+        "blocks.3",
+        "blocks.1",
+        "fc_norm"
+      ],
+      "behavioral_readout_layer": "fc_norm"
+    }
+  }
+}

brainscore_vision/models/timm_models/requirements.txt

@@ -0,0 +1,3 @@
+torch
+torchvision
+timm

brainscore_vision/models/vgg_16/__init__.py

@@ -0,0 +1,7 @@
+from brainscore_vision.model_helpers.brain_transformation import ModelCommitment
+from brainscore_vision import model_registry
+from .model import get_layers,get_model
+
+
+model_registry['vgg_16'] = \
+    lambda: ModelCommitment(identifier='vgg_16', activations_model=get_model('vgg_16'), layers=get_layers('vgg_16'))

brainscore_vision/models/vgg_16/model.py

@@ -0,0 +1,52 @@
+import functools
+from brainscore_vision.model_helpers.activations.pytorch import load_preprocess_images
+from brainscore_vision.model_helpers.activations.pytorch import PytorchWrapper
+from brainscore_vision.model_helpers.check_submission import check_models
+import torchvision
+
+
+
+def get_model(name):
+    """
+    This method fetches an instance of a base model. The instance has to be callable and return a xarray object,
+    containing activations. There exist standard wrapper implementations for common libraries, like pytorch and
+    keras. Checkout the examples folder, to see more. For custom implementations check out the implementation of the
+    wrappers.
+    :param name: the name of the model to fetch
+    :return: the model instance
+    """
+    assert name == 'vgg_16'
+    model = torchvision.models.vgg16(weights='DEFAULT')
+    preprocessing = functools.partial(load_preprocess_images, image_size=224)
+    wrapper = PytorchWrapper(identifier=name, model=model, preprocessing=preprocessing)
+    wrapper.image_size = 224
+    return wrapper
+
+
+def get_layers(name):
+    assert name == 'vgg_16'
+    layer_names = [f'features.{i}' for i in [4, 9, 16, 23, 30]] + [f'classifier.{i}' for i in [0, 3]]
+    return layer_names
+
+
+def get_bibtex(name):
+    """
+    A method returning the bibtex reference of the requested model as a string.
+    """
+    return '''
+@misc{simonyan2015deepconvolutionalnetworkslargescale,
+      title={Very Deep Convolutional Networks for Large-Scale Image Recognition}, 
+      author={Karen Simonyan and Andrew Zisserman},
+      year={2015},
+      eprint={1409.1556},
+      archivePrefix={arXiv},
+      primaryClass={cs.CV},
+      url={https://arxiv.org/abs/1409.1556}, 
+}
+'''
+
+
+if __name__ == '__main__':
+    # Use this method to ensure the correctness of the BaseModel implementations.
+    # It executes a mock run of brain-score benchmarks.
+    check_models.check_base_models(__name__)

brainscore_vision/models/vgg_16/region_layer_map/vgg_16.json

@@ -0,0 +1,6 @@
+{
+    "V1": "features.16",
+    "V2": "features.16",
+    "V4": "features.16",
+    "IT": "features.23"
+}

brainscore_vision/models/vgg_16/requirements.txt

@@ -0,0 +1 @@
+torchvision

brainscore_vision/models/vgg_16/test.py

@@ -0,0 +1,8 @@
+import brainscore_vision
+import pytest
+
+
+@pytest.mark.travis_slow
+def test_has_identifier():
+    model = brainscore_vision.load_model('vgg_16')
+    assert model.identifier == 'vgg_16'

brainscore_vision/models/vgg_19/__init__.py

@@ -0,0 +1,7 @@
+from brainscore_vision.model_helpers.brain_transformation import ModelCommitment
+from brainscore_vision import model_registry
+from .model import get_layers,get_model
+
+
+model_registry['vgg_19'] = \
+    lambda: ModelCommitment(identifier='vgg_19', activations_model=get_model('vgg_19'), layers=get_layers('vgg_19'))