add sam_test_resnet_4 to models (#1313)

Co-authored-by: Jenkins <[email protected]>

brainscore_vision/models/sam_test_resnet_4/__init__.py

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

brainscore_vision/models/sam_test_resnet_4/model.py

@@ -0,0 +1,26 @@
+from brainscore_vision.model_helpers.check_submission import check_models
+import functools
+import torchvision.models
+from brainscore_vision.model_helpers.activations.pytorch import PytorchWrapper
+from brainscore_vision.model_helpers.activations.pytorch import load_preprocess_images
+
+def get_model(name):
+    assert name == 'sam_test_resnet_4'
+    model = torchvision.models.resnet50(pretrained=True)
+    preprocessing = functools.partial(load_preprocess_images, image_size=224)
+    wrapper = PytorchWrapper(identifier='sam_test_resnet', model=model, preprocessing=preprocessing)
+    wrapper.image_size = 224
+    return wrapper
+
+
+def get_layers(name):
+    assert name == 'sam_test_resnet_4'
+    return ['conv1','layer1', 'layer2', 'layer3', 'layer4', 'fc']
+
+
+def get_bibtex(model_identifier):
+    return """"""
+
+
+if __name__ == '__main__':
+    check_models.check_base_models(__name__)

brainscore_vision/models/sam_test_resnet_4/requirements.txt

@@ -0,0 +1,2 @@
+torchvision
+torch

brainscore_vision/models/sam_test_resnet_4/test.py

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