diff --git a/brainscore_vision/models/resnet50_11ad3fa6/__init__.py b/brainscore_vision/models/resnet50_11ad3fa6/__init__.py new file mode 100644 index 000000000..cdfd7fa38 --- /dev/null +++ b/brainscore_vision/models/resnet50_11ad3fa6/__init__.py @@ -0,0 +1,9 @@ +from brainscore_vision import model_registry +from brainscore_vision.model_helpers.brain_transformation import ModelCommitment +from .model import get_model, get_layers + +model_registry['resnet50_11ad3fa6'] = lambda: ModelCommitment( + identifier='resnet50_11ad3fa6', + activations_model=get_model('resnet50_11ad3fa6'), + layers=get_layers('resnet50_11ad3fa6') +) diff --git a/brainscore_vision/models/resnet50_11ad3fa6/model.py b/brainscore_vision/models/resnet50_11ad3fa6/model.py new file mode 100644 index 000000000..1b4294372 --- /dev/null +++ b/brainscore_vision/models/resnet50_11ad3fa6/model.py @@ -0,0 +1,35 @@ +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 + +# This is an example implementation for submitting resnet-50 as a pytorch model + +# Attention: It is important, that the wrapper identifier is unique per model! +# The results will otherwise be the same due to brain-scores internal result caching mechanism. +# Please load your pytorch model for usage in CPU. There won't be GPUs available for scoring your model. +# If the model requires a GPU, contact the brain-score team directly. + + +def get_model(name): + assert name == 'resnet50_11ad3fa6' + model = torchvision.models.resnet50(pretrained=True) + preprocessing = functools.partial(load_preprocess_images, image_size=224) + wrapper = PytorchWrapper( + identifier='resnet50_11ad3fa6', model=model, preprocessing=preprocessing) + wrapper.image_size = 224 + return wrapper + + +def get_layers(name): + assert name == 'resnet50_11ad3fa6' + return ['conv1', 'layer1', 'layer2', 'layer3', 'layer4', 'fc'] + + +def get_bibtex(model_identifier): + return """""" + + +if __name__ == '__main__': + check_models.check_base_models(__name__) diff --git a/brainscore_vision/models/resnet50_11ad3fa6/requirements.txt b/brainscore_vision/models/resnet50_11ad3fa6/requirements.txt new file mode 100644 index 000000000..a56666d38 --- /dev/null +++ b/brainscore_vision/models/resnet50_11ad3fa6/requirements.txt @@ -0,0 +1,2 @@ +torchvision +torch \ No newline at end of file diff --git a/brainscore_vision/models/resnet50_11ad3fa6/test.py b/brainscore_vision/models/resnet50_11ad3fa6/test.py new file mode 100644 index 000000000..28c13e09b --- /dev/null +++ b/brainscore_vision/models/resnet50_11ad3fa6/test.py @@ -0,0 +1,8 @@ +import pytest +import brainscore_vision + + +@pytest.mark.travis_slow +def test_has_identifier(): + model = brainscore_vision.load_model('resnet50_tutorial') + assert model.identifier == 'resnet50_tutorial'