/ concept-collection / benchcompress
Sign in
concept-collection / benchcompress
benchcompress / load_real.py
38 lines · 1.9 KBCodeBlameHistory
a81e75aInitial commitJeremy Magland 1import numpy as np
4def load_real_000876(
5 *, num_samples: int, num_channels: int, start_channel: int
6) -> np.ndarray:
7 import lindi
8 # https://neurosift.app/?p=/nwb&url=https://api.dandiarchive.org/api/assets/7e1de06d-d478-40e2-9b64-9dd04eafaa4c/download/&dandisetId=000876&dandisetVersion=draft
9 nwb_url = "https://api.dandiarchive.org/api/assets/7e1de06d-d478-40e2-9b64-9dd04eafaa4c/download/"
10 h5f = lindi.LindiH5pyFile.from_hdf5_file(nwb_url)
11 ds = h5f["/acquisition/ElectricalSeriesAP/data"]
12 assert isinstance(ds, lindi.LindiH5pyDataset)
13 ret = ds[:num_samples, start_channel:start_channel + num_channels]
14 return ret
16def load_real_000409(
17 *, num_samples: int, num_channels: int, start_channel: int
18) -> np.ndarray:
19 import lindi
20 # https://neurosift.app/?p=/nwb&url=https://api.dandiarchive.org/api/assets/c04f6b30-82bf-40e1-9210-34f0bcd8be24/download/&dandisetId=000409&dandisetVersion=draft
21 nwb_url = "https://api.dandiarchive.org/api/assets/c04f6b30-82bf-40e1-9210-34f0bcd8be24/download/"
22 h5f = lindi.LindiH5pyFile.from_hdf5_file(nwb_url)
23 ds = h5f['/acquisition/ElectricalSeriesAp/data']
24 assert isinstance(ds, lindi.LindiH5pyDataset)
25 ret = ds[:num_samples, start_channel:start_channel + num_channels]
26 return ret
28def load_real_001290(
29 *, num_samples: int, num_channels: int, start_channel: int
30):
31 import lindi
32 # https://neurosift.app/?p=/nwb&url=https://api.dandiarchive.org/api/assets/78c99d23-da88-4ecd-9086-c488a126eac5/download/&dandisetId=001290&dandisetVersion=draft
33 nwb_url = "https://api.dandiarchive.org/api/assets/78c99d23-da88-4ecd-9086-c488a126eac5/download/"
34 h5f = lindi.LindiH5pyFile.from_hdf5_file(nwb_url)
35 ds = h5f['/acquisition/ElectricalSeriesAPImec/data']
36 assert isinstance(ds, lindi.LindiH5pyDataset)
37 ret = ds[:num_samples, start_channel:start_channel + num_channels]
38 return ret
moveopenescclose