brainscore-vision 2.1__py3-none-any.whl

brainscore_vision/data/deng2009/deng2009imagenet.py

@@ -0,0 +1,33 @@
+import numpy as np
+import os
+import pandas as pd
+from pathlib import Path
+
+from brainio_base.stimuli import StimulusSet
+from brainio_collection.packaging import package_stimulus_set
+
+
+def collect_stimuli(data_dir):
+    stimulus_set = pd.read_csv(data_dir / 'imagenet2012.csv')
+    stimulus_set = StimulusSet(stimulus_set)
+    stimulus_set.image_paths = {row.image_id: row.filepath for row in stimulus_set.itertuples()}
+    stimulus_set['image_path_within_store'] = stimulus_set['filename'].apply(
+        lambda filename: os.path.splitext(filename)[0])
+    stimulus_set = stimulus_set[['image_id', 'label', 'synset', 'image_file_sha1', 'image_path_within_store']]
+    assert len(np.unique(stimulus_set['image_id'])) == len(stimulus_set), "duplicate entries"
+    return stimulus_set
+
+
+def main():
+    data_dir = Path('/braintree/home/msch/brain-score/brainscore/benchmarks')
+    assert os.path.isdir(data_dir)
+
+    stimuli = collect_stimuli(data_dir)
+    stimuli.identifier = 'imagenet_val'
+
+    print('Packaging stimuli')
+    package_stimulus_set(stimuli, stimulus_set_identifier=stimuli.identifier, bucket_name='brainio.contrib')
+
+
+if __name__ == '__main__':
+    main()

brainscore_vision/data/deng2009/test.py

@@ -0,0 +1,9 @@
+import pytest
+from brainscore_vision import load_stimulus_set
+
+
+@pytest.mark.private_access
+def test_num_elements():
+    stimulus_set = load_stimulus_set('imagenet_val')
+    assert len(stimulus_set) == 50_000
+    assert len(set(stimulus_set['label'])) == 1_000

brainscore_vision/data/ferguson2024/__init__.py

@@ -0,0 +1,401 @@
+from brainio.assemblies import BehavioralAssembly
+from brainscore_vision import data_registry, stimulus_set_registry, load_stimulus_set
+from brainscore_vision.data_helpers.s3 import load_assembly_from_s3, load_stimulus_set_from_s3
+
+BIBTEX = """TBD"""
+
+# circle_line:
+stimulus_set_registry['Ferguson2024_circle_line'] = lambda: load_stimulus_set_from_s3(
+    identifier='Ferguson2024_circle_line',
+    bucket="brainio-brainscore",
+    csv_sha1="fc59d23ccfb41b4f98cf02865fc335439d2ad222",
+    zip_sha1="1f0065910b01a1a0e12611fe61252eafb9c534c3",
+    csv_version_id="Dcr1JsAE_bYBQwxYqem9JINE3d_bMLGu",
+    zip_version_id="ss4.fqG7b6NaHkbUXO.iH8f32J07_dmo")
+
+data_registry['Ferguson2024_circle_line'] = lambda: load_assembly_from_s3(
+    identifier='Ferguson2024_circle_line',
+    version_id="2EVlerzlieVA1NbfFiOx2xnhJdVagV4j",
+    sha1="586da7b1c7cb5a60fe72bc148513e3159a27b134",
+    bucket="brainio-brainscore",
+    cls=BehavioralAssembly,
+    stimulus_set_loader=lambda: load_stimulus_set('Ferguson2024_circle_line'),
+)
+
+
+# color:
+stimulus_set_registry['Ferguson2024_color'] = lambda: load_stimulus_set_from_s3(
+    identifier='Ferguson2024_color',
+    bucket="brainio-brainscore",
+    csv_sha1="bc351933e1f21eee9704985c1b8231be6955d816",
+    zip_sha1="6ad04c58de8cc8c28b309572cc41c86470f0c322",
+    csv_version_id="jK6ddF6hF_oWmE5ccm4MyprQwjfeti3.",
+    zip_version_id="Eku5pHE1CNJBrcaRstb8PCYCjOliHQmY")
+
+data_registry['Ferguson2024_color'] = lambda: load_assembly_from_s3(
+    identifier='Ferguson2024_color',
+    version_id="VQo0U9ag8r7r9DREexvSlAD_Z326Iumr",
+    sha1="5b5d67fa3189db9984006910d1954586e6a5a9f3",
+    bucket="brainio-brainscore",
+    cls=BehavioralAssembly,
+    stimulus_set_loader=lambda: load_stimulus_set('Ferguson2024_color'),
+)
+
+
+# convergence:
+stimulus_set_registry['Ferguson2024_convergence'] = lambda: load_stimulus_set_from_s3(
+    identifier='Ferguson2024_convergence',
+    bucket="brainio-brainscore",
+    csv_sha1="bc351933e1f21eee9704985c1b8231be6955d816",
+    zip_sha1="d65bdf6654e08c9107a20028281ab5e38a0be981",
+    csv_version_id="PcRg7fdEJI.Ce3wkd0v6sTC3jSw6xiiq",
+    zip_version_id="4EJaobPVM8STsvMKE.hEcePXcLAjB5VG")
+
+data_registry['Ferguson2024_convergence'] = lambda: load_assembly_from_s3(
+    identifier='Ferguson2024_convergence',
+    version_id="l.nJf3IXYqi5euv5xsqS_ip7Bs0ZpZLX",
+    sha1="5165c4b0da30826b89c2c242826bb79a4417b9a5",
+    bucket="brainio-brainscore",
+    cls=BehavioralAssembly,
+    stimulus_set_loader=lambda: load_stimulus_set('Ferguson2024_convergence'),
+)
+
+
+# eighth:
+stimulus_set_registry['Ferguson2024_eighth'] = lambda: load_stimulus_set_from_s3(
+    identifier='Ferguson2024_eighth',
+    bucket="brainio-brainscore",
+    csv_sha1="bc351933e1f21eee9704985c1b8231be6955d816",
+    zip_sha1="6ee1931b316fd4ccf6eeb16788aa42bb7a36aa41",
+    csv_version_id="fVbTijqnoE61rcXNCopHMhXdrTavCIjS",
+    zip_version_id="ifwG3beZ0ePhQGqbo6S7D9Jj1LPCvwsJ")
+
+data_registry['Ferguson2024_eighth'] = lambda: load_assembly_from_s3(
+    identifier='Ferguson2024_eighth',
+    version_id="VklOC2KrpgLJpD1.kGj6Y5D4kLYSwr3s",
+    sha1="984f9498c42b14cfae6c7272a8707df96fea7ee2",
+    bucket="brainio-brainscore",
+    cls=BehavioralAssembly,
+    stimulus_set_loader=lambda: load_stimulus_set('Ferguson2024_eighth'),
+)
+
+
+# gray_easy:
+stimulus_set_registry['Ferguson2024_gray_easy'] = lambda: load_stimulus_set_from_s3(
+    identifier='Ferguson2024_gray_easy',
+    bucket="brainio-brainscore",
+    csv_sha1="bc351933e1f21eee9704985c1b8231be6955d816",
+    zip_sha1="da76bdebf46fe0eb853ea1b877274b1f26f56dfc",
+    csv_version_id="WCw44X7HWimdn3qLi2D9DSOm5i2bLyrd",
+    zip_version_id="UJp9O0lHnMMPMFmwY29g5v1cHXvF1XpH")
+
+data_registry['Ferguson2024_gray_easy'] = lambda: load_assembly_from_s3(
+    identifier='Ferguson2024_gray_easy',
+    version_id="gaK.0mU6IVHjkI6MG9eE5Hz5Jt7_gxc6",
+    sha1="7b09c2f1e8199e680167cfeb124c28dc68c804ab",
+    bucket="brainio-brainscore",
+    cls=BehavioralAssembly,
+    stimulus_set_loader=lambda: load_stimulus_set('Ferguson2024_gray_easy'),
+)
+
+
+# gray_hard:
+stimulus_set_registry['Ferguson2024_gray_hard'] = lambda: load_stimulus_set_from_s3(
+    identifier='Ferguson2024_gray_hard',
+    bucket="brainio-brainscore",
+    csv_sha1="bc351933e1f21eee9704985c1b8231be6955d816",
+    zip_sha1="93f02c051f4d89fe059345c0af7ba6fc83b65b35",
+    csv_version_id="bxWCJhmQw9RYxQx8qzGSltZCnSY4UTRI",
+    zip_version_id="WLtKonQVU9Og0ZbmVRJKx4Zzxb4INsT8")
+
+data_registry['Ferguson2024_gray_hard'] = lambda: load_assembly_from_s3(
+    identifier='Ferguson2024_gray_hard',
+    version_id="KSpwyfIqK6uovFojNd2_w08lKUJvfOWl",
+    sha1="2fa35d41e73053ece6d1f0120ca4dc9bc4a9d4ae",
+    bucket="brainio-brainscore",
+    cls=BehavioralAssembly,
+    stimulus_set_loader=lambda: load_stimulus_set('Ferguson2024_gray_hard'),
+)
+
+
+# half:
+stimulus_set_registry['Ferguson2024_half'] = lambda: load_stimulus_set_from_s3(
+    identifier='Ferguson2024_half',
+    bucket="brainio-brainscore",
+    csv_sha1="bc351933e1f21eee9704985c1b8231be6955d816",
+    zip_sha1="6461a1d19e031943d42e97e8b557a97d14b18c55",
+    csv_version_id="WGrCxoue4oPYUKz81t30jcScz1dWs5Dv",
+    zip_version_id="9pvmNpTauZECPkemXEfLV_wYA9JZT0Iw")
+
+data_registry['Ferguson2024_half'] = lambda: load_assembly_from_s3(
+    identifier='Ferguson2024_half',
+    version_id="Z2Mpv3qH9foT9qggDIxWVHoEuKb6mC.a",
+    sha1="b65e14c5d62fee715438a613e55fffa5e6f76c40",
+    bucket="brainio-brainscore",
+    cls=BehavioralAssembly,
+    stimulus_set_loader=lambda: load_stimulus_set('Ferguson2024_half'),
+)
+
+
+# juncture:
+stimulus_set_registry['Ferguson2024_juncture'] = lambda: load_stimulus_set_from_s3(
+    identifier='Ferguson2024_juncture',
+    bucket="brainio-brainscore",
+    csv_sha1="bc351933e1f21eee9704985c1b8231be6955d816",
+    zip_sha1="832102f1eaa713fdfd03c512df2b8feea422c61d",
+    csv_version_id="J3wrdsSM9LMlGFoC3ks5ees_t1sKjvKc",
+    zip_version_id="zNu6swQFgclcS8.miCuDBk4AQ4G54KT2")
+
+data_registry['Ferguson2024_juncture'] = lambda: load_assembly_from_s3(
+    identifier='Ferguson2024_juncture',
+    version_id="RstO_IgzeE2UbmHbMw6RN7vV8doFZKBq",
+    sha1="b18148383ef2158aa795b3cff8a8e237e08b5070",
+    bucket="brainio-brainscore",
+    cls=BehavioralAssembly,
+    stimulus_set_loader=lambda: load_stimulus_set('Ferguson2024_juncture'),
+)
+
+
+# lle:
+stimulus_set_registry['Ferguson2024_lle'] = lambda: load_stimulus_set_from_s3(
+    identifier='Ferguson2024_lle',
+    bucket="brainio-brainscore",
+    csv_sha1="bc351933e1f21eee9704985c1b8231be6955d816",
+    zip_sha1="27817e955da9e4747d2aeb8757b7f6492bc7767e",
+    csv_version_id="y3epQUp6h7zH5h8251G8DlYzwtk6VYxW",
+    zip_version_id="RCPB0_kLL0GF3xrR0Nl.c11uAL8yYF8c")
+
+data_registry['Ferguson2024_lle'] = lambda: load_assembly_from_s3(
+    identifier='Ferguson2024_lle',
+    version_id="nXWjKJJyGtX.67m.M03oRw7ysfP76e4e",
+    sha1="08e98305657cd374d9ea103df0fe06783a70344a",
+    bucket="brainio-brainscore",
+    cls=BehavioralAssembly,
+    stimulus_set_loader=lambda: load_stimulus_set('Ferguson2024_lle'),
+)
+
+
+# llh (assuming 'llh' is correct and not a placeholder):
+stimulus_set_registry['Ferguson2024_llh'] = lambda: load_stimulus_set_from_s3(
+    identifier='Ferguson2024_llh',
+    bucket="brainio-brainscore",
+    csv_sha1="bc351933e1f21eee9704985c1b8231be6955d816",
+    zip_sha1="56cdf86ecd0b52349b29b2ab0be89daeed9b0eb6",
+    csv_version_id="n3gooGN6lqWT5c.Qa3.kpUGUwogDtQUT",
+    zip_version_id="3A2EgFZ9Un_uFl43xqXIudDHUHdF7le1")
+
+data_registry['Ferguson2024_llh'] = lambda: load_assembly_from_s3(
+    identifier='Ferguson2024_llh',
+    version_id="prACZ4sm395A5yfJEYDG77MfGMJhXaXv",
+    sha1="864d49c00e777f3d464c6c0c59fee087c1de9037",
+    bucket="brainio-brainscore",
+    cls=BehavioralAssembly,
+    stimulus_set_loader=lambda: load_stimulus_set('Ferguson2024_llh'),
+)
+
+
+# quarter:
+stimulus_set_registry['Ferguson2024_quarter'] = lambda: load_stimulus_set_from_s3(
+    identifier='Ferguson2024_quarter',
+    bucket="brainio-brainscore",
+    csv_sha1="bc351933e1f21eee9704985c1b8231be6955d816",
+    zip_sha1="c16c5ecf1f38af0d02255a78a6c438074ec8d446",
+    csv_version_id="lnk7H5WiGe3oB0i5PMTrXA_q058kZSDz",
+    zip_version_id="frHF3zSr4cCUEs7bVYjjaM3c0WQgwiA9")
+
+data_registry['Ferguson2024_quarter'] = lambda: load_assembly_from_s3(
+    identifier='Ferguson2024_quarter',
+    version_id="_q5R_GoANyjQ8DWQsY.2HBtzW8DoSGpm",
+    sha1="921b3b51208cdd5f163eca288ea83be47a2b482f",
+    bucket="brainio-brainscore",
+    cls=BehavioralAssembly,
+    stimulus_set_loader=lambda: load_stimulus_set('Ferguson2024_quarter'),
+)
+
+
+# round_f:
+stimulus_set_registry['Ferguson2024_round_f'] = lambda: load_stimulus_set_from_s3(
+    identifier='Ferguson2024_round_f',
+    bucket="brainio-brainscore",
+    csv_sha1="bc351933e1f21eee9704985c1b8231be6955d816",
+    zip_sha1="7f456e390cb93805187266d05756beb9cf225e1d",
+    csv_version_id="jheoU.xYIbSk2hFPhCue2MGmXckqMooe",
+    zip_version_id="FtLIcpUQzHA_jPdRl_6iSJoqXZKDCeJn")
+
+data_registry['Ferguson2024_round_f'] = lambda: load_assembly_from_s3(
+    identifier='Ferguson2024_round_f',
+    version_id="0E7lr44ha3rV7xpnWnE1MpDV79seDxCe",
+    sha1="acb19ac865b45199a58609db31d3e885ff272fd4",
+    bucket="brainio-brainscore",
+    cls=BehavioralAssembly,
+    stimulus_set_loader=lambda: load_stimulus_set('Ferguson2024_round_f'),
+)
+
+
+# round_v:
+stimulus_set_registry['Ferguson2024_round_v'] = lambda: load_stimulus_set_from_s3(
+    identifier='Ferguson2024_round_v',
+    bucket="brainio-brainscore",
+    csv_sha1="bc351933e1f21eee9704985c1b8231be6955d816",
+    zip_sha1="cebb84d2363c1539368e3e1b1bfd83305ad9ae13",
+    csv_version_id="6_N_s3Cz_g32jncN0bWoDCh.1pWdKCv2",
+    zip_version_id="r8e50KhAeIc0mKz1qE_xN2z4rMYaNsJ_")
+
+data_registry['Ferguson2024_round_v'] = lambda: load_assembly_from_s3(
+    identifier='Ferguson2024_round_v',
+    version_id="VS.8.ocCdNugRJNU6ha2Wm3K1lK4vK5k",
+    sha1="ce0361c4386dc7b8866d78023044b3009c84aa4b",
+    bucket="brainio-brainscore",
+    cls=BehavioralAssembly,
+    stimulus_set_loader=lambda: load_stimulus_set('Ferguson2024_round_v'),
+)
+
+
+# tilted_line:
+stimulus_set_registry['Ferguson2024_tilted_line'] = lambda: load_stimulus_set_from_s3(
+    identifier='Ferguson2024_tilted_line',
+    bucket="brainio-brainscore",
+    csv_sha1="bc351933e1f21eee9704985c1b8231be6955d816",
+    zip_sha1="bb3d7bcb60ba586c8552266839187a59c2b3138f",
+    csv_version_id="7mcYPI8IYpS9Rz7pLm6QOxBne29.WcWp",
+    zip_version_id="5dvzTilCQkDUHG85qCCQZOhB6ZLpfU5_")
+
+data_registry['Ferguson2024_tilted_line'] = lambda: load_assembly_from_s3(
+    identifier='Ferguson2024_tilted_line',
+    version_id="ae4Dbo9JU_PDwTqKGD1G4DQNrdh2cVE2",
+    sha1="1806034da0c25e8625255eb94dc0a05c7e9cda1f",
+    bucket="brainio-brainscore",
+    cls=BehavioralAssembly,
+    stimulus_set_loader=lambda: load_stimulus_set('Ferguson2024_tilted_line'),
+)
+
+
+'''
+Pretraining Stimuli:
+'''
+
+# circle_line
+stimulus_set_registry['Ferguson2024_circle_line_training_stimuli'] = lambda: load_stimulus_set_from_s3(
+    identifier='Ferguson2024_circle_line_training_stimuli',
+    bucket="brainio-brainscore",
+    csv_sha1="098eb5999e9c4b723abc35ade862d2dc45899230",
+    zip_sha1="ba9088601d8c79ea5ff3d513e1a76b1232491918",
+    csv_version_id="fhYDvXCZNhij.2gnNfbTPlD.yOeiuz9G",
+    zip_version_id="i3lS29oWEn3JMReUaKZerehZKvZqaHq7")
+
+# color
+stimulus_set_registry['Ferguson2024_color_training_stimuli'] = lambda: load_stimulus_set_from_s3(
+    identifier='Ferguson2024_color_training_stimuli',
+    bucket="brainio-brainscore",
+    csv_sha1="098eb5999e9c4b723abc35ade862d2dc45899230",
+    zip_sha1="dcd5044c73e7523afc94f91543adb444a437f911",
+    csv_version_id="hfvHFxmWOQUwq0LnSwrhk8xecaa9XhQW",
+    zip_version_id="uwMSKXr5yRYqVDXA9aoS66BBXtsu2kcx")
+
+# convergence
+stimulus_set_registry['Ferguson2024_convergence_training_stimuli'] = lambda: load_stimulus_set_from_s3(
+    identifier='Ferguson2024_convergence_training_stimuli',
+    bucket="brainio-brainscore",
+    csv_sha1="098eb5999e9c4b723abc35ade862d2dc45899230",
+    zip_sha1="00eb401ddfc075a4bb448ec36b8a1c8f3ec1d6e4",
+    csv_version_id=".ZIqJlEMSgY_U5PeXBU33ifj2KMeMz2e",
+    zip_version_id="2bCuP2jVWc2WIuE9tD6b7TyPkuBbxyn0")
+
+# eighth
+stimulus_set_registry['Ferguson2024_eighth_training_stimuli'] = lambda: load_stimulus_set_from_s3(
+    identifier='Ferguson2024_eighth_training_stimuli',
+    bucket="brainio-brainscore",
+    csv_sha1="098eb5999e9c4b723abc35ade862d2dc45899230",
+    zip_sha1="26edaec1d0dd14343a117340283e091a2245f3aa",
+    csv_version_id="X7gv5Rztd.VmOIr8rmEd7XYBWtsGDJdR",
+    zip_version_id="wVfBxoqcy6YIZnFLLu6rj.8XUXQmOQMg")
+
+# gray_easy
+stimulus_set_registry['Ferguson2024_gray_easy_training_stimuli'] = lambda: load_stimulus_set_from_s3(
+    identifier='Ferguson2024_gray_easy_training_stimuli',
+    bucket="brainio-brainscore",
+    csv_sha1="098eb5999e9c4b723abc35ade862d2dc45899230",
+    zip_sha1="18211af83c680d5c916ec15b57b5b871494d6b28",
+    csv_version_id="j25..m3F2t7j.47YEiOHxTZxiq7ViPxc",
+    zip_version_id="h6fYQQ.DIWqr09rrqZIupCdUzJXFMTG9")
+
+# gray hard
+stimulus_set_registry['Ferguson2024_gray_hard_training_stimuli'] = lambda: load_stimulus_set_from_s3(
+    identifier='Ferguson2024_gray_hard_training_stimuli',
+    bucket="brainio-brainscore",
+    csv_sha1="098eb5999e9c4b723abc35ade862d2dc45899230",
+    zip_sha1="a54d84dbb548706bbfeb239113a1d92205dc3f67",
+    csv_version_id="nNGjK3Mgo2h4WVT0yx_yvJeP1htuWSUl",
+    zip_version_id="MeyqeiOhGSRRLYzG_bLsG.Nj4W2ktRf8")
+
+# half
+stimulus_set_registry['Ferguson2024_half_training_stimuli'] = lambda: load_stimulus_set_from_s3(
+    identifier='Ferguson2024_half_training_stimuli',
+    bucket="brainio-brainscore",
+    csv_sha1="098eb5999e9c4b723abc35ade862d2dc45899230",
+    zip_sha1="0db665619307d5c532a3ccd4311611e5a3830a10",
+    csv_version_id="bAg_H4VtFostaowCDqy2htVL9iBWCENh",
+    zip_version_id="rEsJ7ZopuRTyxSnA97ifpiHtkXqnWvR5")
+
+# juncture
+stimulus_set_registry['Ferguson2024_juncture_training_stimuli'] = lambda: load_stimulus_set_from_s3(
+    identifier='Ferguson2024_juncture_training_stimuli',
+    bucket="brainio-brainscore",
+    csv_sha1="098eb5999e9c4b723abc35ade862d2dc45899230",
+    zip_sha1="eb99fd862bec5e61900d037b6b38abf2a278c9f0",
+    csv_version_id="Uikb_kSDojTsL8LXORmShk_cuW8lFxa.",
+    zip_version_id="wRFpwf_J2kC2WtBUDGiv1Enhrj5Ah5Gh")
+
+# lle
+stimulus_set_registry['Ferguson2024_lle_training_stimuli'] = lambda: load_stimulus_set_from_s3(
+    identifier='Ferguson2024_lle_training_stimuli',
+    bucket="brainio-brainscore",
+    csv_sha1="098eb5999e9c4b723abc35ade862d2dc45899230",
+    zip_sha1="a1d19d0b77f0eb17ae886a1b7ccc649c5e51d84e",
+    csv_version_id="QXbtxFHLywcvLQy2enqL2Lxv9.bMgAwo",
+    zip_version_id="3izgx5jOCHDjH1fy_ncHOL7HxZIYt5nr")
+
+# llh
+stimulus_set_registry['Ferguson2024_llh_training_stimuli'] = lambda: load_stimulus_set_from_s3(
+    identifier='Ferguson2024_llh_training_stimuli',
+    bucket="brainio-brainscore",
+    csv_sha1="098eb5999e9c4b723abc35ade862d2dc45899230",
+    zip_sha1="1550f9f71e6930caa15b96aaf811aa97d48d7267",
+    csv_version_id="M3WlC_zVg5m8rYLyJd1KlKo2wQkf36G7",
+    zip_version_id="brEvqix1vzPM6mX8Jnx7pOgJEHETpOXM")
+
+# quarter
+stimulus_set_registry['Ferguson2024_quarter_training_stimuli'] = lambda: load_stimulus_set_from_s3(
+    identifier='Ferguson2024_quarter_training_stimuli',
+    bucket="brainio-brainscore",
+    csv_sha1="098eb5999e9c4b723abc35ade862d2dc45899230",
+    zip_sha1="22669e4a94718b3cbde3f5b2a493044bc091257e",
+    csv_version_id="lP4fsstG0Jfcnistm2H0AUhmPMHqAfTU",
+    zip_version_id="zpwv2_fwsmHk1TyR9_DYdmNGuLykgGX_")
+
+# round_f
+stimulus_set_registry['Ferguson2024_round_f_training_stimuli'] = lambda: load_stimulus_set_from_s3(
+    identifier='Ferguson2024_round_f_training_stimuli',
+    bucket="brainio-brainscore",
+    csv_sha1="098eb5999e9c4b723abc35ade862d2dc45899230",
+    zip_sha1="e33855c899f78a115cf377a228e07d87baa554b7",
+    csv_version_id="csLNw6RL7nen9TFyH552JSahJkKbnNLE",
+    zip_version_id="7YYhm.tjysTS2e.IhjBx0ovOxWdAVv1M")
+
+# round_v
+stimulus_set_registry['Ferguson2024_round_v_training_stimuli'] = lambda: load_stimulus_set_from_s3(
+    identifier='Ferguson2024_round_v_training_stimuli',
+    bucket="brainio-brainscore",
+    csv_sha1="098eb5999e9c4b723abc35ade862d2dc45899230",
+    zip_sha1="b1555f8a140a12e01a87a2f4e452d5863be43a5b",
+    csv_version_id="QeNeoWjAxMZO4AjmB2SZFC4qEzwf1cBw",
+    zip_version_id="gj32aM8zE_VXh_N9hNI42g1Uo5AxNDJh")
+
+# tilted_line
+stimulus_set_registry['Ferguson2024_tilted_line_training_stimuli'] = lambda: load_stimulus_set_from_s3(
+    identifier='Ferguson2024_tilted_line_training_stimuli',
+    bucket="brainio-brainscore",
+    csv_sha1="098eb5999e9c4b723abc35ade862d2dc45899230",
+    zip_sha1="e92533d8aded07ed90ef25650d0cf07c3a458be7",
+    csv_version_id="l.8gS70OruIDfDU9Oj.DAWw6BQNB.LKc",
+    zip_version_id="cAv1IPQkKX8Jey1gFc4VCwItECIiSlLV")

brainscore_vision/data/ferguson2024/data_packaging/data_packaging.py

@@ -0,0 +1,164 @@
+from brainio.packaging import package_data_assembly
+from pathlib import Path
+from shutil import copy
+from brainio.stimuli import StimulusSet
+from brainio.packaging import package_stimulus_set
+from brainio.assemblies import BehavioralAssembly
+import pandas as pd
+
+DATASETS = ['circle_line', 'color', 'convergence', 'eighth',
+            'gray_easy', 'gray_hard', 'half', 'juncture',
+            'lle', 'llh', 'quarter', 'round_f',
+            'round_v', 'tilted_line']
+
+
+# Packages the stimulus_sets for the Ferguson2024 experiment. There are 14 in all.
+def create_stimulus_set_and_upload(name: str, experiment: str, upload_to_s3=True) -> StimulusSet:
+    """
+
+    Sample image from dataset:
+    first_block_0.png
+
+    1) first_block -> what block the stimuli belong two (which image is target, which is distractor)
+    2) 0 -> a number, 0-23 indicating which variation the image is
+
+    There are 24 images in the first block, and 24 in the second block, so the combined stimulus_set is length 48.
+    The packaged stimuli were structured so that the root folder (tilted_line) had two subfolders, /first_block and /second_block.
+
+    :param name: the name of the experiment, usually Ferguson2024
+    :param experiment: the dataset, i.e. color
+    :param upload_to_s3: True if you want to upload this to BrainIO on S3
+    :return: the Stimulus Set
+    """
+
+    stimuli = []
+    stimulus_paths = {}
+    stimuli_directory = f'{experiment}'
+    combine_block_images(stimuli_directory)
+    for filepath in Path(f"{stimuli_directory}/final").glob('**/*.png'):
+        stimulus_id = filepath.stem
+        parts_list = stimulus_id.split("_")
+        block = parts_list[0]
+        image_number = parts_list[2]
+
+        stimulus_paths[stimulus_id] = filepath
+        stimuli.append({
+            'stimulus_id': stimulus_id,
+            'image_number': image_number,
+            'block': block,
+        })
+
+    stimuli = StimulusSet(stimuli)
+    stimuli.stimulus_paths = stimulus_paths
+    stimuli.name = f'{name}_{experiment}'  # give the StimulusSet an identifier name
+
+    # upload to S3
+    if upload_to_s3:
+        init_data = package_stimulus_set(catalog_name=None, proto_stimulus_set=stimuli,
+                                         stimulus_set_identifier=stimuli.name, bucket_name="brainio-brainscore")
+        print(f"{experiment} stimulus_set\n{init_data}")
+    return stimuli
+
+
+# Packages the assemblies for the Ferguson2024 experiment. There are 14 in all.
+def create_assembly_and_upload(name: str, experiment: str, upload_to_s3=True) -> BehavioralAssembly:
+    """
+    Takes in a sanity-processed csv file, converts to an assembly, and uploads it to BrainIO
+
+    :param name: the name of the experiment, usually Ferguson2024
+    :param experiment: the dataset, i.e. color
+    :param upload_to_s3: True if you want to upload this to BrainIO on S3
+    :return: the assmebly
+    """
+    all_subjects = pd.read_csv(f'csvs/{experiment}_sanity_processed.csv')
+
+    # only look at testing data (no warmup or sanity data):
+    all_subjects = all_subjects[all_subjects["trial_type"] == "normal"]
+    all_subjects = bool_to_int(all_subjects, ['correct', 'target_present'])  # cast bool to int for NetCDF
+
+    # create an ID that is equal to the stimulus_set ID
+    all_subjects['stimulus_id'] = all_subjects['stimulus'].apply(extract_and_concatenate)
+
+    assembly = BehavioralAssembly(all_subjects['correct'],
+                                  coords={
+                                      'stimulus_id': ('presentation', all_subjects['stimulus_id']),
+                                      'stimulus_id_long': ('presentation', all_subjects['stimulus']),
+                                      'participant_id': ('presentation', all_subjects['participant_id']),
+                                      'response_time_ms': ('presentation', all_subjects['response_time_ms']),
+                                      'correct': ('presentation', all_subjects['correct']),
+                                      'target_present': ('presentation', all_subjects['target_present']),
+                                      'distractor_nums': ('presentation', all_subjects['distractor_nums']),
+                                      'block': ('presentation', all_subjects['block']),
+                                      'keypress_response': ('presentation', all_subjects['response']),
+                                      'trial_type': ('presentation', all_subjects['trial_type']),
+                                  },
+                                  dims=['presentation']
+                                  )
+
+    assembly.name = f"{name}_{experiment}"
+
+    # upload assembly to S3
+    if upload_to_s3:
+        init_data = package_data_assembly(None, assembly, assembly_identifier=assembly.name,
+                                          stimulus_set_identifier=f"{name}_{experiment}",
+                                          assembly_class_name="BehavioralAssembly",
+                                          bucket_name="brainio-brainscore")
+        print(f"{experiment} assembly\n{init_data}")
+    return assembly
+
+
+# helper function to take in a folder with the structure outlined in the above file docs, and move them
+# all into one folder
+def combine_block_images(stimuli_directory: str) -> None:
+    """
+
+    :param stimuli_directory: the path where your stimuli are located. This folder has two subfolders, /first_block and
+           /second_block
+    """
+    final_directory_path = Path(stimuli_directory) / 'final'
+    final_directory_path.mkdir(exist_ok=True)
+    subfolders = ['first_block', 'second_block']
+    for subfolder in subfolders:
+        current_folder_path = Path(stimuli_directory) / subfolder
+        if not current_folder_path.exists():
+            continue
+        for filepath in current_folder_path.glob('*.png'):
+            stimulus_id = filepath.stem
+            new_file_name = f"{subfolder}_{stimulus_id}.png"
+            new_file_path = final_directory_path / new_file_name
+            copy(filepath, new_file_path)
+
+
+# helper function to get the stimulus_set stimulus_id from the assembly stimulus:
+def extract_and_concatenate(url):
+    parts = url.split('/')
+    block_part = parts[-3]
+    file_name = parts[-1].replace(".png", "")
+    return f"{block_part}_{file_name}"
+
+
+# Converts boolean values to integers in specified columns of a DataFrame.
+def bool_to_int(df, columns):
+    for column in columns:
+        if column in df.columns:
+            df[column] = df[column].map({'True': 1, 'False': 0, True: 1, False: 0}).fillna(df[column])
+        else:
+            print(f"Column '{column}' not found in DataFrame.")
+    return df
+
+
+# wrapper function to loop over all datasets
+def package_all_stimulus_sets(name):
+    for experiment in DATASETS:
+        create_stimulus_set_and_upload(name, experiment)
+
+
+# wrapper function to loop over all datasets:
+def package_all_assemblies(name):
+    for experiment in DATASETS:
+        create_assembly_and_upload(name, experiment)
+
+
+if __name__ == '__main__':
+    package_all_stimulus_sets(name='Ferguson2024')
+    package_all_assemblies(name='Ferguson2024')

brainscore_vision/data/ferguson2024/data_packaging/fitting_stimuli.py

@@ -0,0 +1,20 @@
+from data_packaging import create_stimulus_set_and_upload, DATASETS
+
+# package and upload all 14 training stimuli sets
+all_stimulus_sets = []
+paths = {}
+for dataset in DATASETS:
+
+    """
+    For the fitting stimuli, each stimulus set will have 1920 images in them:
+    320 images x 3 distractors (low, medium, high) x 2 types (target on distractor, distractor on target)
+    """
+
+    stimulus_set = create_stimulus_set_and_upload("Ferguson2024", f"{dataset}_training_stimuli", upload_to_s3=True)
+    all_stimulus_sets.append(stimulus_set)
+
+
+# label each dataset with the name of the dataset
+for df, name in zip(all_stimulus_sets, DATASETS):
+    df['experiment'] = name
+

brainscore_vision/data/ferguson2024/requirements.txt

@@ -0,0 +1,2 @@
+pathlib
+pandas