add resnet50_4_all to models

brain-score · Dec 2, 2024 · 6dcf7a3 · 6dcf7a3
1 parent 9da6818
commit 6dcf7a3
Show file tree

Hide file tree

Showing 6 changed files with 2,920 additions and 0 deletions.
diff --git a/brainscore_vision/models/resnet50_4_all/__init__.py b/brainscore_vision/models/resnet50_4_all/__init__.py
@@ -0,0 +1,11 @@
+
+from brainscore_vision import model_registry
+from brainscore_vision.model_helpers.brain_transformation import ModelCommitment
+from .model import get_model, MODEL_CONFIGS, get_layers
+import os
+
+network = 'resnet50'
+for keyword in [ 'less_variation', 'no_variation', 'z_axis', 'wo_shading','wo_shadows', 'ambient', 'textures', 'no_specular']:
+    for iteration in range(1,6):
+        if f'{network}_{keyword}_iteration={iteration}' not in model_registry:
+            model_registry[f'{network}_{keyword}_iteration={iteration}'] = lambda: ModelCommitment(identifier=f'{network}_{keyword}_iteration={iteration}', activations_model=get_model(f'{network}_{keyword}_iteration={iteration}'), layers=get_layers(f'{network}_{keyword}_iteration={iteration}'))