1import os
2import spikeinterface as si
3import numpy as np
4from s3_utils import download_s3_folder
6s3_base_url = "s3://aind-benchmark-data/ephys-compression/"
8folder_names = [
9 ("aind-np2/612962_2022-04-13_19-18-04_ProbeB", "aind-np2-probeB"),
10 ("aind-np1/625749_2022-08-03_15-15-06_ProbeA", "aind-np1-probeA")
11]
13for folder_name, name0 in folder_names:
14 s3_folder_name = f"{s3_base_url}/{folder_name}"
15 local_folder_name = f"{folder_name}.si"
17 if not os.path.exists(local_folder_name):
18 # make parent directories if needed
19 os.makedirs(os.path.dirname(local_folder_name), exist_ok=True)
20 print(f'Downloading {s3_folder_name} to {local_folder_name}...')
21 download_s3_folder(s3_folder_name, local_folder_name)
23 recording = si.load(
24 local_folder_name
25 )
27 channel_ids = [
28 'CH101',
29 'CH102',
30 'CH103',
31 'CH104',
32 'CH105',
33 'CH106',
34 'CH107',
35 'CH108',
36 'CH109',
37 'CH110'
38 ]
40 fname = f'{name0}-ch101-110.raw.npy'
41 if not os.path.exists(fname):
42 print(f'Writing {fname}...')
43 X = recording.get_traces(channel_ids=channel_ids, start_frame=30000, end_frame=30000 + 30000 * 10)
44 print(f'X.shape = {X.shape}')
45 np.save(fname, X)