brainscore-vision 2.1__py3-none-any.whl
Sign up to get free protection for your applications and to get access to all the features.
- brainscore_vision/__init__.py +105 -0
- brainscore_vision/__main__.py +20 -0
- brainscore_vision/benchmark_helpers/__init__.py +67 -0
- brainscore_vision/benchmark_helpers/neural_common.py +70 -0
- brainscore_vision/benchmark_helpers/properties_common.py +424 -0
- brainscore_vision/benchmark_helpers/screen.py +126 -0
- brainscore_vision/benchmark_helpers/test_helper.py +160 -0
- brainscore_vision/benchmarks/README.md +7 -0
- brainscore_vision/benchmarks/__init__.py +122 -0
- brainscore_vision/benchmarks/baker2022/__init__.py +9 -0
- brainscore_vision/benchmarks/baker2022/benchmark.py +125 -0
- brainscore_vision/benchmarks/baker2022/requirements.txt +1 -0
- brainscore_vision/benchmarks/baker2022/test.py +90 -0
- brainscore_vision/benchmarks/bmd2024/__init__.py +8 -0
- brainscore_vision/benchmarks/bmd2024/benchmark.py +51 -0
- brainscore_vision/benchmarks/bmd2024/test.py +29 -0
- brainscore_vision/benchmarks/bracci2019/__init__.py +8 -0
- brainscore_vision/benchmarks/bracci2019/benchmark.py +286 -0
- brainscore_vision/benchmarks/bracci2019/requirements.txt +3 -0
- brainscore_vision/benchmarks/cadena2017/__init__.py +5 -0
- brainscore_vision/benchmarks/cadena2017/benchmark.py +91 -0
- brainscore_vision/benchmarks/cadena2017/test.py +35 -0
- brainscore_vision/benchmarks/coggan2024_behavior/__init__.py +8 -0
- brainscore_vision/benchmarks/coggan2024_behavior/benchmark.py +133 -0
- brainscore_vision/benchmarks/coggan2024_behavior/test.py +21 -0
- brainscore_vision/benchmarks/coggan2024_fMRI/__init__.py +15 -0
- brainscore_vision/benchmarks/coggan2024_fMRI/benchmark.py +201 -0
- brainscore_vision/benchmarks/coggan2024_fMRI/test.py +25 -0
- brainscore_vision/benchmarks/ferguson2024/__init__.py +24 -0
- brainscore_vision/benchmarks/ferguson2024/benchmark.py +210 -0
- brainscore_vision/benchmarks/ferguson2024/helpers/helpers.py +251 -0
- brainscore_vision/benchmarks/ferguson2024/requirements.txt +5 -0
- brainscore_vision/benchmarks/ferguson2024/test.py +114 -0
- brainscore_vision/benchmarks/freemanziemba2013/__init__.py +10 -0
- brainscore_vision/benchmarks/freemanziemba2013/benchmarks/benchmark.py +53 -0
- brainscore_vision/benchmarks/freemanziemba2013/benchmarks/public_benchmarks.py +37 -0
- brainscore_vision/benchmarks/freemanziemba2013/test.py +98 -0
- brainscore_vision/benchmarks/geirhos2021/__init__.py +59 -0
- brainscore_vision/benchmarks/geirhos2021/benchmark.py +132 -0
- brainscore_vision/benchmarks/geirhos2021/test.py +189 -0
- brainscore_vision/benchmarks/hebart2023/__init__.py +4 -0
- brainscore_vision/benchmarks/hebart2023/benchmark.py +72 -0
- brainscore_vision/benchmarks/hebart2023/test.py +19 -0
- brainscore_vision/benchmarks/hermann2020/__init__.py +6 -0
- brainscore_vision/benchmarks/hermann2020/benchmark.py +63 -0
- brainscore_vision/benchmarks/hermann2020/test.py +28 -0
- brainscore_vision/benchmarks/igustibagus2024/__init__.py +11 -0
- brainscore_vision/benchmarks/igustibagus2024/domain_transfer_analysis.py +306 -0
- brainscore_vision/benchmarks/igustibagus2024/domain_transfer_neural.py +134 -0
- brainscore_vision/benchmarks/igustibagus2024/test.py +45 -0
- brainscore_vision/benchmarks/imagenet/__init__.py +4 -0
- brainscore_vision/benchmarks/imagenet/benchmark.py +50 -0
- brainscore_vision/benchmarks/imagenet/imagenet2012.csv +50001 -0
- brainscore_vision/benchmarks/imagenet/test.py +32 -0
- brainscore_vision/benchmarks/imagenet_c/__init__.py +7 -0
- brainscore_vision/benchmarks/imagenet_c/benchmark.py +204 -0
- brainscore_vision/benchmarks/imagenet_c/test.py +57 -0
- brainscore_vision/benchmarks/islam2021/__init__.py +11 -0
- brainscore_vision/benchmarks/islam2021/benchmark.py +107 -0
- brainscore_vision/benchmarks/islam2021/test.py +47 -0
- brainscore_vision/benchmarks/kar2019/__init__.py +4 -0
- brainscore_vision/benchmarks/kar2019/benchmark.py +88 -0
- brainscore_vision/benchmarks/kar2019/test.py +93 -0
- brainscore_vision/benchmarks/majajhong2015/__init__.py +18 -0
- brainscore_vision/benchmarks/majajhong2015/benchmark.py +96 -0
- brainscore_vision/benchmarks/majajhong2015/test.py +103 -0
- brainscore_vision/benchmarks/malania2007/__init__.py +13 -0
- brainscore_vision/benchmarks/malania2007/benchmark.py +235 -0
- brainscore_vision/benchmarks/malania2007/test.py +64 -0
- brainscore_vision/benchmarks/maniquet2024/__init__.py +6 -0
- brainscore_vision/benchmarks/maniquet2024/benchmark.py +199 -0
- brainscore_vision/benchmarks/maniquet2024/test.py +17 -0
- brainscore_vision/benchmarks/marques2020/__init__.py +76 -0
- brainscore_vision/benchmarks/marques2020/benchmarks/cavanaugh2002a_benchmark.py +119 -0
- brainscore_vision/benchmarks/marques2020/benchmarks/devalois1982a_benchmark.py +84 -0
- brainscore_vision/benchmarks/marques2020/benchmarks/devalois1982b_benchmark.py +88 -0
- brainscore_vision/benchmarks/marques2020/benchmarks/freemanZiemba2013_benchmark.py +138 -0
- brainscore_vision/benchmarks/marques2020/benchmarks/ringach2002_benchmark.py +167 -0
- brainscore_vision/benchmarks/marques2020/benchmarks/schiller1976_benchmark.py +100 -0
- brainscore_vision/benchmarks/marques2020/test.py +135 -0
- brainscore_vision/benchmarks/objectnet/__init__.py +4 -0
- brainscore_vision/benchmarks/objectnet/benchmark.py +52 -0
- brainscore_vision/benchmarks/objectnet/test.py +33 -0
- brainscore_vision/benchmarks/rajalingham2018/__init__.py +10 -0
- brainscore_vision/benchmarks/rajalingham2018/benchmarks/benchmark.py +74 -0
- brainscore_vision/benchmarks/rajalingham2018/benchmarks/public_benchmark.py +10 -0
- brainscore_vision/benchmarks/rajalingham2018/test.py +125 -0
- brainscore_vision/benchmarks/rajalingham2018/test_resources/alexnet-probabilities.nc +0 -0
- brainscore_vision/benchmarks/rajalingham2018/test_resources/identifier=alexnet,stimuli_identifier=objectome-240.nc +0 -0
- brainscore_vision/benchmarks/rajalingham2018/test_resources/identifier=resnet18,stimuli_identifier=objectome-240.nc +0 -0
- brainscore_vision/benchmarks/rajalingham2018/test_resources/identifier=resnet34,stimuli_identifier=objectome-240.nc +0 -0
- brainscore_vision/benchmarks/rajalingham2018/test_resources/resnet18-probabilities.nc +0 -0
- brainscore_vision/benchmarks/rajalingham2018/test_resources/resnet34-probabilities.nc +0 -0
- brainscore_vision/benchmarks/rajalingham2020/__init__.py +4 -0
- brainscore_vision/benchmarks/rajalingham2020/benchmark.py +52 -0
- brainscore_vision/benchmarks/rajalingham2020/test.py +39 -0
- brainscore_vision/benchmarks/sanghavi2020/__init__.py +17 -0
- brainscore_vision/benchmarks/sanghavi2020/benchmarks/sanghavi2020_benchmark.py +44 -0
- brainscore_vision/benchmarks/sanghavi2020/benchmarks/sanghavijozwik2020_benchmark.py +44 -0
- brainscore_vision/benchmarks/sanghavi2020/benchmarks/sanghavimurty2020_benchmark.py +44 -0
- brainscore_vision/benchmarks/sanghavi2020/test.py +83 -0
- brainscore_vision/benchmarks/scialom2024/__init__.py +52 -0
- brainscore_vision/benchmarks/scialom2024/benchmark.py +97 -0
- brainscore_vision/benchmarks/scialom2024/test.py +162 -0
- brainscore_vision/data/__init__.py +0 -0
- brainscore_vision/data/baker2022/__init__.py +40 -0
- brainscore_vision/data/baker2022/data_packaging/inverted_distortion_data_assembly.py +43 -0
- brainscore_vision/data/baker2022/data_packaging/inverted_distortion_stimulus_set.py +81 -0
- brainscore_vision/data/baker2022/data_packaging/mapping.py +60 -0
- brainscore_vision/data/baker2022/data_packaging/normal_distortion_data_assembly.py +46 -0
- brainscore_vision/data/baker2022/data_packaging/normal_distortion_stimulus_set.py +94 -0
- brainscore_vision/data/baker2022/test.py +135 -0
- brainscore_vision/data/barbumayo2019/BarbuMayo2019.py +26 -0
- brainscore_vision/data/barbumayo2019/__init__.py +23 -0
- brainscore_vision/data/barbumayo2019/test.py +10 -0
- brainscore_vision/data/bashivankar2019/__init__.py +52 -0
- brainscore_vision/data/bashivankar2019/data_packaging/2020-08-17_npc_v4_data.h5.png +0 -0
- brainscore_vision/data/bashivankar2019/data_packaging/requirements.txt +4 -0
- brainscore_vision/data/bashivankar2019/data_packaging/synthetic.py +162 -0
- brainscore_vision/data/bashivankar2019/test.py +15 -0
- brainscore_vision/data/bmd2024/__init__.py +69 -0
- brainscore_vision/data/bmd2024/data_packaging/BMD_2024_data_assembly.py +91 -0
- brainscore_vision/data/bmd2024/data_packaging/BMD_2024_simulus_set.py +48 -0
- brainscore_vision/data/bmd2024/data_packaging/stim_meta.csv +401 -0
- brainscore_vision/data/bmd2024/test.py +130 -0
- brainscore_vision/data/bracci2019/__init__.py +36 -0
- brainscore_vision/data/bracci2019/data_packaging.py +221 -0
- brainscore_vision/data/bracci2019/test.py +16 -0
- brainscore_vision/data/cadena2017/__init__.py +52 -0
- brainscore_vision/data/cadena2017/data_packaging/2018-08-07_tolias_v1.ipynb +25880 -0
- brainscore_vision/data/cadena2017/data_packaging/analysis.py +26 -0
- brainscore_vision/data/cadena2017/test.py +24 -0
- brainscore_vision/data/cichy2019/__init__.py +38 -0
- brainscore_vision/data/cichy2019/test.py +8 -0
- brainscore_vision/data/coggan2024_behavior/__init__.py +36 -0
- brainscore_vision/data/coggan2024_behavior/data_packaging.py +166 -0
- brainscore_vision/data/coggan2024_behavior/test.py +32 -0
- brainscore_vision/data/coggan2024_fMRI/__init__.py +27 -0
- brainscore_vision/data/coggan2024_fMRI/data_packaging.py +123 -0
- brainscore_vision/data/coggan2024_fMRI/test.py +25 -0
- brainscore_vision/data/david2004/__init__.py +34 -0
- brainscore_vision/data/david2004/data_packaging/2018-05-10_gallant_data.ipynb +3647 -0
- brainscore_vision/data/david2004/data_packaging/2018-05-23_gallant_data.ipynb +3149 -0
- brainscore_vision/data/david2004/data_packaging/2018-06-05_gallant_data.ipynb +3628 -0
- brainscore_vision/data/david2004/data_packaging/__init__.py +61 -0
- brainscore_vision/data/david2004/data_packaging/convertGallant.m +100 -0
- brainscore_vision/data/david2004/data_packaging/convertGallantV1Aligned.m +58 -0
- brainscore_vision/data/david2004/data_packaging/lib/DataHash_20160618/DataHash.m +484 -0
- brainscore_vision/data/david2004/data_packaging/lib/DataHash_20160618/license.txt +24 -0
- brainscore_vision/data/david2004/data_packaging/lib/GetMD5/GetMD5.c +895 -0
- brainscore_vision/data/david2004/data_packaging/lib/GetMD5/GetMD5.m +107 -0
- brainscore_vision/data/david2004/data_packaging/lib/GetMD5/GetMD5.mexw64 +0 -0
- brainscore_vision/data/david2004/data_packaging/lib/GetMD5/GetMD5_helper.m +91 -0
- brainscore_vision/data/david2004/data_packaging/lib/GetMD5/InstallMex.m +307 -0
- brainscore_vision/data/david2004/data_packaging/lib/GetMD5/license.txt +24 -0
- brainscore_vision/data/david2004/data_packaging/lib/GetMD5/uTest_GetMD5.m +290 -0
- brainscore_vision/data/david2004/data_packaging/lib/glob/glob.m +472 -0
- brainscore_vision/data/david2004/data_packaging/lib/glob/license.txt +27 -0
- brainscore_vision/data/david2004/data_packaging/xr_align_debug.py +137 -0
- brainscore_vision/data/david2004/test.py +8 -0
- brainscore_vision/data/deng2009/__init__.py +22 -0
- brainscore_vision/data/deng2009/deng2009imagenet.py +33 -0
- brainscore_vision/data/deng2009/test.py +9 -0
- brainscore_vision/data/ferguson2024/__init__.py +401 -0
- brainscore_vision/data/ferguson2024/data_packaging/data_packaging.py +164 -0
- brainscore_vision/data/ferguson2024/data_packaging/fitting_stimuli.py +20 -0
- brainscore_vision/data/ferguson2024/requirements.txt +2 -0
- brainscore_vision/data/ferguson2024/test.py +155 -0
- brainscore_vision/data/freemanziemba2013/__init__.py +133 -0
- brainscore_vision/data/freemanziemba2013/data_packaging/2018-10-05_movshon.ipynb +2002 -0
- brainscore_vision/data/freemanziemba2013/data_packaging/2020-02-21_movshon_aperture.ipynb +4730 -0
- brainscore_vision/data/freemanziemba2013/data_packaging/2020-02-26_movshon_aperture_test.ipynb +2228 -0
- brainscore_vision/data/freemanziemba2013/data_packaging/aperture_correct.py +160 -0
- brainscore_vision/data/freemanziemba2013/data_packaging/data_packaging.py +57 -0
- brainscore_vision/data/freemanziemba2013/data_packaging/movshon.py +202 -0
- brainscore_vision/data/freemanziemba2013/test.py +97 -0
- brainscore_vision/data/geirhos2021/__init__.py +358 -0
- brainscore_vision/data/geirhos2021/creating_geirhos_ids.ipynb +468 -0
- brainscore_vision/data/geirhos2021/data_packaging/colour/colour_data_assembly.py +87 -0
- brainscore_vision/data/geirhos2021/data_packaging/colour/colour_stimulus_set.py +81 -0
- brainscore_vision/data/geirhos2021/data_packaging/contrast/contrast_data_assembly.py +83 -0
- brainscore_vision/data/geirhos2021/data_packaging/contrast/contrast_stimulus_set.py +82 -0
- brainscore_vision/data/geirhos2021/data_packaging/cue-conflict/cue-conflict_data_assembly.py +100 -0
- brainscore_vision/data/geirhos2021/data_packaging/cue-conflict/cue-conflict_stimulus_set.py +84 -0
- brainscore_vision/data/geirhos2021/data_packaging/edge/edge_data_assembly.py +96 -0
- brainscore_vision/data/geirhos2021/data_packaging/edge/edge_stimulus_set.py +69 -0
- brainscore_vision/data/geirhos2021/data_packaging/eidolonI/eidolonI_data_assembly.py +92 -0
- brainscore_vision/data/geirhos2021/data_packaging/eidolonI/eidolonI_stimulus_set.py +82 -0
- brainscore_vision/data/geirhos2021/data_packaging/eidolonII/eidolonII_data_assembly.py +92 -0
- brainscore_vision/data/geirhos2021/data_packaging/eidolonII/eidolonII_stimulus_set.py +82 -0
- brainscore_vision/data/geirhos2021/data_packaging/eidolonIII/eidolonIII_data_assembly.py +92 -0
- brainscore_vision/data/geirhos2021/data_packaging/eidolonIII/eidolonIII_stimulus_set.py +82 -0
- brainscore_vision/data/geirhos2021/data_packaging/false-colour/false-colour_data_assembly.py +83 -0
- brainscore_vision/data/geirhos2021/data_packaging/false-colour/false-colour_stimulus_set.py +87 -0
- brainscore_vision/data/geirhos2021/data_packaging/high-pass/high-pass_data_assembly.py +84 -0
- brainscore_vision/data/geirhos2021/data_packaging/high-pass/high-pass_stimulus_set.py +82 -0
- brainscore_vision/data/geirhos2021/data_packaging/low-pass/low-pass_data_assembly.py +84 -0
- brainscore_vision/data/geirhos2021/data_packaging/low-pass/low-pass_stimulus_set.py +81 -0
- brainscore_vision/data/geirhos2021/data_packaging/phase-scrambling/phase-scrambling_data_assembly.py +84 -0
- brainscore_vision/data/geirhos2021/data_packaging/phase-scrambling/phase-scrambling_stimulus_set.py +82 -0
- brainscore_vision/data/geirhos2021/data_packaging/power-equalisation/power-equalisation_data_assembly.py +88 -0
- brainscore_vision/data/geirhos2021/data_packaging/power-equalisation/power-equalisation_stimulus_set.py +82 -0
- brainscore_vision/data/geirhos2021/data_packaging/rotation/rotation_data_assembly.py +87 -0
- brainscore_vision/data/geirhos2021/data_packaging/rotation/rotation_stimulus_set.py +82 -0
- brainscore_vision/data/geirhos2021/data_packaging/silhouette/silhouette_data_assembly.py +100 -0
- brainscore_vision/data/geirhos2021/data_packaging/silhouette/silhouette_stimulus_set.py +71 -0
- brainscore_vision/data/geirhos2021/data_packaging/sketch/sketch_data_assembly.py +88 -0
- brainscore_vision/data/geirhos2021/data_packaging/sketch/sketch_stimulus_set.py +75 -0
- brainscore_vision/data/geirhos2021/data_packaging/stylized/stylized_data_assembly.py +87 -0
- brainscore_vision/data/geirhos2021/data_packaging/stylized/stylized_stimulus_set.py +75 -0
- brainscore_vision/data/geirhos2021/data_packaging/uniform-noise/uniform-noise_data_assembly.py +86 -0
- brainscore_vision/data/geirhos2021/data_packaging/uniform-noise/uniform-noise_stimulus_set.py +82 -0
- brainscore_vision/data/geirhos2021/geirhos_hashes.csv +52 -0
- brainscore_vision/data/geirhos2021/test.py +330 -0
- brainscore_vision/data/hebart2023/__init__.py +23 -0
- brainscore_vision/data/hebart2023/packaging/data_assembly.py +40 -0
- brainscore_vision/data/hebart2023/packaging/stimulus_set.py +72 -0
- brainscore_vision/data/hebart2023/test.py +42 -0
- brainscore_vision/data/hendrycks2019/__init__.py +45 -0
- brainscore_vision/data/hendrycks2019/test.py +26 -0
- brainscore_vision/data/igustibagus2024/__init__.py +23 -0
- brainscore_vision/data/igustibagus2024/dependencies/data_pico/stimulus_dicarlo_domain_transfer.csv +3139 -0
- brainscore_vision/data/igustibagus2024/investigation_consistency.ipynb +346 -0
- brainscore_vision/data/igustibagus2024/merged_assembly/__init__.py +0 -0
- brainscore_vision/data/igustibagus2024/merged_assembly/create_merged_assembly.ipynb +649 -0
- brainscore_vision/data/igustibagus2024/merged_assembly/create_merged_assembly_and_stim.py +152 -0
- brainscore_vision/data/igustibagus2024/merged_assembly/create_stimulus_set_with_background-id.py +45 -0
- brainscore_vision/data/igustibagus2024/merged_assembly/helpers_background_id.py +849 -0
- brainscore_vision/data/igustibagus2024/merged_assembly/merged_stimulus_set.csv +3139 -0
- brainscore_vision/data/igustibagus2024/oleo_pico_exploration.ipynb +410 -0
- brainscore_vision/data/igustibagus2024/test.py +26 -0
- brainscore_vision/data/imagenetslim15000/ImageNetSlim15000.py +30 -0
- brainscore_vision/data/imagenetslim15000/__init__.py +11 -0
- brainscore_vision/data/imagenetslim15000/test.py +8 -0
- brainscore_vision/data/islam2021/__init__.py +18 -0
- brainscore_vision/data/islam2021/data_packaging.py +64 -0
- brainscore_vision/data/islam2021/test.py +11 -0
- brainscore_vision/data/kar2018/__init__.py +58 -0
- brainscore_vision/data/kar2018/data_packaging/kar_coco.py +97 -0
- brainscore_vision/data/kar2018/data_packaging/kar_hvm.py +77 -0
- brainscore_vision/data/kar2018/data_packaging/requirements.txt +1 -0
- brainscore_vision/data/kar2018/test.py +10 -0
- brainscore_vision/data/kar2019/__init__.py +43 -0
- brainscore_vision/data/kar2019/data_packaging.py +116 -0
- brainscore_vision/data/kar2019/test.py +8 -0
- brainscore_vision/data/kuzovkin2018/__init__.py +36 -0
- brainscore_vision/data/kuzovkin2018/createAssembliesBrainScore.py +103 -0
- brainscore_vision/data/kuzovkin2018/test.py +8 -0
- brainscore_vision/data/majajhong2015/__init__.py +113 -0
- brainscore_vision/data/majajhong2015/data_packaging/darren10ms.py +32 -0
- brainscore_vision/data/majajhong2015/data_packaging/data_packaging.py +65 -0
- brainscore_vision/data/majajhong2015/test.py +38 -0
- brainscore_vision/data/malania2007/__init__.py +254 -0
- brainscore_vision/data/malania2007/data_packaging/malania_data_assembly.py +79 -0
- brainscore_vision/data/malania2007/data_packaging/malania_stimulus_set.py +79 -0
- brainscore_vision/data/malania2007/test.py +147 -0
- brainscore_vision/data/maniquet2024/__init__.py +57 -0
- brainscore_vision/data/maniquet2024/data_packaging.py +151 -0
- brainscore_vision/data/maniquet2024/test.py +16 -0
- brainscore_vision/data/marques2020/__init__.py +123 -0
- brainscore_vision/data/marques2020/data_packaging/marques_cavanaugh2002a.py +84 -0
- brainscore_vision/data/marques2020/data_packaging/marques_devalois1982a.py +44 -0
- brainscore_vision/data/marques2020/data_packaging/marques_devalois1982b.py +54 -0
- brainscore_vision/data/marques2020/data_packaging/marques_freemanZiemba2013.py +252 -0
- brainscore_vision/data/marques2020/data_packaging/marques_gen_stim.py +95 -0
- brainscore_vision/data/marques2020/data_packaging/marques_ringach2002.py +95 -0
- brainscore_vision/data/marques2020/data_packaging/marques_schiller1976c.py +60 -0
- brainscore_vision/data/marques2020/data_packaging/marques_stim_common.py +389 -0
- brainscore_vision/data/marques2020/data_packaging/marques_utils.py +21 -0
- brainscore_vision/data/marques2020/data_packaging/setup.py +13 -0
- brainscore_vision/data/marques2020/test.py +54 -0
- brainscore_vision/data/rajalingham2018/__init__.py +56 -0
- brainscore_vision/data/rajalingham2018/rajalingham2018objectome.py +193 -0
- brainscore_vision/data/rajalingham2018/test.py +10 -0
- brainscore_vision/data/rajalingham2020/__init__.py +39 -0
- brainscore_vision/data/rajalingham2020/rajalingham2020orthographic_IT.py +97 -0
- brainscore_vision/data/rajalingham2020/test.py +8 -0
- brainscore_vision/data/rust2012/2020-12-28_rust.ipynb +3301 -0
- brainscore_vision/data/rust2012/__init__.py +45 -0
- brainscore_vision/data/rust2012/rust305.py +35 -0
- brainscore_vision/data/rust2012/test.py +47 -0
- brainscore_vision/data/sanghavi2020/__init__.py +119 -0
- brainscore_vision/data/sanghavi2020/data_packaging/environment.yml +36 -0
- brainscore_vision/data/sanghavi2020/data_packaging/requirements.txt +4 -0
- brainscore_vision/data/sanghavi2020/data_packaging/sanghavi2020.py +101 -0
- brainscore_vision/data/sanghavi2020/data_packaging/sanghavijozwik2020.py +148 -0
- brainscore_vision/data/sanghavi2020/data_packaging/sanghavikar2020.py +131 -0
- brainscore_vision/data/sanghavi2020/data_packaging/sanghavimurty2020.py +120 -0
- brainscore_vision/data/sanghavi2020/data_packaging/sanghavimurty2020things.py +138 -0
- brainscore_vision/data/sanghavi2020/data_packaging/sanghavimurty2020things1.py +118 -0
- brainscore_vision/data/sanghavi2020/data_packaging/sanghavimurty2020things2.py +118 -0
- brainscore_vision/data/sanghavi2020/test.py +13 -0
- brainscore_vision/data/scialom2024/__init__.py +386 -0
- brainscore_vision/data/scialom2024/data_packaging/scialom_data_assembly.py +164 -0
- brainscore_vision/data/scialom2024/data_packaging/scialom_stimulus_set.py +117 -0
- brainscore_vision/data/scialom2024/test.py +301 -0
- brainscore_vision/data/seibert2019/__init__.py +25 -0
- brainscore_vision/data/seibert2019/data_packaging/2020-10-13_juvenile.ipynb +35703 -0
- brainscore_vision/data/seibert2019/data_packaging/2020-11-18_juvenile_scratch.txt +556 -0
- brainscore_vision/data/seibert2019/data_packaging/2020-11-22_juvenile_dldata.ipynb +3614 -0
- brainscore_vision/data/seibert2019/data_packaging/juvenile.py +103 -0
- brainscore_vision/data/seibert2019/test.py +35 -0
- brainscore_vision/data/zhang2018/__init__.py +38 -0
- brainscore_vision/data/zhang2018/test.py +29 -0
- brainscore_vision/data_helpers/__init__.py +0 -0
- brainscore_vision/data_helpers/lookup_legacy.py +15 -0
- brainscore_vision/data_helpers/s3.py +79 -0
- brainscore_vision/metric_helpers/__init__.py +5 -0
- brainscore_vision/metric_helpers/temporal.py +119 -0
- brainscore_vision/metric_helpers/transformations.py +379 -0
- brainscore_vision/metric_helpers/utils.py +71 -0
- brainscore_vision/metric_helpers/xarray_utils.py +151 -0
- brainscore_vision/metrics/__init__.py +7 -0
- brainscore_vision/metrics/accuracy/__init__.py +4 -0
- brainscore_vision/metrics/accuracy/metric.py +16 -0
- brainscore_vision/metrics/accuracy/test.py +11 -0
- brainscore_vision/metrics/accuracy_distance/__init__.py +4 -0
- brainscore_vision/metrics/accuracy_distance/metric.py +109 -0
- brainscore_vision/metrics/accuracy_distance/test.py +57 -0
- brainscore_vision/metrics/baker_accuracy_delta/__init__.py +4 -0
- brainscore_vision/metrics/baker_accuracy_delta/metric.py +94 -0
- brainscore_vision/metrics/baker_accuracy_delta/requirements.txt +1 -0
- brainscore_vision/metrics/baker_accuracy_delta/test.py +1 -0
- brainscore_vision/metrics/cka/__init__.py +14 -0
- brainscore_vision/metrics/cka/metric.py +105 -0
- brainscore_vision/metrics/cka/test.py +28 -0
- brainscore_vision/metrics/dimensionality/__init__.py +13 -0
- brainscore_vision/metrics/dimensionality/metric.py +45 -0
- brainscore_vision/metrics/distribution_similarity/__init__.py +14 -0
- brainscore_vision/metrics/distribution_similarity/metric.py +84 -0
- brainscore_vision/metrics/distribution_similarity/test.py +10 -0
- brainscore_vision/metrics/error_consistency/__init__.py +13 -0
- brainscore_vision/metrics/error_consistency/metric.py +93 -0
- brainscore_vision/metrics/error_consistency/test.py +39 -0
- brainscore_vision/metrics/i1i2/__init__.py +16 -0
- brainscore_vision/metrics/i1i2/metric.py +299 -0
- brainscore_vision/metrics/i1i2/requirements.txt +2 -0
- brainscore_vision/metrics/i1i2/test.py +36 -0
- brainscore_vision/metrics/i1i2/test_resources/alexnet-probabilities.nc +0 -0
- brainscore_vision/metrics/i1i2/test_resources/resnet18-probabilities.nc +0 -0
- brainscore_vision/metrics/i1i2/test_resources/resnet34-probabilities.nc +0 -0
- brainscore_vision/metrics/internal_consistency/__init__.py +8 -0
- brainscore_vision/metrics/internal_consistency/ceiling.py +127 -0
- brainscore_vision/metrics/internal_consistency/requirements.txt +1 -0
- brainscore_vision/metrics/internal_consistency/test.py +39 -0
- brainscore_vision/metrics/maniquet2024_metrics/__init__.py +19 -0
- brainscore_vision/metrics/maniquet2024_metrics/metric.py +416 -0
- brainscore_vision/metrics/maniquet2024_metrics/test.py +8 -0
- brainscore_vision/metrics/mask_regression/__init__.py +16 -0
- brainscore_vision/metrics/mask_regression/metric.py +242 -0
- brainscore_vision/metrics/mask_regression/requirements.txt +1 -0
- brainscore_vision/metrics/mask_regression/test.py +0 -0
- brainscore_vision/metrics/ost/__init__.py +23 -0
- brainscore_vision/metrics/ost/metric.py +350 -0
- brainscore_vision/metrics/ost/requirements.txt +2 -0
- brainscore_vision/metrics/ost/test.py +0 -0
- brainscore_vision/metrics/rdm/__init__.py +14 -0
- brainscore_vision/metrics/rdm/metric.py +101 -0
- brainscore_vision/metrics/rdm/requirements.txt +2 -0
- brainscore_vision/metrics/rdm/test.py +63 -0
- brainscore_vision/metrics/regression_correlation/__init__.py +48 -0
- brainscore_vision/metrics/regression_correlation/mask_regression.py +232 -0
- brainscore_vision/metrics/regression_correlation/metric.py +125 -0
- brainscore_vision/metrics/regression_correlation/requirements.txt +3 -0
- brainscore_vision/metrics/regression_correlation/test.py +36 -0
- brainscore_vision/metrics/threshold/__init__.py +5 -0
- brainscore_vision/metrics/threshold/metric.py +481 -0
- brainscore_vision/metrics/threshold/test.py +71 -0
- brainscore_vision/metrics/value_delta/__init__.py +4 -0
- brainscore_vision/metrics/value_delta/metric.py +30 -0
- brainscore_vision/metrics/value_delta/requirements.txt +1 -0
- brainscore_vision/metrics/value_delta/test.py +40 -0
- brainscore_vision/model_helpers/__init__.py +3 -0
- brainscore_vision/model_helpers/activations/__init__.py +1 -0
- brainscore_vision/model_helpers/activations/core.py +635 -0
- brainscore_vision/model_helpers/activations/pca.py +117 -0
- brainscore_vision/model_helpers/activations/pytorch.py +152 -0
- brainscore_vision/model_helpers/activations/temporal/__init__.py +0 -0
- brainscore_vision/model_helpers/activations/temporal/core/__init__.py +3 -0
- brainscore_vision/model_helpers/activations/temporal/core/executor.py +219 -0
- brainscore_vision/model_helpers/activations/temporal/core/extractor.py +282 -0
- brainscore_vision/model_helpers/activations/temporal/core/inferencer/__init__.py +2 -0
- brainscore_vision/model_helpers/activations/temporal/core/inferencer/base.py +274 -0
- brainscore_vision/model_helpers/activations/temporal/core/inferencer/video/__init__.py +2 -0
- brainscore_vision/model_helpers/activations/temporal/core/inferencer/video/base.py +134 -0
- brainscore_vision/model_helpers/activations/temporal/core/inferencer/video/temporal_context/__init__.py +2 -0
- brainscore_vision/model_helpers/activations/temporal/core/inferencer/video/temporal_context/base.py +99 -0
- brainscore_vision/model_helpers/activations/temporal/core/inferencer/video/temporal_context/block.py +77 -0
- brainscore_vision/model_helpers/activations/temporal/core/inferencer/video/temporal_context/causal.py +86 -0
- brainscore_vision/model_helpers/activations/temporal/core/inferencer/video/time_aligner.py +73 -0
- brainscore_vision/model_helpers/activations/temporal/inputs/__init__.py +3 -0
- brainscore_vision/model_helpers/activations/temporal/inputs/base.py +17 -0
- brainscore_vision/model_helpers/activations/temporal/inputs/image.py +50 -0
- brainscore_vision/model_helpers/activations/temporal/inputs/video.py +186 -0
- brainscore_vision/model_helpers/activations/temporal/model/__init__.py +2 -0
- brainscore_vision/model_helpers/activations/temporal/model/base.py +33 -0
- brainscore_vision/model_helpers/activations/temporal/model/pytorch.py +107 -0
- brainscore_vision/model_helpers/activations/temporal/utils.py +228 -0
- brainscore_vision/model_helpers/brain_transformation/__init__.py +97 -0
- brainscore_vision/model_helpers/brain_transformation/behavior.py +348 -0
- brainscore_vision/model_helpers/brain_transformation/imagenet_classes.txt +1000 -0
- brainscore_vision/model_helpers/brain_transformation/neural.py +159 -0
- brainscore_vision/model_helpers/brain_transformation/temporal.py +199 -0
- brainscore_vision/model_helpers/check_submission/__init__.py +0 -0
- brainscore_vision/model_helpers/check_submission/check_models.py +87 -0
- brainscore_vision/model_helpers/check_submission/images/1.png +0 -0
- brainscore_vision/model_helpers/check_submission/images/10.png +0 -0
- brainscore_vision/model_helpers/check_submission/images/11.png +0 -0
- brainscore_vision/model_helpers/check_submission/images/12.png +0 -0
- brainscore_vision/model_helpers/check_submission/images/13.png +0 -0
- brainscore_vision/model_helpers/check_submission/images/14.png +0 -0
- brainscore_vision/model_helpers/check_submission/images/15.png +0 -0
- brainscore_vision/model_helpers/check_submission/images/16.png +0 -0
- brainscore_vision/model_helpers/check_submission/images/17.png +0 -0
- brainscore_vision/model_helpers/check_submission/images/18.png +0 -0
- brainscore_vision/model_helpers/check_submission/images/19.png +0 -0
- brainscore_vision/model_helpers/check_submission/images/2.png +0 -0
- brainscore_vision/model_helpers/check_submission/images/20.png +0 -0
- brainscore_vision/model_helpers/check_submission/images/3.png +0 -0
- brainscore_vision/model_helpers/check_submission/images/4.png +0 -0
- brainscore_vision/model_helpers/check_submission/images/5.png +0 -0
- brainscore_vision/model_helpers/check_submission/images/6.png +0 -0
- brainscore_vision/model_helpers/check_submission/images/7.png +0 -0
- brainscore_vision/model_helpers/check_submission/images/8.png +0 -0
- brainscore_vision/model_helpers/check_submission/images/9.png +0 -0
- brainscore_vision/model_helpers/conftest.py +3 -0
- brainscore_vision/model_helpers/generic_plugin_tests.py +119 -0
- brainscore_vision/model_helpers/s3.py +62 -0
- brainscore_vision/model_helpers/utils/__init__.py +15 -0
- brainscore_vision/model_helpers/utils/s3.py +42 -0
- brainscore_vision/model_interface.py +214 -0
- brainscore_vision/models/AdvProp_efficientne_b6/__init__.py +5 -0
- brainscore_vision/models/AdvProp_efficientne_b6/model.py +75 -0
- brainscore_vision/models/AdvProp_efficientne_b6/requirements.txt +1 -0
- brainscore_vision/models/AdvProp_efficientne_b6/test.py +9 -0
- brainscore_vision/models/AlexNet_SIN/__init__.py +8 -0
- brainscore_vision/models/AlexNet_SIN/model.py +29 -0
- brainscore_vision/models/AlexNet_SIN/requirements.txt +2 -0
- brainscore_vision/models/AlexNet_SIN/test.py +1 -0
- brainscore_vision/models/Soumyadeep_inf_1/__init__.py +5 -0
- brainscore_vision/models/Soumyadeep_inf_1/model.py +60 -0
- brainscore_vision/models/Soumyadeep_inf_1/setup.py +26 -0
- brainscore_vision/models/Soumyadeep_inf_1/test.py +1 -0
- brainscore_vision/models/ViT_L_32_imagenet1k/__init__.py +8 -0
- brainscore_vision/models/ViT_L_32_imagenet1k/model.py +43 -0
- brainscore_vision/models/ViT_L_32_imagenet1k/requirements.txt +4 -0
- brainscore_vision/models/ViT_L_32_imagenet1k/test.py +8 -0
- brainscore_vision/models/__init__.py +0 -0
- brainscore_vision/models/alexnet/__init__.py +8 -0
- brainscore_vision/models/alexnet/model.py +28 -0
- brainscore_vision/models/alexnet/requirements.txt +2 -0
- brainscore_vision/models/alexnet/test.py +15 -0
- brainscore_vision/models/alexnet_7be5be79/__init__.py +7 -0
- brainscore_vision/models/alexnet_7be5be79/model.py +44 -0
- brainscore_vision/models/alexnet_7be5be79/setup.py +26 -0
- brainscore_vision/models/alexnet_7be5be79/test.py +1 -0
- brainscore_vision/models/alexnet_7be5be79_convs/__init__.py +5 -0
- brainscore_vision/models/alexnet_7be5be79_convs/model.py +42 -0
- brainscore_vision/models/alexnet_7be5be79_convs/setup.py +25 -0
- brainscore_vision/models/alexnet_7be5be79_convs/test.py +1 -0
- brainscore_vision/models/alexnet_ks_torevert/__init__.py +8 -0
- brainscore_vision/models/alexnet_ks_torevert/model.py +28 -0
- brainscore_vision/models/alexnet_ks_torevert/requirements.txt +2 -0
- brainscore_vision/models/alexnet_ks_torevert/test.py +15 -0
- brainscore_vision/models/alexnet_simclr_run1/__init__.py +7 -0
- brainscore_vision/models/alexnet_simclr_run1/model.py +267 -0
- brainscore_vision/models/alexnet_simclr_run1/requirements.txt +2 -0
- brainscore_vision/models/alexnet_simclr_run1/test.py +1 -0
- brainscore_vision/models/alexnet_testing/__init__.py +8 -0
- brainscore_vision/models/alexnet_testing/model.py +28 -0
- brainscore_vision/models/alexnet_testing/requirements.txt +2 -0
- brainscore_vision/models/alexnet_testing/setup.py +24 -0
- brainscore_vision/models/alexnet_testing/test.py +15 -0
- brainscore_vision/models/antialias_resnet152/__init__.py +7 -0
- brainscore_vision/models/antialias_resnet152/model.py +35 -0
- brainscore_vision/models/antialias_resnet152/requirements.txt +3 -0
- brainscore_vision/models/antialias_resnet152/test.py +8 -0
- brainscore_vision/models/antialiased_rnext101_32x8d/__init__.py +7 -0
- brainscore_vision/models/antialiased_rnext101_32x8d/model.py +35 -0
- brainscore_vision/models/antialiased_rnext101_32x8d/requirements.txt +1 -0
- brainscore_vision/models/antialiased_rnext101_32x8d/test.py +8 -0
- brainscore_vision/models/bp_resnet50_julios/__init__.py +5 -0
- brainscore_vision/models/bp_resnet50_julios/model.py +52 -0
- brainscore_vision/models/bp_resnet50_julios/setup.py +24 -0
- brainscore_vision/models/bp_resnet50_julios/test.py +1 -0
- brainscore_vision/models/clip/__init__.py +5 -0
- brainscore_vision/models/clip/model.py +179 -0
- brainscore_vision/models/clip/requirements.txt +4 -0
- brainscore_vision/models/clip/test.py +1 -0
- brainscore_vision/models/clipvision/__init__.py +5 -0
- brainscore_vision/models/clipvision/model.py +179 -0
- brainscore_vision/models/clipvision/requirements.txt +4 -0
- brainscore_vision/models/clipvision/test.py +1 -0
- brainscore_vision/models/cornet_s/__init__.py +8 -0
- brainscore_vision/models/cornet_s/helpers/helpers.py +215 -0
- brainscore_vision/models/cornet_s/model.py +77 -0
- brainscore_vision/models/cornet_s/requirements.txt +7 -0
- brainscore_vision/models/cornet_s/test.py +8 -0
- brainscore_vision/models/cornet_s_ynshah/__init__.py +388 -0
- brainscore_vision/models/cornet_s_ynshah/model.py +192 -0
- brainscore_vision/models/cornet_s_ynshah/setup.py +24 -0
- brainscore_vision/models/cornet_s_ynshah/test.py +0 -0
- brainscore_vision/models/custom_model_cv_18_dagger_408/__init__.py +7 -0
- brainscore_vision/models/custom_model_cv_18_dagger_408/model.py +75 -0
- brainscore_vision/models/custom_model_cv_18_dagger_408/requirements.txt +4 -0
- brainscore_vision/models/custom_model_cv_18_dagger_408/test.py +8 -0
- brainscore_vision/models/cv_18_dagger_408_pretrained/__init__.py +8 -0
- brainscore_vision/models/cv_18_dagger_408_pretrained/model.py +57 -0
- brainscore_vision/models/cv_18_dagger_408_pretrained/requirements.txt +3 -0
- brainscore_vision/models/cv_18_dagger_408_pretrained/test.py +25 -0
- brainscore_vision/models/cvt_cvt_w24_384_in22k_finetuned_in1k_4/__init__.py +9 -0
- brainscore_vision/models/cvt_cvt_w24_384_in22k_finetuned_in1k_4/model.py +134 -0
- brainscore_vision/models/cvt_cvt_w24_384_in22k_finetuned_in1k_4/requirements.txt +4 -0
- brainscore_vision/models/cvt_cvt_w24_384_in22k_finetuned_in1k_4/test.py +8 -0
- brainscore_vision/models/dbp_resnet50_julios/__init__.py +5 -0
- brainscore_vision/models/dbp_resnet50_julios/model.py +52 -0
- brainscore_vision/models/dbp_resnet50_julios/setup.py +24 -0
- brainscore_vision/models/dbp_resnet50_julios/test.py +1 -0
- brainscore_vision/models/densenet_201_pytorch/__init__.py +7 -0
- brainscore_vision/models/densenet_201_pytorch/model.py +59 -0
- brainscore_vision/models/densenet_201_pytorch/requirements.txt +3 -0
- brainscore_vision/models/densenet_201_pytorch/test.py +8 -0
- brainscore_vision/models/eBarlow_Vanilla/__init__.py +9 -0
- brainscore_vision/models/eBarlow_Vanilla/model.py +50 -0
- brainscore_vision/models/eBarlow_Vanilla/requirements.txt +2 -0
- brainscore_vision/models/eBarlow_Vanilla/setup.py +24 -0
- brainscore_vision/models/eBarlow_Vanilla/test.py +1 -0
- brainscore_vision/models/eBarlow_Vanilla_1/__init__.py +9 -0
- brainscore_vision/models/eBarlow_Vanilla_1/model.py +64 -0
- brainscore_vision/models/eBarlow_Vanilla_1/setup.py +24 -0
- brainscore_vision/models/eBarlow_Vanilla_1/test.py +1 -0
- brainscore_vision/models/eBarlow_Vanilla_1_full/__init__.py +9 -0
- brainscore_vision/models/eBarlow_Vanilla_1_full/model.py +84 -0
- brainscore_vision/models/eBarlow_Vanilla_1_full/setup.py +25 -0
- brainscore_vision/models/eBarlow_Vanilla_1_full/test.py +1 -0
- brainscore_vision/models/eBarlow_Vanilla_2/__init__.py +9 -0
- brainscore_vision/models/eBarlow_Vanilla_2/model.py +64 -0
- brainscore_vision/models/eBarlow_Vanilla_2/setup.py +24 -0
- brainscore_vision/models/eBarlow_Vanilla_2/test.py +1 -0
- brainscore_vision/models/eBarlow_augself_linear_1/__init__.py +9 -0
- brainscore_vision/models/eBarlow_augself_linear_1/model.py +65 -0
- brainscore_vision/models/eBarlow_augself_linear_1/setup.py +24 -0
- brainscore_vision/models/eBarlow_augself_linear_1/test.py +1 -0
- brainscore_vision/models/eBarlow_augself_mlp_1/__init__.py +9 -0
- brainscore_vision/models/eBarlow_augself_mlp_1/model.py +65 -0
- brainscore_vision/models/eBarlow_augself_mlp_1/setup.py +24 -0
- brainscore_vision/models/eBarlow_augself_mlp_1/test.py +1 -0
- brainscore_vision/models/eBarlow_lmda_0001_1/__init__.py +9 -0
- brainscore_vision/models/eBarlow_lmda_0001_1/model.py +65 -0
- brainscore_vision/models/eBarlow_lmda_0001_1/setup.py +24 -0
- brainscore_vision/models/eBarlow_lmda_0001_1/test.py +1 -0
- brainscore_vision/models/eBarlow_lmda_001_1/__init__.py +9 -0
- brainscore_vision/models/eBarlow_lmda_001_1/model.py +65 -0
- brainscore_vision/models/eBarlow_lmda_001_1/setup.py +24 -0
- brainscore_vision/models/eBarlow_lmda_001_1/test.py +1 -0
- brainscore_vision/models/eBarlow_lmda_001_2/__init__.py +9 -0
- brainscore_vision/models/eBarlow_lmda_001_2/model.py +65 -0
- brainscore_vision/models/eBarlow_lmda_001_2/setup.py +24 -0
- brainscore_vision/models/eBarlow_lmda_001_2/test.py +1 -0
- brainscore_vision/models/eBarlow_lmda_001_3/__init__.py +9 -0
- brainscore_vision/models/eBarlow_lmda_001_3/model.py +65 -0
- brainscore_vision/models/eBarlow_lmda_001_3/setup.py +24 -0
- brainscore_vision/models/eBarlow_lmda_001_3/test.py +1 -0
- brainscore_vision/models/eBarlow_lmda_01/__init__.py +9 -0
- brainscore_vision/models/eBarlow_lmda_01/model.py +50 -0
- brainscore_vision/models/eBarlow_lmda_01/requirements.txt +2 -0
- brainscore_vision/models/eBarlow_lmda_01/setup.py +24 -0
- brainscore_vision/models/eBarlow_lmda_01/test.py +1 -0
- brainscore_vision/models/eBarlow_lmda_01_1/__init__.py +9 -0
- brainscore_vision/models/eBarlow_lmda_01_1/model.py +65 -0
- brainscore_vision/models/eBarlow_lmda_01_1/setup.py +24 -0
- brainscore_vision/models/eBarlow_lmda_01_1/test.py +1 -0
- brainscore_vision/models/eBarlow_lmda_01_2/__init__.py +9 -0
- brainscore_vision/models/eBarlow_lmda_01_2/model.py +65 -0
- brainscore_vision/models/eBarlow_lmda_01_2/setup.py +24 -0
- brainscore_vision/models/eBarlow_lmda_01_2/test.py +1 -0
- brainscore_vision/models/eBarlow_lmda_02_1/__init__.py +9 -0
- brainscore_vision/models/eBarlow_lmda_02_1/model.py +65 -0
- brainscore_vision/models/eBarlow_lmda_02_1/setup.py +24 -0
- brainscore_vision/models/eBarlow_lmda_02_1/test.py +1 -0
- brainscore_vision/models/eBarlow_lmda_02_1000ep/__init__.py +9 -0
- brainscore_vision/models/eBarlow_lmda_02_1000ep/model.py +84 -0
- brainscore_vision/models/eBarlow_lmda_02_1000ep/setup.py +25 -0
- brainscore_vision/models/eBarlow_lmda_02_1000ep/test.py +1 -0
- brainscore_vision/models/eBarlow_lmda_02_1_full/__init__.py +9 -0
- brainscore_vision/models/eBarlow_lmda_02_1_full/model.py +85 -0
- brainscore_vision/models/eBarlow_lmda_02_1_full/setup.py +25 -0
- brainscore_vision/models/eBarlow_lmda_02_1_full/test.py +1 -0
- brainscore_vision/models/eBarlow_lmda_02_200_full/__init__.py +9 -0
- brainscore_vision/models/eBarlow_lmda_02_200_full/model.py +85 -0
- brainscore_vision/models/eBarlow_lmda_02_200_full/setup.py +25 -0
- brainscore_vision/models/eBarlow_lmda_02_200_full/test.py +1 -0
- brainscore_vision/models/eBarlow_lmda_03_1/__init__.py +9 -0
- brainscore_vision/models/eBarlow_lmda_03_1/model.py +65 -0
- brainscore_vision/models/eBarlow_lmda_03_1/setup.py +24 -0
- brainscore_vision/models/eBarlow_lmda_03_1/test.py +1 -0
- brainscore_vision/models/eBarlow_lmda_04_1/__init__.py +9 -0
- brainscore_vision/models/eBarlow_lmda_04_1/model.py +65 -0
- brainscore_vision/models/eBarlow_lmda_04_1/setup.py +24 -0
- brainscore_vision/models/eBarlow_lmda_04_1/test.py +1 -0
- brainscore_vision/models/eBarlow_lmda_05_1/__init__.py +9 -0
- brainscore_vision/models/eBarlow_lmda_05_1/model.py +65 -0
- brainscore_vision/models/eBarlow_lmda_05_1/setup.py +24 -0
- brainscore_vision/models/eBarlow_lmda_05_1/test.py +1 -0
- brainscore_vision/models/eMMCR_Mom_Vanilla_1/__init__.py +9 -0
- brainscore_vision/models/eMMCR_Mom_Vanilla_1/model.py +64 -0
- brainscore_vision/models/eMMCR_Mom_Vanilla_1/setup.py +24 -0
- brainscore_vision/models/eMMCR_Mom_Vanilla_1/test.py +1 -0
- brainscore_vision/models/eMMCR_Mom_Vanilla_2/__init__.py +9 -0
- brainscore_vision/models/eMMCR_Mom_Vanilla_2/model.py +64 -0
- brainscore_vision/models/eMMCR_Mom_Vanilla_2/setup.py +24 -0
- brainscore_vision/models/eMMCR_Mom_Vanilla_2/test.py +1 -0
- brainscore_vision/models/eMMCR_Mom_lmda_0001_1/__init__.py +9 -0
- brainscore_vision/models/eMMCR_Mom_lmda_0001_1/model.py +65 -0
- brainscore_vision/models/eMMCR_Mom_lmda_0001_1/setup.py +24 -0
- brainscore_vision/models/eMMCR_Mom_lmda_0001_1/test.py +1 -0
- brainscore_vision/models/eMMCR_Mom_lmda_001_1/__init__.py +9 -0
- brainscore_vision/models/eMMCR_Mom_lmda_001_1/model.py +65 -0
- brainscore_vision/models/eMMCR_Mom_lmda_001_1/setup.py +24 -0
- brainscore_vision/models/eMMCR_Mom_lmda_001_1/test.py +1 -0
- brainscore_vision/models/eMMCR_Mom_lmda_01_1/__init__.py +9 -0
- brainscore_vision/models/eMMCR_Mom_lmda_01_1/model.py +65 -0
- brainscore_vision/models/eMMCR_Mom_lmda_01_1/setup.py +24 -0
- brainscore_vision/models/eMMCR_Mom_lmda_01_1/test.py +1 -0
- brainscore_vision/models/eMMCR_Mom_lmda_01_2/__init__.py +9 -0
- brainscore_vision/models/eMMCR_Mom_lmda_01_2/model.py +65 -0
- brainscore_vision/models/eMMCR_Mom_lmda_01_2/setup.py +24 -0
- brainscore_vision/models/eMMCR_Mom_lmda_01_2/test.py +1 -0
- brainscore_vision/models/eMMCR_Mom_lmda_02_1/__init__.py +9 -0
- brainscore_vision/models/eMMCR_Mom_lmda_02_1/model.py +65 -0
- brainscore_vision/models/eMMCR_Mom_lmda_02_1/setup.py +24 -0
- brainscore_vision/models/eMMCR_Mom_lmda_02_1/test.py +1 -0
- brainscore_vision/models/eMMCR_Mom_lmda_03_1/__init__.py +9 -0
- brainscore_vision/models/eMMCR_Mom_lmda_03_1/model.py +65 -0
- brainscore_vision/models/eMMCR_Mom_lmda_03_1/setup.py +24 -0
- brainscore_vision/models/eMMCR_Mom_lmda_03_1/test.py +1 -0
- brainscore_vision/models/eMMCR_Mom_lmda_04_1/__init__.py +9 -0
- brainscore_vision/models/eMMCR_Mom_lmda_04_1/model.py +65 -0
- brainscore_vision/models/eMMCR_Mom_lmda_04_1/setup.py +24 -0
- brainscore_vision/models/eMMCR_Mom_lmda_04_1/test.py +1 -0
- brainscore_vision/models/eMMCR_Mom_lmda_05_1/__init__.py +9 -0
- brainscore_vision/models/eMMCR_Mom_lmda_05_1/model.py +65 -0
- brainscore_vision/models/eMMCR_Mom_lmda_05_1/setup.py +24 -0
- brainscore_vision/models/eMMCR_Mom_lmda_05_1/test.py +1 -0
- brainscore_vision/models/eMMCR_Vanilla/__init__.py +9 -0
- brainscore_vision/models/eMMCR_Vanilla/model.py +50 -0
- brainscore_vision/models/eMMCR_Vanilla/setup.py +24 -0
- brainscore_vision/models/eMMCR_Vanilla/test.py +1 -0
- brainscore_vision/models/eMMCR_VanillaV2/__init__.py +9 -0
- brainscore_vision/models/eMMCR_VanillaV2/model.py +50 -0
- brainscore_vision/models/eMMCR_VanillaV2/setup.py +24 -0
- brainscore_vision/models/eMMCR_VanillaV2/test.py +1 -0
- brainscore_vision/models/eMMCR_Vanilla_1/__init__.py +9 -0
- brainscore_vision/models/eMMCR_Vanilla_1/model.py +64 -0
- brainscore_vision/models/eMMCR_Vanilla_1/setup.py +24 -0
- brainscore_vision/models/eMMCR_Vanilla_1/test.py +1 -0
- brainscore_vision/models/eMMCR_Vanilla_2/__init__.py +9 -0
- brainscore_vision/models/eMMCR_Vanilla_2/model.py +64 -0
- brainscore_vision/models/eMMCR_Vanilla_2/setup.py +24 -0
- brainscore_vision/models/eMMCR_Vanilla_2/test.py +1 -0
- brainscore_vision/models/eMMCR_lmda_01/__init__.py +9 -0
- brainscore_vision/models/eMMCR_lmda_01/model.py +50 -0
- brainscore_vision/models/eMMCR_lmda_01/setup.py +24 -0
- brainscore_vision/models/eMMCR_lmda_01/test.py +1 -0
- brainscore_vision/models/eMMCR_lmda_01V2/__init__.py +9 -0
- brainscore_vision/models/eMMCR_lmda_01V2/model.py +50 -0
- brainscore_vision/models/eMMCR_lmda_01V2/requirements.txt +2 -0
- brainscore_vision/models/eMMCR_lmda_01V2/setup.py +24 -0
- brainscore_vision/models/eMMCR_lmda_01V2/test.py +1 -0
- brainscore_vision/models/eMMCR_lmda_01_1/__init__.py +9 -0
- brainscore_vision/models/eMMCR_lmda_01_1/model.py +65 -0
- brainscore_vision/models/eMMCR_lmda_01_1/setup.py +24 -0
- brainscore_vision/models/eMMCR_lmda_01_1/test.py +1 -0
- brainscore_vision/models/eMMCR_lmda_01_2/__init__.py +9 -0
- brainscore_vision/models/eMMCR_lmda_01_2/model.py +65 -0
- brainscore_vision/models/eMMCR_lmda_01_2/setup.py +24 -0
- brainscore_vision/models/eMMCR_lmda_01_2/test.py +1 -0
- brainscore_vision/models/eMMCR_lmda_01_3/__init__.py +9 -0
- brainscore_vision/models/eMMCR_lmda_01_3/model.py +65 -0
- brainscore_vision/models/eMMCR_lmda_01_3/setup.py +24 -0
- brainscore_vision/models/eMMCR_lmda_01_3/test.py +1 -0
- brainscore_vision/models/eSimCLR_Vanilla_1/__init__.py +9 -0
- brainscore_vision/models/eSimCLR_Vanilla_1/model.py +64 -0
- brainscore_vision/models/eSimCLR_Vanilla_1/setup.py +24 -0
- brainscore_vision/models/eSimCLR_Vanilla_1/test.py +1 -0
- brainscore_vision/models/eSimCLR_Vanilla_2/__init__.py +9 -0
- brainscore_vision/models/eSimCLR_Vanilla_2/model.py +64 -0
- brainscore_vision/models/eSimCLR_Vanilla_2/setup.py +24 -0
- brainscore_vision/models/eSimCLR_Vanilla_2/test.py +1 -0
- brainscore_vision/models/eSimCLR_lmda_0001_1/__init__.py +9 -0
- brainscore_vision/models/eSimCLR_lmda_0001_1/model.py +65 -0
- brainscore_vision/models/eSimCLR_lmda_0001_1/setup.py +24 -0
- brainscore_vision/models/eSimCLR_lmda_0001_1/test.py +1 -0
- brainscore_vision/models/eSimCLR_lmda_001_1/__init__.py +9 -0
- brainscore_vision/models/eSimCLR_lmda_001_1/model.py +65 -0
- brainscore_vision/models/eSimCLR_lmda_001_1/setup.py +24 -0
- brainscore_vision/models/eSimCLR_lmda_001_1/test.py +1 -0
- brainscore_vision/models/eSimCLR_lmda_01_1/__init__.py +9 -0
- brainscore_vision/models/eSimCLR_lmda_01_1/model.py +65 -0
- brainscore_vision/models/eSimCLR_lmda_01_1/setup.py +24 -0
- brainscore_vision/models/eSimCLR_lmda_01_1/test.py +1 -0
- brainscore_vision/models/eSimCLR_lmda_01_2/__init__.py +9 -0
- brainscore_vision/models/eSimCLR_lmda_01_2/model.py +65 -0
- brainscore_vision/models/eSimCLR_lmda_01_2/setup.py +24 -0
- brainscore_vision/models/eSimCLR_lmda_01_2/test.py +1 -0
- brainscore_vision/models/eSimCLR_lmda_02_1/__init__.py +9 -0
- brainscore_vision/models/eSimCLR_lmda_02_1/model.py +65 -0
- brainscore_vision/models/eSimCLR_lmda_02_1/setup.py +24 -0
- brainscore_vision/models/eSimCLR_lmda_02_1/test.py +1 -0
- brainscore_vision/models/eSimCLR_lmda_02_1_1/__init__.py +9 -0
- brainscore_vision/models/eSimCLR_lmda_02_1_1/model.py +65 -0
- brainscore_vision/models/eSimCLR_lmda_02_1_1/setup.py +24 -0
- brainscore_vision/models/eSimCLR_lmda_02_1_1/test.py +1 -0
- brainscore_vision/models/eSimCLR_lmda_03_1/__init__.py +9 -0
- brainscore_vision/models/eSimCLR_lmda_03_1/model.py +65 -0
- brainscore_vision/models/eSimCLR_lmda_03_1/setup.py +24 -0
- brainscore_vision/models/eSimCLR_lmda_03_1/test.py +1 -0
- brainscore_vision/models/eSimCLR_lmda_04_1/__init__.py +9 -0
- brainscore_vision/models/eSimCLR_lmda_04_1/model.py +65 -0
- brainscore_vision/models/eSimCLR_lmda_04_1/setup.py +24 -0
- brainscore_vision/models/eSimCLR_lmda_04_1/test.py +1 -0
- brainscore_vision/models/eSimCLR_lmda_04_1_1/__init__.py +9 -0
- brainscore_vision/models/eSimCLR_lmda_04_1_1/model.py +65 -0
- brainscore_vision/models/eSimCLR_lmda_04_1_1/setup.py +24 -0
- brainscore_vision/models/eSimCLR_lmda_04_1_1/test.py +1 -0
- brainscore_vision/models/eSimCLR_lmda_05_1/__init__.py +9 -0
- brainscore_vision/models/eSimCLR_lmda_05_1/model.py +65 -0
- brainscore_vision/models/eSimCLR_lmda_05_1/setup.py +24 -0
- brainscore_vision/models/eSimCLR_lmda_05_1/test.py +1 -0
- brainscore_vision/models/effnetb1_272x240/__init__.py +5 -0
- brainscore_vision/models/effnetb1_272x240/model.py +126 -0
- brainscore_vision/models/effnetb1_272x240/requirements.txt +3 -0
- brainscore_vision/models/effnetb1_272x240/test.py +9 -0
- brainscore_vision/models/effnetb1_cutmix_augmix_sam_e1_5avg_424x377/__init__.py +9 -0
- brainscore_vision/models/effnetb1_cutmix_augmix_sam_e1_5avg_424x377/model.py +111 -0
- brainscore_vision/models/effnetb1_cutmix_augmix_sam_e1_5avg_424x377/requirements.txt +6 -0
- brainscore_vision/models/effnetb1_cutmix_augmix_sam_e1_5avg_424x377/test.py +8 -0
- brainscore_vision/models/effnetb1_cutmixpatch_SAM_robust32_avge6e8e9e10_manylayers_324x288/__init__.py +5 -0
- brainscore_vision/models/effnetb1_cutmixpatch_SAM_robust32_avge6e8e9e10_manylayers_324x288/model.py +142 -0
- brainscore_vision/models/effnetb1_cutmixpatch_SAM_robust32_avge6e8e9e10_manylayers_324x288/requirements.txt +5 -0
- brainscore_vision/models/effnetb1_cutmixpatch_SAM_robust32_avge6e8e9e10_manylayers_324x288/test.py +8 -0
- brainscore_vision/models/effnetb1_cutmixpatch_augmix_robust32_avge4e7_manylayers_324x288/__init__.py +9 -0
- brainscore_vision/models/effnetb1_cutmixpatch_augmix_robust32_avge4e7_manylayers_324x288/model.py +140 -0
- brainscore_vision/models/effnetb1_cutmixpatch_augmix_robust32_avge4e7_manylayers_324x288/requirements.txt +5 -0
- brainscore_vision/models/effnetb1_cutmixpatch_augmix_robust32_avge4e7_manylayers_324x288/test.py +8 -0
- brainscore_vision/models/focalnet_tiny_in1k_submission/__init__.py +5 -0
- brainscore_vision/models/focalnet_tiny_in1k_submission/model.py +62 -0
- brainscore_vision/models/focalnet_tiny_in1k_submission/requirements.txt +3 -0
- brainscore_vision/models/focalnet_tiny_in1k_submission/test.py +8 -0
- brainscore_vision/models/hmax/__init__.py +7 -0
- brainscore_vision/models/hmax/helpers/hmax.py +438 -0
- brainscore_vision/models/hmax/helpers/pytorch.py +216 -0
- brainscore_vision/models/hmax/model.py +69 -0
- brainscore_vision/models/hmax/requirements.txt +5 -0
- brainscore_vision/models/hmax/test.py +8 -0
- brainscore_vision/models/inception_v3_pytorch/__init__.py +7 -0
- brainscore_vision/models/inception_v3_pytorch/model.py +68 -0
- brainscore_vision/models/inception_v3_pytorch/requirements.txt +3 -0
- brainscore_vision/models/inception_v3_pytorch/test.py +8 -0
- brainscore_vision/models/mobilenet_v2_1_4_224_pytorch/__init__.py +7 -0
- brainscore_vision/models/mobilenet_v2_1_4_224_pytorch/model.py +60 -0
- brainscore_vision/models/mobilenet_v2_1_4_224_pytorch/requirements.txt +3 -0
- brainscore_vision/models/mobilenet_v2_1_4_224_pytorch/test.py +8 -0
- brainscore_vision/models/mobilevit_small/__init__.py +7 -0
- brainscore_vision/models/mobilevit_small/model.py +49 -0
- brainscore_vision/models/mobilevit_small/requirements.txt +3 -0
- brainscore_vision/models/mobilevit_small/test.py +8 -0
- brainscore_vision/models/pixels/__init__.py +8 -0
- brainscore_vision/models/pixels/model.py +35 -0
- brainscore_vision/models/pixels/test.py +15 -0
- brainscore_vision/models/pnasnet_large_pytorch/__init__.py +7 -0
- brainscore_vision/models/pnasnet_large_pytorch/model.py +59 -0
- brainscore_vision/models/pnasnet_large_pytorch/requirements.txt +3 -0
- brainscore_vision/models/pnasnet_large_pytorch/test.py +8 -0
- brainscore_vision/models/r101_eBarlow_Vanilla_1/__init__.py +9 -0
- brainscore_vision/models/r101_eBarlow_Vanilla_1/model.py +64 -0
- brainscore_vision/models/r101_eBarlow_Vanilla_1/setup.py +25 -0
- brainscore_vision/models/r101_eBarlow_Vanilla_1/test.py +1 -0
- brainscore_vision/models/r101_eBarlow_lmda_01_1/__init__.py +9 -0
- brainscore_vision/models/r101_eBarlow_lmda_01_1/model.py +65 -0
- brainscore_vision/models/r101_eBarlow_lmda_01_1/setup.py +25 -0
- brainscore_vision/models/r101_eBarlow_lmda_01_1/test.py +1 -0
- brainscore_vision/models/r101_eBarlow_lmda_02_1/__init__.py +9 -0
- brainscore_vision/models/r101_eBarlow_lmda_02_1/model.py +65 -0
- brainscore_vision/models/r101_eBarlow_lmda_02_1/setup.py +25 -0
- brainscore_vision/models/r101_eBarlow_lmda_02_1/test.py +1 -0
- brainscore_vision/models/r101_eBarlow_lmda_02_1_copy/__init__.py +9 -0
- brainscore_vision/models/r101_eBarlow_lmda_02_1_copy/model.py +67 -0
- brainscore_vision/models/r101_eBarlow_lmda_02_1_copy/setup.py +25 -0
- brainscore_vision/models/r101_eBarlow_lmda_02_1_copy/test.py +1 -0
- brainscore_vision/models/r34_eMMCR_Mom_Vanilla_1/__init__.py +9 -0
- brainscore_vision/models/r34_eMMCR_Mom_Vanilla_1/model.py +66 -0
- brainscore_vision/models/r34_eMMCR_Mom_Vanilla_1/setup.py +25 -0
- brainscore_vision/models/r34_eMMCR_Mom_Vanilla_1/test.py +1 -0
- brainscore_vision/models/r34_eMMCR_Mom_lmda_01_1/__init__.py +9 -0
- brainscore_vision/models/r34_eMMCR_Mom_lmda_01_1/model.py +66 -0
- brainscore_vision/models/r34_eMMCR_Mom_lmda_01_1/setup.py +25 -0
- brainscore_vision/models/r34_eMMCR_Mom_lmda_01_1/test.py +1 -0
- brainscore_vision/models/r34_eMMCR_Mom_lmda_02_1/__init__.py +9 -0
- brainscore_vision/models/r34_eMMCR_Mom_lmda_02_1/model.py +66 -0
- brainscore_vision/models/r34_eMMCR_Mom_lmda_02_1/setup.py +25 -0
- brainscore_vision/models/r34_eMMCR_Mom_lmda_02_1/test.py +1 -0
- brainscore_vision/models/r50_tvpt/__init__.py +9 -0
- brainscore_vision/models/r50_tvpt/model.py +47 -0
- brainscore_vision/models/r50_tvpt/setup.py +24 -0
- brainscore_vision/models/r50_tvpt/test.py +1 -0
- brainscore_vision/models/regnet/__init__.py +14 -0
- brainscore_vision/models/regnet/model.py +17 -0
- brainscore_vision/models/regnet/requirements.txt +2 -0
- brainscore_vision/models/regnet/test.py +17 -0
- brainscore_vision/models/resnet18_imagenet21kP/__init__.py +6 -0
- brainscore_vision/models/resnet18_imagenet21kP/model.py +119 -0
- brainscore_vision/models/resnet18_imagenet21kP/setup.py +18 -0
- brainscore_vision/models/resnet18_imagenet21kP/test.py +0 -0
- brainscore_vision/models/resnet50_eMMCR_Vanilla/__init__.py +5 -0
- brainscore_vision/models/resnet50_eMMCR_Vanilla/model.py +59 -0
- brainscore_vision/models/resnet50_eMMCR_Vanilla/setup.py +24 -0
- brainscore_vision/models/resnet50_eMMCR_Vanilla/test.py +1 -0
- brainscore_vision/models/resnet50_eMMCR_VanillaV2/__init__.py +9 -0
- brainscore_vision/models/resnet50_eMMCR_VanillaV2/model.py +72 -0
- brainscore_vision/models/resnet50_eMMCR_VanillaV2/setup.py +24 -0
- brainscore_vision/models/resnet50_eMMCR_VanillaV2/test.py +1 -0
- brainscore_vision/models/resnet50_eMMCR_eqp10_lm1/__init__.py +9 -0
- brainscore_vision/models/resnet50_eMMCR_eqp10_lm1/model.py +72 -0
- brainscore_vision/models/resnet50_eMMCR_eqp10_lm1/setup.py +24 -0
- brainscore_vision/models/resnet50_eMMCR_eqp10_lm1/test.py +1 -0
- brainscore_vision/models/resnet50_julios/__init__.py +5 -0
- brainscore_vision/models/resnet50_julios/model.py +54 -0
- brainscore_vision/models/resnet50_julios/setup.py +24 -0
- brainscore_vision/models/resnet50_julios/test.py +1 -0
- brainscore_vision/models/resnet50_tutorial/__init__.py +5 -0
- brainscore_vision/models/resnet50_tutorial/model.py +34 -0
- brainscore_vision/models/resnet50_tutorial/requirements.txt +2 -0
- brainscore_vision/models/resnet50_tutorial/test.py +8 -0
- brainscore_vision/models/resnet_152_v2_pytorch/__init__.py +7 -0
- brainscore_vision/models/resnet_152_v2_pytorch/model.py +59 -0
- brainscore_vision/models/resnet_152_v2_pytorch/requirements.txt +2 -0
- brainscore_vision/models/resnet_152_v2_pytorch/test.py +8 -0
- brainscore_vision/models/resnet_50_robust/__init__.py +7 -0
- brainscore_vision/models/resnet_50_robust/model.py +55 -0
- brainscore_vision/models/resnet_50_robust/requirements.txt +3 -0
- brainscore_vision/models/resnet_50_robust/test.py +8 -0
- brainscore_vision/models/resnext101_32x16d_wsl/__init__.py +7 -0
- brainscore_vision/models/resnext101_32x16d_wsl/model.py +38 -0
- brainscore_vision/models/resnext101_32x16d_wsl/requirements.txt +2 -0
- brainscore_vision/models/resnext101_32x16d_wsl/test.py +8 -0
- brainscore_vision/models/resnext101_32x32d_wsl/__init__.py +7 -0
- brainscore_vision/models/resnext101_32x32d_wsl/model.py +40 -0
- brainscore_vision/models/resnext101_32x32d_wsl/requirements.txt +2 -0
- brainscore_vision/models/resnext101_32x32d_wsl/test.py +8 -0
- brainscore_vision/models/resnext101_32x48d_wsl/__init__.py +7 -0
- brainscore_vision/models/resnext101_32x48d_wsl/model.py +38 -0
- brainscore_vision/models/resnext101_32x48d_wsl/requirements.txt +3 -0
- brainscore_vision/models/resnext101_32x48d_wsl/test.py +8 -0
- brainscore_vision/models/resnext101_32x8d_wsl/__init__.py +7 -0
- brainscore_vision/models/resnext101_32x8d_wsl/model.py +44 -0
- brainscore_vision/models/resnext101_32x8d_wsl/requirements.txt +2 -0
- brainscore_vision/models/resnext101_32x8d_wsl/test.py +8 -0
- brainscore_vision/models/temporal_model_AVID_CMA/__init__.py +17 -0
- brainscore_vision/models/temporal_model_AVID_CMA/model.py +92 -0
- brainscore_vision/models/temporal_model_AVID_CMA/requirements.txt +3 -0
- brainscore_vision/models/temporal_model_AVID_CMA/test.py +18 -0
- brainscore_vision/models/temporal_model_GDT/__init__.py +16 -0
- brainscore_vision/models/temporal_model_GDT/model.py +72 -0
- brainscore_vision/models/temporal_model_GDT/requirements.txt +3 -0
- brainscore_vision/models/temporal_model_GDT/test.py +17 -0
- brainscore_vision/models/temporal_model_S3D_text_video/__init__.py +14 -0
- brainscore_vision/models/temporal_model_S3D_text_video/model.py +65 -0
- brainscore_vision/models/temporal_model_S3D_text_video/requirements.txt +1 -0
- brainscore_vision/models/temporal_model_S3D_text_video/test.py +15 -0
- brainscore_vision/models/temporal_model_SeLaVi/__init__.py +17 -0
- brainscore_vision/models/temporal_model_SeLaVi/model.py +68 -0
- brainscore_vision/models/temporal_model_SeLaVi/requirements.txt +3 -0
- brainscore_vision/models/temporal_model_SeLaVi/test.py +18 -0
- brainscore_vision/models/temporal_model_VideoMAE/__init__.py +15 -0
- brainscore_vision/models/temporal_model_VideoMAE/model.py +100 -0
- brainscore_vision/models/temporal_model_VideoMAE/requirements.txt +6 -0
- brainscore_vision/models/temporal_model_VideoMAE/test.py +16 -0
- brainscore_vision/models/temporal_model_VideoMAEv2/__init__.py +14 -0
- brainscore_vision/models/temporal_model_VideoMAEv2/model.py +109 -0
- brainscore_vision/models/temporal_model_VideoMAEv2/requirements.txt +4 -0
- brainscore_vision/models/temporal_model_VideoMAEv2/test.py +16 -0
- brainscore_vision/models/temporal_model_mae_st/__init__.py +15 -0
- brainscore_vision/models/temporal_model_mae_st/model.py +120 -0
- brainscore_vision/models/temporal_model_mae_st/requirements.txt +3 -0
- brainscore_vision/models/temporal_model_mae_st/test.py +16 -0
- brainscore_vision/models/temporal_model_mmaction2/__init__.py +23 -0
- brainscore_vision/models/temporal_model_mmaction2/mmaction2.csv +24 -0
- brainscore_vision/models/temporal_model_mmaction2/model.py +226 -0
- brainscore_vision/models/temporal_model_mmaction2/requirements.txt +5 -0
- brainscore_vision/models/temporal_model_mmaction2/test.py +24 -0
- brainscore_vision/models/temporal_model_openstl/__init__.py +18 -0
- brainscore_vision/models/temporal_model_openstl/model.py +206 -0
- brainscore_vision/models/temporal_model_openstl/requirements.txt +3 -0
- brainscore_vision/models/temporal_model_openstl/test.py +19 -0
- brainscore_vision/models/temporal_model_torchvision/__init__.py +19 -0
- brainscore_vision/models/temporal_model_torchvision/model.py +92 -0
- brainscore_vision/models/temporal_model_torchvision/requirements.txt +2 -0
- brainscore_vision/models/temporal_model_torchvision/test.py +20 -0
- brainscore_vision/models/tv_efficientnet_b1/__init__.py +5 -0
- brainscore_vision/models/tv_efficientnet_b1/model.py +54 -0
- brainscore_vision/models/tv_efficientnet_b1/setup.py +24 -0
- brainscore_vision/models/tv_efficientnet_b1/test.py +1 -0
- brainscore_vision/models/voneresnet_50_non_stochastic/__init__.py +7 -0
- brainscore_vision/models/voneresnet_50_non_stochastic/model.py +104 -0
- brainscore_vision/models/voneresnet_50_non_stochastic/requirements.txt +8 -0
- brainscore_vision/models/voneresnet_50_non_stochastic/test.py +8 -0
- brainscore_vision/models/voneresnet_50_non_stochastic/vonenet/LICENSE +674 -0
- brainscore_vision/models/voneresnet_50_non_stochastic/vonenet/README.md +105 -0
- brainscore_vision/models/voneresnet_50_non_stochastic/vonenet/run.py +136 -0
- brainscore_vision/models/voneresnet_50_non_stochastic/vonenet/setup.py +41 -0
- brainscore_vision/models/voneresnet_50_non_stochastic/vonenet/train.py +383 -0
- brainscore_vision/models/voneresnet_50_non_stochastic/vonenet/vonenet/__init__.py +71 -0
- brainscore_vision/models/voneresnet_50_non_stochastic/vonenet/vonenet/back_ends.py +337 -0
- brainscore_vision/models/voneresnet_50_non_stochastic/vonenet/vonenet/modules.py +126 -0
- brainscore_vision/models/voneresnet_50_non_stochastic/vonenet/vonenet/params.py +100 -0
- brainscore_vision/models/voneresnet_50_non_stochastic/vonenet/vonenet/utils.py +32 -0
- brainscore_vision/models/voneresnet_50_non_stochastic/vonenet/vonenet/vonenet.py +68 -0
- brainscore_vision/models/voneresnet_50_non_stochastic/vonenet/vonenet_tutorial-activations.ipynb +352 -0
- brainscore_vision/models/yudixie_resnet18_240719_0/__init__.py +11 -0
- brainscore_vision/models/yudixie_resnet18_240719_0/model.py +60 -0
- brainscore_vision/models/yudixie_resnet18_240719_0/setup.py +25 -0
- brainscore_vision/models/yudixie_resnet18_240719_0/test.py +1 -0
- brainscore_vision/models/yudixie_resnet18_240719_1/__init__.py +11 -0
- brainscore_vision/models/yudixie_resnet18_240719_1/model.py +60 -0
- brainscore_vision/models/yudixie_resnet18_240719_1/setup.py +25 -0
- brainscore_vision/models/yudixie_resnet18_240719_1/test.py +1 -0
- brainscore_vision/models/yudixie_resnet18_240719_10/__init__.py +11 -0
- brainscore_vision/models/yudixie_resnet18_240719_10/model.py +60 -0
- brainscore_vision/models/yudixie_resnet18_240719_10/setup.py +25 -0
- brainscore_vision/models/yudixie_resnet18_240719_10/test.py +1 -0
- brainscore_vision/models/yudixie_resnet18_240719_2/__init__.py +11 -0
- brainscore_vision/models/yudixie_resnet18_240719_2/model.py +60 -0
- brainscore_vision/models/yudixie_resnet18_240719_2/setup.py +25 -0
- brainscore_vision/models/yudixie_resnet18_240719_2/test.py +1 -0
- brainscore_vision/models/yudixie_resnet50_imagenet1kpret_0_240222/__init__.py +7 -0
- brainscore_vision/models/yudixie_resnet50_imagenet1kpret_0_240222/model.py +66 -0
- brainscore_vision/models/yudixie_resnet50_imagenet1kpret_0_240222/setup.py +24 -0
- brainscore_vision/models/yudixie_resnet50_imagenet1kpret_0_240222/test.py +1 -0
- brainscore_vision/models/yudixie_resnet50_imagenet1kpret_0_240312/__init__.py +7 -0
- brainscore_vision/models/yudixie_resnet50_imagenet1kpret_0_240312/model.py +68 -0
- brainscore_vision/models/yudixie_resnet50_imagenet1kpret_0_240312/setup.py +24 -0
- brainscore_vision/models/yudixie_resnet50_imagenet1kpret_0_240312/test.py +1 -0
- brainscore_vision/submission/__init__.py +0 -0
- brainscore_vision/submission/actions_helpers.py +153 -0
- brainscore_vision/submission/config.py +7 -0
- brainscore_vision/submission/endpoints.py +58 -0
- brainscore_vision/utils/__init__.py +91 -0
- brainscore_vision-2.1.dist-info/LICENSE +11 -0
- brainscore_vision-2.1.dist-info/METADATA +152 -0
- brainscore_vision-2.1.dist-info/RECORD +1009 -0
- brainscore_vision-2.1.dist-info/WHEEL +5 -0
- brainscore_vision-2.1.dist-info/top_level.txt +4 -0
- docs/Makefile +20 -0
- docs/source/conf.py +78 -0
- docs/source/index.rst +21 -0
- docs/source/modules/api_reference.rst +10 -0
- docs/source/modules/benchmarks.rst +8 -0
- docs/source/modules/brainscore_submission.png +0 -0
- docs/source/modules/developer_clarifications.rst +36 -0
- docs/source/modules/metrics.rst +8 -0
- docs/source/modules/model_interface.rst +8 -0
- docs/source/modules/submission.rst +112 -0
- docs/source/modules/tutorial_screenshots/brain-score_logo.png +0 -0
- docs/source/modules/tutorial_screenshots/final_submit.png +0 -0
- docs/source/modules/tutorial_screenshots/init_py.png +0 -0
- docs/source/modules/tutorial_screenshots/mms.png +0 -0
- docs/source/modules/tutorial_screenshots/setup.png +0 -0
- docs/source/modules/tutorial_screenshots/sms.png +0 -0
- docs/source/modules/tutorial_screenshots/subfolders.png +0 -0
- docs/source/modules/utils.rst +22 -0
- migrations/2020-12-20_pkl_to_nc.py +90 -0
- tests/__init__.py +6 -0
- tests/conftest.py +26 -0
- tests/test_benchmark_helpers/__init__.py +0 -0
- tests/test_benchmark_helpers/test_screen.py +75 -0
- tests/test_examples.py +41 -0
- tests/test_integration.py +43 -0
- tests/test_metric_helpers/__init__.py +0 -0
- tests/test_metric_helpers/test_temporal.py +80 -0
- tests/test_metric_helpers/test_transformations.py +171 -0
- tests/test_metric_helpers/test_xarray_utils.py +85 -0
- tests/test_model_helpers/__init__.py +6 -0
- tests/test_model_helpers/activations/__init__.py +0 -0
- tests/test_model_helpers/activations/test___init__.py +404 -0
- tests/test_model_helpers/brain_transformation/__init__.py +0 -0
- tests/test_model_helpers/brain_transformation/test___init__.py +18 -0
- tests/test_model_helpers/brain_transformation/test_behavior.py +181 -0
- tests/test_model_helpers/brain_transformation/test_neural.py +70 -0
- tests/test_model_helpers/brain_transformation/test_temporal.py +66 -0
- tests/test_model_helpers/temporal/__init__.py +0 -0
- tests/test_model_helpers/temporal/activations/__init__.py +0 -0
- tests/test_model_helpers/temporal/activations/test_extractor.py +96 -0
- tests/test_model_helpers/temporal/activations/test_inferencer.py +189 -0
- tests/test_model_helpers/temporal/activations/test_inputs.py +103 -0
- tests/test_model_helpers/temporal/brain_transformation/__init__.py +0 -0
- tests/test_model_helpers/temporal/brain_transformation/test_temporal_ops.py +122 -0
- tests/test_model_helpers/temporal/test_utils.py +61 -0
- tests/test_model_helpers/test_generic_plugin_tests.py +310 -0
- tests/test_model_helpers/test_imports.py +10 -0
- tests/test_model_helpers/test_s3.py +38 -0
- tests/test_models.py +15 -0
- tests/test_stimuli.py +0 -0
- tests/test_submission/__init__.py +0 -0
- tests/test_submission/mock_config.py +3 -0
- tests/test_submission/test_actions_helpers.py +67 -0
- tests/test_submission/test_db.py +54 -0
- tests/test_submission/test_endpoints.py +125 -0
- tests/test_utils.py +21 -0
@@ -0,0 +1,138 @@
|
|
1
|
+
import numpy as np
|
2
|
+
from result_caching import store
|
3
|
+
|
4
|
+
from brainio.assemblies import DataAssembly
|
5
|
+
from brainscore_vision import load_dataset, load_metric
|
6
|
+
from brainscore_vision.benchmark_helpers.properties_common import PropertiesBenchmark, _assert_texture_activations
|
7
|
+
from brainscore_vision.benchmark_helpers.properties_common import calc_texture_modulation, calc_sparseness, \
|
8
|
+
calc_variance_ratio
|
9
|
+
from brainscore_vision.data.freemanziemba2013 import BIBTEX
|
10
|
+
from brainscore_vision.metrics.distribution_similarity import NeuronalPropertyCeiling
|
11
|
+
|
12
|
+
ASSEMBLY_NAME = 'FreemanZiemba2013_V1_properties'
|
13
|
+
REGION = 'V1'
|
14
|
+
TIMEBINS = [(70, 170)]
|
15
|
+
PARENT_TEXTURE_MODULATION = 'V1-texture_modulation'
|
16
|
+
PARENT_SELECTIVITY = 'V1-response_selectivity'
|
17
|
+
PARENT_MAGNITUDE = 'V1-response_magnitude'
|
18
|
+
|
19
|
+
PROPERTY_NAMES = ['texture_modulation_index', 'absolute_texture_modulation_index', 'texture_selectivity',
|
20
|
+
'noise_selectivity', 'texture_sparseness', 'noise_sparseness', 'variance_ratio', 'sample_variance',
|
21
|
+
'family_variance', 'max_texture', 'max_noise']
|
22
|
+
|
23
|
+
RESPONSE_THRESHOLD = 5
|
24
|
+
|
25
|
+
|
26
|
+
def _MarquesFreemanZiemba2013V1Property(property_name, parent, property_identifier=None):
|
27
|
+
assembly = load_dataset(ASSEMBLY_NAME)
|
28
|
+
similarity_metric = load_metric('ks_similarity', property_name=property_name)
|
29
|
+
ceil_func = NeuronalPropertyCeiling(similarity_metric)
|
30
|
+
|
31
|
+
if not property_identifier:
|
32
|
+
property_identifier = property_name
|
33
|
+
return PropertiesBenchmark(identifier=f'Marques2020_FreemanZiemba2013-{property_identifier}', assembly=assembly,
|
34
|
+
neuronal_property=freemanziemba2013_properties, similarity_metric=similarity_metric,
|
35
|
+
timebins=TIMEBINS,
|
36
|
+
parent=parent, ceiling_func=ceil_func, bibtex=BIBTEX, version=1)
|
37
|
+
|
38
|
+
|
39
|
+
def MarquesFreemanZiemba2013V1TextureModulationIndex():
|
40
|
+
property_name = 'texture_modulation_index'
|
41
|
+
parent = PARENT_TEXTURE_MODULATION
|
42
|
+
return _MarquesFreemanZiemba2013V1Property(property_name=property_name, parent=parent)
|
43
|
+
|
44
|
+
|
45
|
+
def MarquesFreemanZiemba2013V1AbsoluteTextureModulationIndex():
|
46
|
+
property_name = 'absolute_texture_modulation_index'
|
47
|
+
property_identifier = 'abs_texture_modulation_index'
|
48
|
+
parent = PARENT_TEXTURE_MODULATION
|
49
|
+
return _MarquesFreemanZiemba2013V1Property(property_name=property_name, parent=parent, property_identifier=property_identifier)
|
50
|
+
|
51
|
+
|
52
|
+
def MarquesFreemanZiemba2013V1TextureSelectivity():
|
53
|
+
property_name = 'texture_selectivity'
|
54
|
+
parent = PARENT_SELECTIVITY
|
55
|
+
return _MarquesFreemanZiemba2013V1Property(property_name=property_name, parent=parent)
|
56
|
+
|
57
|
+
|
58
|
+
def MarquesFreemanZiemba2013V1TextureSparseness():
|
59
|
+
property_name = 'texture_sparseness'
|
60
|
+
parent = PARENT_SELECTIVITY
|
61
|
+
return _MarquesFreemanZiemba2013V1Property(property_name=property_name, parent=parent)
|
62
|
+
|
63
|
+
|
64
|
+
def MarquesFreemanZiemba2013V1VarianceRatio():
|
65
|
+
property_name = 'variance_ratio'
|
66
|
+
property_identifier = 'texture_variance_ratio'
|
67
|
+
parent = PARENT_SELECTIVITY
|
68
|
+
return _MarquesFreemanZiemba2013V1Property(property_name=property_name, parent=parent, property_identifier=property_identifier)
|
69
|
+
|
70
|
+
|
71
|
+
def MarquesFreemanZiemba2013V1MaxTexture():
|
72
|
+
property_name = 'max_texture'
|
73
|
+
parent = PARENT_MAGNITUDE
|
74
|
+
return _MarquesFreemanZiemba2013V1Property(property_name=property_name, parent=parent)
|
75
|
+
|
76
|
+
|
77
|
+
def MarquesFreemanZiemba2013V1MaxNoise():
|
78
|
+
property_name = 'max_noise'
|
79
|
+
parent = PARENT_MAGNITUDE
|
80
|
+
return _MarquesFreemanZiemba2013V1Property(property_name=property_name, parent=parent)
|
81
|
+
|
82
|
+
|
83
|
+
@store(identifier_ignore=['responses', 'baseline'])
|
84
|
+
def freemanziemba2013_properties(model_identifier, responses, baseline):
|
85
|
+
_assert_texture_activations(responses)
|
86
|
+
responses = responses.sortby(['type', 'family', 'sample'])
|
87
|
+
type = np.array(sorted(set(responses.type.values)))
|
88
|
+
family = np.array(sorted(set(responses.family.values)))
|
89
|
+
sample = np.array(sorted(set(responses.sample.values)))
|
90
|
+
|
91
|
+
responses = responses.values
|
92
|
+
baseline = baseline.values
|
93
|
+
assert responses.shape[0] == baseline.shape[0]
|
94
|
+
n_neuroids = responses.shape[0]
|
95
|
+
|
96
|
+
responses = responses.reshape(n_neuroids, len(type), len(family), len(sample))
|
97
|
+
responses_spikes = responses / 10
|
98
|
+
responses_spikes = np.sqrt(responses_spikes) + np.sqrt(responses_spikes + 1)
|
99
|
+
responses -= baseline.reshape((-1, 1, 1, 1))
|
100
|
+
|
101
|
+
max_texture = np.max((responses.reshape((n_neuroids, 2, -1)))[:, 1, :], axis=1, keepdims=True)
|
102
|
+
max_noise = np.max((responses.reshape((n_neuroids, 2, -1)))[:, 0, :], axis=1, keepdims=True)
|
103
|
+
max_response = np.max(responses.reshape((n_neuroids, -1)), axis=1, keepdims=True)
|
104
|
+
|
105
|
+
responses_family = responses.mean(axis=3)
|
106
|
+
|
107
|
+
texture_modulation_index = np.zeros((n_neuroids, 1))
|
108
|
+
texture_selectivity = np.zeros((n_neuroids, 1))
|
109
|
+
noise_selectivity = np.zeros((n_neuroids, 1))
|
110
|
+
texture_sparseness = np.zeros((n_neuroids, 1))
|
111
|
+
noise_sparseness = np.zeros((n_neuroids, 1))
|
112
|
+
variance_ratio = np.zeros((n_neuroids, 1))
|
113
|
+
sample_variance = np.zeros((n_neuroids, 1))
|
114
|
+
family_variance = np.zeros((n_neuroids, 1))
|
115
|
+
|
116
|
+
for neur in range(n_neuroids):
|
117
|
+
texture_modulation_index[neur] = calc_texture_modulation(responses_family[neur])[0]
|
118
|
+
texture_selectivity[neur] = calc_sparseness(responses_family[neur, 1])
|
119
|
+
noise_selectivity[neur] = calc_sparseness(responses_family[neur, 0])
|
120
|
+
texture_sparseness[neur] = calc_sparseness(responses[neur, 1])
|
121
|
+
noise_sparseness[neur] = calc_sparseness(responses[neur, 0])
|
122
|
+
variance_ratio[neur], sample_variance[neur], family_variance[neur] = \
|
123
|
+
calc_variance_ratio(responses_spikes[neur, 1])
|
124
|
+
|
125
|
+
absolute_texture_modulation_index = np.abs(texture_modulation_index)
|
126
|
+
|
127
|
+
properties_data = np.concatenate((texture_modulation_index, absolute_texture_modulation_index, texture_selectivity,
|
128
|
+
noise_selectivity, texture_sparseness, noise_sparseness, variance_ratio,
|
129
|
+
sample_variance, family_variance, max_texture, max_noise), axis=1)
|
130
|
+
|
131
|
+
good_neuroids = max_response > RESPONSE_THRESHOLD
|
132
|
+
properties_data = properties_data[np.argwhere(good_neuroids)[:, 0], :]
|
133
|
+
|
134
|
+
properties_data = DataAssembly(properties_data, coords={'neuroid_id': ('neuroid', range(properties_data.shape[0])),
|
135
|
+
'region': ('neuroid', ['V1'] * properties_data.shape[0]),
|
136
|
+
'neuronal_property': PROPERTY_NAMES},
|
137
|
+
dims=['neuroid', 'neuronal_property'])
|
138
|
+
return properties_data
|
@@ -0,0 +1,167 @@
|
|
1
|
+
import numpy as np
|
2
|
+
from result_caching import store
|
3
|
+
|
4
|
+
from brainio.assemblies import DataAssembly
|
5
|
+
from brainscore_vision import load_dataset, load_metric
|
6
|
+
from brainscore_vision.benchmark_helpers.properties_common import PropertiesBenchmark, _assert_grating_activations
|
7
|
+
from brainscore_vision.benchmark_helpers.properties_common import calc_circular_variance, calc_bandwidth, \
|
8
|
+
calc_orthogonal_preferred_ratio
|
9
|
+
from brainscore_vision.metrics.distribution_similarity import NeuronalPropertyCeiling
|
10
|
+
|
11
|
+
ASSEMBLY_NAME = 'Ringach2002'
|
12
|
+
REGION = 'V1'
|
13
|
+
TIMEBINS = [(70, 170)]
|
14
|
+
PARENT_ORIENTATION = 'V1-orientation'
|
15
|
+
PARENT_MAGNITUDE = 'V1-response_magnitude'
|
16
|
+
PARENT_SELECTIVITY = 'V1-response_selectivity'
|
17
|
+
|
18
|
+
PROPERTY_NAMES = ['baseline', 'max_dc', 'min_dc', 'max_ac', 'modulation_ratio', 'circular_variance', 'bandwidth',
|
19
|
+
'orthogonal_preferred_ratio', 'orientation_selective', 'circular_variance_bandwidth_ratio',
|
20
|
+
'orthogonal_preferred_ratio_circular_variance_difference',
|
21
|
+
'orthogonal_preferred_ratio_bandwidth_ratio']
|
22
|
+
|
23
|
+
BIBTEX = """@article{Ringach2002,
|
24
|
+
author = {Ringach, Dario L and Shapley, Robert M and Hawken, Michael J},
|
25
|
+
doi = {20026567},
|
26
|
+
issn = {1529-2401},
|
27
|
+
journal = {The Journal of Neuroscience},
|
28
|
+
number = {13},
|
29
|
+
pages = {5639--5651},
|
30
|
+
pmid = {12097515},
|
31
|
+
title = {{Orientation selectivity in macaque V1: diversity and laminar dependence.}},
|
32
|
+
volume = {22},
|
33
|
+
year = {2002}
|
34
|
+
}"""
|
35
|
+
|
36
|
+
RESPONSE_THRESHOLD = 5
|
37
|
+
|
38
|
+
|
39
|
+
def _MarquesRingach2002V1Property(property_name, parent, property_identifier=None):
|
40
|
+
assembly = load_dataset(ASSEMBLY_NAME)
|
41
|
+
similarity_metric = load_metric('ks_similarity', property_name=property_name)
|
42
|
+
ceil_func = NeuronalPropertyCeiling(similarity_metric)
|
43
|
+
|
44
|
+
if not property_identifier:
|
45
|
+
property_identifier = property_name
|
46
|
+
return PropertiesBenchmark(identifier=f'Marques2020_Ringach2002-{property_identifier}', assembly=assembly,
|
47
|
+
neuronal_property=ringach2002_properties, similarity_metric=similarity_metric,
|
48
|
+
timebins=TIMEBINS,
|
49
|
+
parent=parent, ceiling_func=ceil_func, bibtex=BIBTEX, version=1)
|
50
|
+
|
51
|
+
|
52
|
+
def MarquesRingach2002V1CircularVariance():
|
53
|
+
property_name = 'circular_variance'
|
54
|
+
parent = PARENT_ORIENTATION
|
55
|
+
return _MarquesRingach2002V1Property(property_name=property_name, parent=parent)
|
56
|
+
|
57
|
+
|
58
|
+
def MarquesRingach2002V1Bandwidth():
|
59
|
+
property_name = 'bandwidth'
|
60
|
+
property_identifier = 'or_bandwidth'
|
61
|
+
parent = PARENT_ORIENTATION
|
62
|
+
return _MarquesRingach2002V1Property(property_name=property_name, parent=parent, property_identifier=property_identifier)
|
63
|
+
|
64
|
+
|
65
|
+
def MarquesRingach2002V1OrthogonalPreferredRatio():
|
66
|
+
property_name = 'orthogonal_preferred_ratio'
|
67
|
+
property_identifier = 'orth_pref_ratio'
|
68
|
+
parent = PARENT_ORIENTATION
|
69
|
+
return _MarquesRingach2002V1Property(property_name=property_name, parent=parent, property_identifier=property_identifier)
|
70
|
+
|
71
|
+
|
72
|
+
def MarquesRingach2002V1OrientationSelective():
|
73
|
+
property_name = 'orientation_selective'
|
74
|
+
property_identifier = 'or_selective'
|
75
|
+
parent = PARENT_ORIENTATION
|
76
|
+
return _MarquesRingach2002V1Property(property_name=property_name, parent=parent, property_identifier=property_identifier)
|
77
|
+
|
78
|
+
|
79
|
+
def MarquesRingach2002V1CircularVarianceBandwidthRatio():
|
80
|
+
property_name = 'circular_variance_bandwidth_ratio'
|
81
|
+
property_identifier = 'cv_bandwidth_ratio'
|
82
|
+
parent = PARENT_ORIENTATION
|
83
|
+
return _MarquesRingach2002V1Property(property_name=property_name, parent=parent, property_identifier=property_identifier)
|
84
|
+
|
85
|
+
|
86
|
+
def MarquesRingach2002V1OrthogonalPrefferredRatioCircularVarianceDifference():
|
87
|
+
property_name = 'orthogonal_preferred_ratio_circular_variance_difference'
|
88
|
+
property_identifier = 'opr_cv_diff'
|
89
|
+
parent = PARENT_ORIENTATION
|
90
|
+
return _MarquesRingach2002V1Property(property_name=property_name, parent=parent, property_identifier=property_identifier)
|
91
|
+
|
92
|
+
|
93
|
+
def MarquesRingach2002V1MaxDC():
|
94
|
+
property_name = 'max_dc'
|
95
|
+
parent = PARENT_MAGNITUDE
|
96
|
+
return _MarquesRingach2002V1Property(property_name=property_name, parent=parent)
|
97
|
+
|
98
|
+
|
99
|
+
def MarquesRingach2002V1ModulationRatio():
|
100
|
+
property_name = 'modulation_ratio'
|
101
|
+
parent = PARENT_SELECTIVITY
|
102
|
+
return _MarquesRingach2002V1Property(property_name=property_name, parent=parent)
|
103
|
+
|
104
|
+
|
105
|
+
@store(identifier_ignore=['responses', 'baseline'])
|
106
|
+
def ringach2002_properties(model_identifier, responses, baseline):
|
107
|
+
_assert_grating_activations(responses)
|
108
|
+
spatial_frequency = np.array(sorted(set(responses.spatial_frequency.values)))
|
109
|
+
orientation = np.array(sorted(set(responses.orientation.values)))
|
110
|
+
phase = np.array(sorted(set(responses.phase.values)))
|
111
|
+
nStim = responses.values.shape[1]
|
112
|
+
n_cycles = nStim // (len(phase) * len(orientation) * len(spatial_frequency))
|
113
|
+
|
114
|
+
responses = responses.values
|
115
|
+
baseline = baseline.values
|
116
|
+
assert responses.shape[0] == baseline.shape[0]
|
117
|
+
n_neuroids = responses.shape[0]
|
118
|
+
|
119
|
+
responses = responses.reshape((n_neuroids, n_cycles, len(spatial_frequency), len(orientation), len(phase)))
|
120
|
+
responses_dc = responses.mean(axis=4)
|
121
|
+
responses_ac = np.absolute(np.fft.fft(responses)) / len(phase)
|
122
|
+
responses_ac = responses_ac[:, :, :, :, 1] * 2
|
123
|
+
del responses
|
124
|
+
|
125
|
+
max_dc = np.zeros((n_neuroids, 1))
|
126
|
+
max_ac = np.zeros((n_neuroids, 1))
|
127
|
+
min_dc = np.zeros((n_neuroids, 1))
|
128
|
+
circular_variance = np.zeros((n_neuroids, 1))
|
129
|
+
bandwidth = np.zeros((n_neuroids, 1))
|
130
|
+
orthogonal_preferred_ratio = np.zeros((n_neuroids, 1))
|
131
|
+
orientation_selective = np.ones((n_neuroids, 1))
|
132
|
+
|
133
|
+
for neur in range(n_neuroids):
|
134
|
+
pref_cycle, pref_spatial_frequency, pref_orientation = np.unravel_index(np.argmax(responses_dc[neur]),
|
135
|
+
(n_cycles, len(spatial_frequency),
|
136
|
+
len(orientation)))
|
137
|
+
|
138
|
+
max_dc[neur] = responses_dc[neur, pref_cycle, pref_spatial_frequency, pref_orientation]
|
139
|
+
max_ac[neur] = responses_ac[neur, pref_cycle, pref_spatial_frequency, pref_orientation]
|
140
|
+
|
141
|
+
orientation_curve = responses_dc[neur, pref_cycle, pref_spatial_frequency, :]
|
142
|
+
min_dc[neur] = orientation_curve.min()
|
143
|
+
|
144
|
+
circular_variance[neur] = calc_circular_variance(orientation_curve, orientation)
|
145
|
+
bandwidth[neur] = \
|
146
|
+
calc_bandwidth(orientation_curve, orientation, filt_type='hanning', thrsh=0.707, mode='half')[0]
|
147
|
+
orthogonal_preferred_ratio[neur] = calc_orthogonal_preferred_ratio(orientation_curve, orientation)
|
148
|
+
|
149
|
+
orientation_selective[np.isnan(bandwidth)] = 0
|
150
|
+
modulation_ratio = max_ac / max_dc
|
151
|
+
circular_variance_bandwidth_ratio = circular_variance / bandwidth
|
152
|
+
orthogonal_preferred_ratio_circular_variance_difference = orthogonal_preferred_ratio - circular_variance
|
153
|
+
orthogonal_preferred_ratio_bandwidth_ratio = orthogonal_preferred_ratio / bandwidth
|
154
|
+
|
155
|
+
properties_data = np.concatenate(
|
156
|
+
(baseline, max_dc, min_dc, max_ac, modulation_ratio, circular_variance, bandwidth, orthogonal_preferred_ratio,
|
157
|
+
orientation_selective, circular_variance_bandwidth_ratio,
|
158
|
+
orthogonal_preferred_ratio_circular_variance_difference, orthogonal_preferred_ratio_bandwidth_ratio), axis=1)
|
159
|
+
|
160
|
+
good_neuroids = max_dc > baseline + RESPONSE_THRESHOLD
|
161
|
+
properties_data = properties_data[np.argwhere(good_neuroids)[:, 0], :]
|
162
|
+
|
163
|
+
properties_data = DataAssembly(properties_data, coords={'neuroid_id': ('neuroid', range(properties_data.shape[0])),
|
164
|
+
'region': ('neuroid', ['V1'] * properties_data.shape[0]),
|
165
|
+
'neuronal_property': PROPERTY_NAMES},
|
166
|
+
dims=['neuroid', 'neuronal_property'])
|
167
|
+
return properties_data
|
@@ -0,0 +1,100 @@
|
|
1
|
+
import numpy as np
|
2
|
+
from result_caching import store
|
3
|
+
|
4
|
+
from brainio.assemblies import DataAssembly
|
5
|
+
from brainscore_vision import load_metric, load_dataset
|
6
|
+
from brainscore_vision.benchmark_helpers.properties_common import PropertiesBenchmark, _assert_grating_activations
|
7
|
+
from brainscore_vision.benchmark_helpers.properties_common import calc_spatial_frequency_tuning
|
8
|
+
from brainscore_vision.metrics.distribution_similarity import NeuronalPropertyCeiling
|
9
|
+
|
10
|
+
ASSEMBLY_NAME = 'Schiller1976c'
|
11
|
+
REGION = 'V1'
|
12
|
+
TIMEBINS = [(70, 170)]
|
13
|
+
PARENT = 'V1-spatial_frequency'
|
14
|
+
|
15
|
+
PROPERTY_NAMES = ['spatial_frequency_selective', 'spatial_frequency_bandwidth']
|
16
|
+
|
17
|
+
BIBTEX = """@article{Schiller1976,
|
18
|
+
author = {Schiller, P. H. and Finlay, B. L. and Volman, S. F.},
|
19
|
+
doi = {10.1152/jn.1976.39.6.1352},
|
20
|
+
issn = {0022-3077},
|
21
|
+
journal = {Journal of neurophysiology},
|
22
|
+
number = {6},
|
23
|
+
pages = {1334--1351},
|
24
|
+
pmid = {825624},
|
25
|
+
title = {{Quantitative studies of single-cell properties in monkey striate cortex. III. Spatial Frequency}},
|
26
|
+
url = {http://www.ncbi.nlm.nih.gov/pubmed/825624},
|
27
|
+
volume = {39},
|
28
|
+
year = {1976}
|
29
|
+
}
|
30
|
+
"""
|
31
|
+
|
32
|
+
RESPONSE_THRESHOLD = 5
|
33
|
+
|
34
|
+
|
35
|
+
def _MarquesSchiller1976V1Property(property_name):
|
36
|
+
assembly = load_dataset(ASSEMBLY_NAME)
|
37
|
+
similarity_metric = load_metric('ks_similarity', property_name=property_name)
|
38
|
+
ceil_func = NeuronalPropertyCeiling(similarity_metric)
|
39
|
+
parent = PARENT
|
40
|
+
property_identifier = property_name.replace("spatial_frequency", "sf")
|
41
|
+
return PropertiesBenchmark(identifier=f'Marques2020_Schiller1976-{property_identifier}', assembly=assembly,
|
42
|
+
neuronal_property=schiller1976_properties, similarity_metric=similarity_metric,
|
43
|
+
timebins=TIMEBINS,
|
44
|
+
parent=parent, ceiling_func=ceil_func, bibtex=BIBTEX, version=1)
|
45
|
+
|
46
|
+
|
47
|
+
def MarquesSchiller1976V1SpatialFrequencySelective():
|
48
|
+
property_name = 'spatial_frequency_selective'
|
49
|
+
return _MarquesSchiller1976V1Property(property_name=property_name)
|
50
|
+
|
51
|
+
|
52
|
+
def MarquesSchiller1976V1SpatialFrequencyBandwidth():
|
53
|
+
property_name = 'spatial_frequency_bandwidth'
|
54
|
+
return _MarquesSchiller1976V1Property(property_name=property_name)
|
55
|
+
|
56
|
+
|
57
|
+
@store(identifier_ignore=['responses', 'baseline'])
|
58
|
+
def schiller1976_properties(model_identifier, responses, baseline):
|
59
|
+
_assert_grating_activations(responses)
|
60
|
+
radius = np.array(sorted(set(responses.radius.values)))
|
61
|
+
spatial_frequency = np.array(sorted(set(responses.spatial_frequency.values)))
|
62
|
+
orientation = np.array(sorted(set(responses.orientation.values)))
|
63
|
+
phase = np.array(sorted(set(responses.phase.values)))
|
64
|
+
|
65
|
+
responses = responses.values
|
66
|
+
baseline = baseline.values
|
67
|
+
assert responses.shape[0] == baseline.shape[0]
|
68
|
+
n_neuroids = responses.shape[0]
|
69
|
+
|
70
|
+
responses = responses.reshape((n_neuroids, len(radius), len(spatial_frequency), len(orientation), len(phase)))
|
71
|
+
responses = responses.mean(axis=4)
|
72
|
+
|
73
|
+
max_response = responses.reshape((n_neuroids, -1)).max(axis=1, keepdims=True)
|
74
|
+
|
75
|
+
spatial_frequency_bandwidth = np.zeros((n_neuroids, 1))
|
76
|
+
spatial_frequency_selective = np.ones((n_neuroids, 1))
|
77
|
+
|
78
|
+
for neur in range(n_neuroids):
|
79
|
+
pref_radius, pref_spatial_frequency, pref_orientation = \
|
80
|
+
np.unravel_index(np.argmax(responses[neur, :, :, :]),
|
81
|
+
(len(radius), len(spatial_frequency), len(orientation)))
|
82
|
+
|
83
|
+
spatial_frequency_curve = responses[neur, pref_radius, :, pref_orientation]
|
84
|
+
|
85
|
+
spatial_frequency_bandwidth[neur] = \
|
86
|
+
calc_spatial_frequency_tuning(spatial_frequency_curve, spatial_frequency, thrsh=0.707, filt_type='smooth',
|
87
|
+
mode='ratio')[0]
|
88
|
+
|
89
|
+
spatial_frequency_selective[np.isnan(spatial_frequency_bandwidth)] = 0
|
90
|
+
|
91
|
+
properties_data = np.concatenate((spatial_frequency_selective, spatial_frequency_bandwidth), axis=1)
|
92
|
+
|
93
|
+
good_neuroids = max_response > baseline + RESPONSE_THRESHOLD
|
94
|
+
properties_data = properties_data[np.argwhere(good_neuroids)[:, 0], :]
|
95
|
+
|
96
|
+
properties_data = DataAssembly(properties_data, coords={'neuroid_id': ('neuroid', range(properties_data.shape[0])),
|
97
|
+
'region': ('neuroid', ['V1'] * properties_data.shape[0]),
|
98
|
+
'neuronal_property': PROPERTY_NAMES},
|
99
|
+
dims=['neuroid', 'neuronal_property'])
|
100
|
+
return properties_data
|
@@ -0,0 +1,135 @@
|
|
1
|
+
import pytest
|
2
|
+
from pytest import approx
|
3
|
+
|
4
|
+
from brainscore_vision import benchmark_registry
|
5
|
+
from brainscore_vision.benchmark_helpers.test_helper import StandardizedTests, PrecomputedTests, NumberOfTrialsTests
|
6
|
+
|
7
|
+
standardized_tests = StandardizedTests()
|
8
|
+
precomputed_test = PrecomputedTests()
|
9
|
+
num_trials_test = NumberOfTrialsTests()
|
10
|
+
|
11
|
+
|
12
|
+
@pytest.mark.parametrize('benchmark', [
|
13
|
+
'Marques2020_Cavanaugh2002-grating_summation_field',
|
14
|
+
'Marques2020_Cavanaugh2002-surround_diameter',
|
15
|
+
'Marques2020_Cavanaugh2002-surround_suppression_index',
|
16
|
+
'Marques2020_DeValois1982-pref_or',
|
17
|
+
'Marques2020_DeValois1982-peak_sf',
|
18
|
+
'Marques2020_FreemanZiemba2013-texture_sparseness',
|
19
|
+
'Marques2020_FreemanZiemba2013-texture_selectivity',
|
20
|
+
'Marques2020_FreemanZiemba2013-texture_variance_ratio',
|
21
|
+
'Marques2020_FreemanZiemba2013-texture_modulation_index',
|
22
|
+
'Marques2020_FreemanZiemba2013-abs_texture_modulation_index',
|
23
|
+
'Marques2020_FreemanZiemba2013-max_noise',
|
24
|
+
'Marques2020_FreemanZiemba2013-max_texture',
|
25
|
+
'Marques2020_Ringach2002-or_bandwidth',
|
26
|
+
'Marques2020_Ringach2002-or_selective',
|
27
|
+
'Marques2020_Ringach2002-circular_variance',
|
28
|
+
'Marques2020_Ringach2002-orth_pref_ratio',
|
29
|
+
'Marques2020_Ringach2002-cv_bandwidth_ratio',
|
30
|
+
'Marques2020_Ringach2002-opr_cv_diff',
|
31
|
+
'Marques2020_Ringach2002-modulation_ratio',
|
32
|
+
'Marques2020_Ringach2002-max_dc',
|
33
|
+
'Marques2020_Schiller1976-sf_bandwidth',
|
34
|
+
'Marques2020_Schiller1976-sf_selective',
|
35
|
+
])
|
36
|
+
def test_benchmark_registry(benchmark):
|
37
|
+
assert benchmark in benchmark_registry
|
38
|
+
|
39
|
+
|
40
|
+
@pytest.mark.private_access
|
41
|
+
@pytest.mark.parametrize('benchmark, expected', [
|
42
|
+
('Marques2020_Cavanaugh2002-grating_summation_field', approx(0.956, abs=.005)),
|
43
|
+
('Marques2020_Cavanaugh2002-surround_diameter', approx(0.955, abs=.005)),
|
44
|
+
('Marques2020_Cavanaugh2002-surround_suppression_index', approx(0.958, abs=.005)),
|
45
|
+
('Marques2020_DeValois1982-pref_or', approx(0.962, abs=.005)),
|
46
|
+
('Marques2020_DeValois1982-peak_sf', approx(0.967, abs=.005)),
|
47
|
+
('Marques2020_FreemanZiemba2013-texture_modulation_index', approx(0.948, abs=.005)),
|
48
|
+
('Marques2020_FreemanZiemba2013-abs_texture_modulation_index', approx(0.958, abs=.005)),
|
49
|
+
('Marques2020_FreemanZiemba2013-texture_selectivity', approx(0.940, abs=.005)),
|
50
|
+
('Marques2020_FreemanZiemba2013-texture_sparseness', approx(0.935, abs=.005)),
|
51
|
+
('Marques2020_FreemanZiemba2013-texture_variance_ratio', approx(0.939, abs=.005)),
|
52
|
+
('Marques2020_FreemanZiemba2013-max_texture', approx(0.946, abs=.005)),
|
53
|
+
('Marques2020_FreemanZiemba2013-max_noise', approx(0.945, abs=.005)),
|
54
|
+
('Marques2020_Ringach2002-circular_variance', approx(0.959, abs=.005)),
|
55
|
+
('Marques2020_Ringach2002-or_bandwidth', approx(0.964, abs=.005)),
|
56
|
+
('Marques2020_Ringach2002-orth_pref_ratio', approx(0.962, abs=.005)),
|
57
|
+
('Marques2020_Ringach2002-or_selective', approx(0.994, abs=.005)),
|
58
|
+
('Marques2020_Ringach2002-cv_bandwidth_ratio', approx(0.968, abs=.005)),
|
59
|
+
('Marques2020_Ringach2002-opr_cv_diff', approx(0.967, abs=.005)),
|
60
|
+
('Marques2020_Ringach2002-modulation_ratio', approx(0.959, abs=.005)),
|
61
|
+
('Marques2020_Ringach2002-max_dc', approx(0.968, abs=.005)),
|
62
|
+
('Marques2020_Schiller1976-sf_selective', approx(0.963, abs=.005)),
|
63
|
+
('Marques2020_Schiller1976-sf_bandwidth', approx(0.933, abs=.005)),
|
64
|
+
])
|
65
|
+
def test_ceilings(benchmark, expected):
|
66
|
+
standardized_tests.ceilings_test(benchmark, expected)
|
67
|
+
|
68
|
+
|
69
|
+
@pytest.mark.private_access
|
70
|
+
@pytest.mark.memory_intense
|
71
|
+
@pytest.mark.slow
|
72
|
+
@pytest.mark.parametrize('benchmark, expected', [
|
73
|
+
('Marques2020_Cavanaugh2002-grating_summation_field', approx(.599, abs=.01)),
|
74
|
+
('Marques2020_Cavanaugh2002-surround_diameter', approx(.367, abs=.01)),
|
75
|
+
('Marques2020_Cavanaugh2002-surround_suppression_index', approx(.365, abs=.01)),
|
76
|
+
('Marques2020_DeValois1982-pref_or', approx(.895, abs=.01)),
|
77
|
+
('Marques2020_DeValois1982-peak_sf', approx(.775, abs=.01)),
|
78
|
+
('Marques2020_FreemanZiemba2013-texture_modulation_index', approx(.636, abs=.01)),
|
79
|
+
('Marques2020_FreemanZiemba2013-abs_texture_modulation_index', approx(.861, abs=.01)),
|
80
|
+
('Marques2020_FreemanZiemba2013-texture_selectivity', approx(.646, abs=.01)),
|
81
|
+
('Marques2020_FreemanZiemba2013-texture_sparseness', approx(.508, abs=.01)),
|
82
|
+
('Marques2020_FreemanZiemba2013-texture_variance_ratio', approx(.827, abs=.01)),
|
83
|
+
('Marques2020_FreemanZiemba2013-max_texture', approx(.823, abs=.01)),
|
84
|
+
('Marques2020_FreemanZiemba2013-max_noise', approx(.684, abs=.01)),
|
85
|
+
('Marques2020_Ringach2002-circular_variance', approx(.830, abs=.01)),
|
86
|
+
('Marques2020_Ringach2002-or_bandwidth', approx(.844, abs=.01)),
|
87
|
+
('Marques2020_Ringach2002-orth_pref_ratio', approx(.876, abs=.01)),
|
88
|
+
('Marques2020_Ringach2002-or_selective', approx(.895, abs=.01)),
|
89
|
+
('Marques2020_Ringach2002-cv_bandwidth_ratio', approx(.841, abs=.01)),
|
90
|
+
('Marques2020_Ringach2002-opr_cv_diff', approx(.909, abs=.01)),
|
91
|
+
('Marques2020_Ringach2002-modulation_ratio', approx(.371, abs=.01)),
|
92
|
+
('Marques2020_Ringach2002-max_dc', approx(.904, abs=.01)),
|
93
|
+
('Marques2020_Schiller1976-sf_selective', approx(.808, abs=.01)),
|
94
|
+
('Marques2020_Schiller1976-sf_bandwidth', approx(.869, abs=.01)),
|
95
|
+
])
|
96
|
+
def test_Marques2020(benchmark, expected):
|
97
|
+
precomputed_test.run_test_properties(
|
98
|
+
benchmark=benchmark,
|
99
|
+
files={'Marques2020_blank': 'alexnet-dicarlo.Marques2020_blank.nc',
|
100
|
+
'Marques2020_receptive_field': 'alexnet-dicarlo.Marques2020_receptive_field.nc',
|
101
|
+
'Marques2020_orientation': 'alexnet-dicarlo.Marques2020_orientation.nc',
|
102
|
+
'Marques2020_spatial_frequency': 'alexnet-dicarlo.Marques2020_spatial_frequency.nc',
|
103
|
+
'Marques2020_size': 'alexnet-dicarlo.Marques2020_size.nc',
|
104
|
+
'FreemanZiemba2013_properties': 'alexnet-movshon.FreemanZiemba2013_properties.nc',
|
105
|
+
},
|
106
|
+
expected=expected)
|
107
|
+
|
108
|
+
|
109
|
+
@pytest.mark.private_access
|
110
|
+
@pytest.mark.parametrize('benchmark_identifier', [
|
111
|
+
'Marques2020_Cavanaugh2002-grating_summation_field',
|
112
|
+
'Marques2020_Cavanaugh2002-surround_diameter',
|
113
|
+
'Marques2020_Cavanaugh2002-surround_suppression_index',
|
114
|
+
'Marques2020_DeValois1982-pref_or',
|
115
|
+
'Marques2020_DeValois1982-peak_sf',
|
116
|
+
'Marques2020_FreemanZiemba2013-texture_sparseness',
|
117
|
+
'Marques2020_FreemanZiemba2013-texture_selectivity',
|
118
|
+
'Marques2020_FreemanZiemba2013-texture_variance_ratio',
|
119
|
+
'Marques2020_FreemanZiemba2013-texture_modulation_index',
|
120
|
+
'Marques2020_FreemanZiemba2013-abs_texture_modulation_index',
|
121
|
+
'Marques2020_FreemanZiemba2013-max_noise',
|
122
|
+
'Marques2020_FreemanZiemba2013-max_texture',
|
123
|
+
'Marques2020_Ringach2002-or_bandwidth',
|
124
|
+
'Marques2020_Ringach2002-or_selective',
|
125
|
+
'Marques2020_Ringach2002-circular_variance',
|
126
|
+
'Marques2020_Ringach2002-orth_pref_ratio',
|
127
|
+
'Marques2020_Ringach2002-cv_bandwidth_ratio',
|
128
|
+
'Marques2020_Ringach2002-opr_cv_diff',
|
129
|
+
'Marques2020_Ringach2002-modulation_ratio',
|
130
|
+
'Marques2020_Ringach2002-max_dc',
|
131
|
+
'Marques2020_Schiller1976-sf_bandwidth',
|
132
|
+
'Marques2020_Schiller1976-sf_selective',
|
133
|
+
])
|
134
|
+
def test_repetitions(benchmark_identifier):
|
135
|
+
num_trials_test.repetitions_test(benchmark_identifier)
|
@@ -0,0 +1,52 @@
|
|
1
|
+
import logging
|
2
|
+
|
3
|
+
from brainscore_core import Score
|
4
|
+
from brainscore_vision import load_stimulus_set, load_metric
|
5
|
+
from brainscore_vision.benchmarks import BenchmarkBase
|
6
|
+
from brainscore_vision.model_interface import BrainModel
|
7
|
+
|
8
|
+
BIBTEX = """@inproceedings{DBLP:conf/nips/BarbuMALWGTK19,
|
9
|
+
author = {Andrei Barbu and
|
10
|
+
David Mayo and
|
11
|
+
Julian Alverio and
|
12
|
+
William Luo and
|
13
|
+
Christopher Wang and
|
14
|
+
Dan Gutfreund and
|
15
|
+
Josh Tenenbaum and
|
16
|
+
Boris Katz},
|
17
|
+
title = {ObjectNet: {A} large-scale bias-controlled dataset for pushing the
|
18
|
+
limits of object recognition models},
|
19
|
+
booktitle = {NeurIPS 2019},
|
20
|
+
pages = {9448--9458},
|
21
|
+
year = {2019},
|
22
|
+
url = {https://proceedings.neurips.cc/paper/2019/hash/97af07a14cacba681feacf3012730892-Abstract.html},
|
23
|
+
}"""
|
24
|
+
|
25
|
+
NUMBER_OF_TRIALS = 10
|
26
|
+
|
27
|
+
_logger = logging.getLogger(__name__)
|
28
|
+
|
29
|
+
|
30
|
+
class Objectnet(BenchmarkBase):
|
31
|
+
def __init__(self):
|
32
|
+
self._stimulus_set = load_stimulus_set('BarbuMayo2019')
|
33
|
+
self._similarity_metric = load_metric('accuracy')
|
34
|
+
ceiling = Score(1)
|
35
|
+
|
36
|
+
super(Objectnet, self).__init__(identifier='BarbuMayo2019-top1', version=1,
|
37
|
+
ceiling_func=lambda: ceiling,
|
38
|
+
parent='engineering',
|
39
|
+
bibtex=BIBTEX)
|
40
|
+
|
41
|
+
def __call__(self, candidate):
|
42
|
+
# The proper `fitting_stimuli` to pass to the candidate would be the imagenet training set.
|
43
|
+
# For now, since almost all models in our hands were trained with imagenet, we'll just short-cut this
|
44
|
+
# by telling the candidate to use its pre-trained imagenet weights.
|
45
|
+
candidate.start_task(BrainModel.Task.label, 'imagenet')
|
46
|
+
stimulus_set = self._stimulus_set[list(set(self._stimulus_set.columns) - {'synset'})] # do not show label
|
47
|
+
predictions = candidate.look_at(stimulus_set, number_of_trials=NUMBER_OF_TRIALS)
|
48
|
+
score = self._similarity_metric(
|
49
|
+
predictions.sortby('filename'),
|
50
|
+
self._stimulus_set.sort_values('filename')['synset'].values
|
51
|
+
)
|
52
|
+
return score
|
@@ -0,0 +1,33 @@
|
|
1
|
+
import numpy as np
|
2
|
+
import pytest
|
3
|
+
|
4
|
+
from brainio.assemblies import BehavioralAssembly
|
5
|
+
from brainscore_vision.benchmarks.objectnet import Objectnet
|
6
|
+
from brainscore_vision.model_interface import BrainModel
|
7
|
+
|
8
|
+
|
9
|
+
@pytest.mark.private_access
|
10
|
+
class TestObjectnet:
|
11
|
+
def test_groundtruth(self):
|
12
|
+
benchmark = Objectnet()
|
13
|
+
source = benchmark._stimulus_set
|
14
|
+
|
15
|
+
class GroundTruth(BrainModel):
|
16
|
+
def start_task(self, task, fitting_stimuli):
|
17
|
+
assert task == BrainModel.Task.label
|
18
|
+
assert fitting_stimuli == 'imagenet' # shortcut
|
19
|
+
|
20
|
+
def look_at(self, stimuli, number_of_trials=1):
|
21
|
+
source_image_ids = source['stimulus_id'].values
|
22
|
+
stimuli_image_ids = stimuli['stimulus_id'].values
|
23
|
+
sorted_x = source_image_ids[np.argsort(source_image_ids)]
|
24
|
+
sorted_index = np.searchsorted(sorted_x, stimuli_image_ids)
|
25
|
+
aligned_source = source.loc[sorted_index]
|
26
|
+
labels = aligned_source['synset'].values
|
27
|
+
return BehavioralAssembly([labels], coords={
|
28
|
+
**{column: ('presentation', aligned_source[column].values) for column in aligned_source.columns},
|
29
|
+
**{'choice': ('choice', ['dummy'])}}, dims=['choice', 'presentation'])
|
30
|
+
|
31
|
+
candidate = GroundTruth()
|
32
|
+
score = benchmark(candidate)
|
33
|
+
assert score == pytest.approx(1)
|
@@ -0,0 +1,10 @@
|
|
1
|
+
from brainscore_vision import benchmark_registry
|
2
|
+
from .benchmarks.benchmark import DicarloRajalingham2018I2n
|
3
|
+
|
4
|
+
benchmark_registry['Rajalingham2018-i2n'] = DicarloRajalingham2018I2n
|
5
|
+
|
6
|
+
|
7
|
+
# public benchmark
|
8
|
+
from.benchmarks.public_benchmark import RajalinghamMatchtosamplePublicBenchmark
|
9
|
+
|
10
|
+
benchmark_registry['Rajalingham2018public-i2n'] = RajalinghamMatchtosamplePublicBenchmark
|