spikezoo 0.1.2__py3-none-any.whl → 0.2.1__py3-none-any.whl

Sign up to get free protection for your applications and to get access to all the features.
Files changed (192) hide show
  1. spikezoo/__init__.py +13 -0
  2. spikezoo/archs/__pycache__/__init__.cpython-39.pyc +0 -0
  3. spikezoo/archs/base/__pycache__/nets.cpython-39.pyc +0 -0
  4. spikezoo/archs/base/nets.py +34 -0
  5. spikezoo/archs/bsf/README.md +92 -0
  6. spikezoo/archs/bsf/datasets/datasets.py +328 -0
  7. spikezoo/archs/bsf/datasets/ds_utils.py +64 -0
  8. spikezoo/archs/bsf/main.py +398 -0
  9. spikezoo/archs/bsf/metrics/psnr.py +22 -0
  10. spikezoo/archs/bsf/metrics/ssim.py +54 -0
  11. spikezoo/archs/bsf/models/bsf/__pycache__/align.cpython-39.pyc +0 -0
  12. spikezoo/archs/bsf/models/bsf/__pycache__/bsf.cpython-39.pyc +0 -0
  13. spikezoo/archs/bsf/models/bsf/__pycache__/rep.cpython-39.pyc +0 -0
  14. spikezoo/archs/bsf/models/bsf/align.py +154 -0
  15. spikezoo/archs/bsf/models/bsf/bsf.py +105 -0
  16. spikezoo/archs/bsf/models/bsf/dsft_convert.py +96 -0
  17. spikezoo/archs/bsf/models/bsf/rep.py +44 -0
  18. spikezoo/archs/bsf/models/get_model.py +7 -0
  19. spikezoo/archs/bsf/prepare_data/DSFT.py +62 -0
  20. spikezoo/archs/bsf/prepare_data/crop_dataset_train.py +135 -0
  21. spikezoo/archs/bsf/prepare_data/crop_dataset_val.py +139 -0
  22. spikezoo/archs/bsf/prepare_data/crop_train.sh +4 -0
  23. spikezoo/archs/bsf/prepare_data/crop_val.sh +4 -0
  24. spikezoo/archs/bsf/prepare_data/io_utils.py +64 -0
  25. spikezoo/archs/bsf/requirements.txt +9 -0
  26. spikezoo/archs/bsf/test.py +16 -0
  27. spikezoo/archs/bsf/utils.py +154 -0
  28. spikezoo/archs/spikeclip/__pycache__/nets.cpython-39.pyc +0 -0
  29. spikezoo/archs/spikeclip/nets.py +40 -0
  30. spikezoo/archs/spikeformer/CheckPoints/readme +1 -0
  31. spikezoo/archs/spikeformer/DataProcess/DataExtactor.py +60 -0
  32. spikezoo/archs/spikeformer/DataProcess/DataLoader.py +115 -0
  33. spikezoo/archs/spikeformer/DataProcess/LoadSpike.py +39 -0
  34. spikezoo/archs/spikeformer/EvalResults/readme +1 -0
  35. spikezoo/archs/spikeformer/LICENSE +21 -0
  36. spikezoo/archs/spikeformer/Metrics/Metrics.py +50 -0
  37. spikezoo/archs/spikeformer/Metrics/__init__.py +0 -0
  38. spikezoo/archs/spikeformer/Model/Loss.py +89 -0
  39. spikezoo/archs/spikeformer/Model/SpikeFormer.py +230 -0
  40. spikezoo/archs/spikeformer/Model/__init__.py +0 -0
  41. spikezoo/archs/spikeformer/Model/__pycache__/SpikeFormer.cpython-39.pyc +0 -0
  42. spikezoo/archs/spikeformer/Model/__pycache__/__init__.cpython-39.pyc +0 -0
  43. spikezoo/archs/spikeformer/README.md +30 -0
  44. spikezoo/archs/spikeformer/evaluate.py +87 -0
  45. spikezoo/archs/spikeformer/recon_real_data.py +97 -0
  46. spikezoo/archs/spikeformer/requirements.yml +95 -0
  47. spikezoo/archs/spikeformer/train.py +173 -0
  48. spikezoo/archs/spikeformer/utils.py +22 -0
  49. spikezoo/archs/spk2imgnet/.github/workflows/pylint.yml +23 -0
  50. spikezoo/archs/spk2imgnet/.gitignore +150 -0
  51. spikezoo/archs/spk2imgnet/DCNv2.py +135 -0
  52. spikezoo/archs/spk2imgnet/__pycache__/DCNv2.cpython-39.pyc +0 -0
  53. spikezoo/archs/spk2imgnet/__pycache__/align_arch.cpython-39.pyc +0 -0
  54. spikezoo/archs/spk2imgnet/__pycache__/nets.cpython-39.pyc +0 -0
  55. spikezoo/archs/spk2imgnet/align_arch.py +159 -0
  56. spikezoo/archs/spk2imgnet/dataset.py +144 -0
  57. spikezoo/archs/spk2imgnet/nets.py +230 -0
  58. spikezoo/archs/spk2imgnet/readme.md +86 -0
  59. spikezoo/archs/spk2imgnet/test_gen_imgseq.py +118 -0
  60. spikezoo/archs/spk2imgnet/train.py +189 -0
  61. spikezoo/archs/spk2imgnet/utils.py +64 -0
  62. spikezoo/archs/ssir/README.md +87 -0
  63. spikezoo/archs/ssir/configs/SSIR.yml +37 -0
  64. spikezoo/archs/ssir/configs/yml_parser.py +78 -0
  65. spikezoo/archs/ssir/datasets/dataset_sreds.py +170 -0
  66. spikezoo/archs/ssir/datasets/ds_utils.py +66 -0
  67. spikezoo/archs/ssir/losses.py +21 -0
  68. spikezoo/archs/ssir/main.py +326 -0
  69. spikezoo/archs/ssir/metrics/psnr.py +22 -0
  70. spikezoo/archs/ssir/metrics/ssim.py +54 -0
  71. spikezoo/archs/ssir/models/Vgg19.py +42 -0
  72. spikezoo/archs/ssir/models/__pycache__/layers.cpython-39.pyc +0 -0
  73. spikezoo/archs/ssir/models/__pycache__/networks.cpython-39.pyc +0 -0
  74. spikezoo/archs/ssir/models/layers.py +110 -0
  75. spikezoo/archs/ssir/models/networks.py +61 -0
  76. spikezoo/archs/ssir/requirements.txt +8 -0
  77. spikezoo/archs/ssir/shells/eval_SREDS.sh +6 -0
  78. spikezoo/archs/ssir/shells/train_SSIR.sh +12 -0
  79. spikezoo/archs/ssir/test.py +3 -0
  80. spikezoo/archs/ssir/utils.py +154 -0
  81. spikezoo/archs/ssml/__pycache__/cbam.cpython-39.pyc +0 -0
  82. spikezoo/archs/ssml/__pycache__/model.cpython-39.pyc +0 -0
  83. spikezoo/archs/ssml/cbam.py +224 -0
  84. spikezoo/archs/ssml/model.py +290 -0
  85. spikezoo/archs/ssml/res.png +0 -0
  86. spikezoo/archs/ssml/test.py +67 -0
  87. spikezoo/archs/stir/.git-credentials +0 -0
  88. spikezoo/archs/stir/README.md +65 -0
  89. spikezoo/archs/stir/ckpt_outputs/Descriptions.txt +1 -0
  90. spikezoo/archs/stir/configs/STIR.yml +37 -0
  91. spikezoo/archs/stir/configs/utils.py +155 -0
  92. spikezoo/archs/stir/configs/yml_parser.py +78 -0
  93. spikezoo/archs/stir/datasets/dataset_sreds.py +180 -0
  94. spikezoo/archs/stir/datasets/ds_utils.py +66 -0
  95. spikezoo/archs/stir/eval_SREDS.sh +5 -0
  96. spikezoo/archs/stir/main.py +397 -0
  97. spikezoo/archs/stir/metrics/losses.py +219 -0
  98. spikezoo/archs/stir/metrics/psnr.py +22 -0
  99. spikezoo/archs/stir/metrics/ssim.py +54 -0
  100. spikezoo/archs/stir/models/Vgg19.py +42 -0
  101. spikezoo/archs/stir/models/__pycache__/networks_STIR.cpython-39.pyc +0 -0
  102. spikezoo/archs/stir/models/__pycache__/submodules.cpython-39.pyc +0 -0
  103. spikezoo/archs/stir/models/__pycache__/transformer_new.cpython-39.pyc +0 -0
  104. spikezoo/archs/stir/models/networks_STIR.py +361 -0
  105. spikezoo/archs/stir/models/submodules.py +86 -0
  106. spikezoo/archs/stir/models/transformer_new.py +151 -0
  107. spikezoo/archs/stir/package_core/build/lib/package_core/__init__.py +0 -0
  108. spikezoo/archs/stir/package_core/build/lib/package_core/convertions.py +721 -0
  109. spikezoo/archs/stir/package_core/build/lib/package_core/disp_netS.py +133 -0
  110. spikezoo/archs/stir/package_core/build/lib/package_core/flow_utils.py +167 -0
  111. spikezoo/archs/stir/package_core/build/lib/package_core/generic_train_test.py +76 -0
  112. spikezoo/archs/stir/package_core/build/lib/package_core/geometry.py +458 -0
  113. spikezoo/archs/stir/package_core/build/lib/package_core/image_proc.py +183 -0
  114. spikezoo/archs/stir/package_core/build/lib/package_core/linalg.py +40 -0
  115. spikezoo/archs/stir/package_core/build/lib/package_core/losses.py +198 -0
  116. spikezoo/archs/stir/package_core/build/lib/package_core/metrics.py +51 -0
  117. spikezoo/archs/stir/package_core/build/lib/package_core/model_base.py +53 -0
  118. spikezoo/archs/stir/package_core/build/lib/package_core/net_basics.py +100 -0
  119. spikezoo/archs/stir/package_core/build/lib/package_core/resnet.py +333 -0
  120. spikezoo/archs/stir/package_core/build/lib/package_core/transforms.py +123 -0
  121. spikezoo/archs/stir/package_core/build/lib/package_core/utils.py +72 -0
  122. spikezoo/archs/stir/package_core/dist/package_core-0.0.0-py3.9.egg +0 -0
  123. spikezoo/archs/stir/package_core/package_core/__init__.py +0 -0
  124. spikezoo/archs/stir/package_core/package_core/__pycache__/__init__.cpython-39.pyc +0 -0
  125. spikezoo/archs/stir/package_core/package_core/__pycache__/net_basics.cpython-39.pyc +0 -0
  126. spikezoo/archs/stir/package_core/package_core/convertions.py +721 -0
  127. spikezoo/archs/stir/package_core/package_core/disp_netS.py +133 -0
  128. spikezoo/archs/stir/package_core/package_core/flow_utils.py +167 -0
  129. spikezoo/archs/stir/package_core/package_core/generic_train_test.py +76 -0
  130. spikezoo/archs/stir/package_core/package_core/geometry.py +458 -0
  131. spikezoo/archs/stir/package_core/package_core/image_proc.py +183 -0
  132. spikezoo/archs/stir/package_core/package_core/linalg.py +40 -0
  133. spikezoo/archs/stir/package_core/package_core/losses.py +198 -0
  134. spikezoo/archs/stir/package_core/package_core/metrics.py +51 -0
  135. spikezoo/archs/stir/package_core/package_core/model_base.py +53 -0
  136. spikezoo/archs/stir/package_core/package_core/net_basics.py +100 -0
  137. spikezoo/archs/stir/package_core/package_core/resnet.py +333 -0
  138. spikezoo/archs/stir/package_core/package_core/transforms.py +123 -0
  139. spikezoo/archs/stir/package_core/package_core/utils.py +72 -0
  140. spikezoo/archs/stir/package_core/package_core.egg-info/PKG-INFO +3 -0
  141. spikezoo/archs/stir/package_core/package_core.egg-info/SOURCES.txt +20 -0
  142. spikezoo/archs/stir/package_core/package_core.egg-info/dependency_links.txt +1 -0
  143. spikezoo/archs/stir/package_core/package_core.egg-info/top_level.txt +1 -0
  144. spikezoo/archs/stir/package_core/setup.py +5 -0
  145. spikezoo/archs/stir/requirements.txt +12 -0
  146. spikezoo/archs/stir/train_STIR.sh +9 -0
  147. spikezoo/archs/tfi/__pycache__/nets.cpython-39.pyc +0 -0
  148. spikezoo/archs/tfi/nets.py +43 -0
  149. spikezoo/archs/tfp/__pycache__/nets.cpython-39.pyc +0 -0
  150. spikezoo/archs/tfp/nets.py +13 -0
  151. spikezoo/archs/wgse/README.md +64 -0
  152. spikezoo/archs/wgse/__pycache__/dwtnets.cpython-39.pyc +0 -0
  153. spikezoo/archs/wgse/__pycache__/submodules.cpython-39.pyc +0 -0
  154. spikezoo/archs/wgse/dataset.py +59 -0
  155. spikezoo/archs/wgse/demo.png +0 -0
  156. spikezoo/archs/wgse/demo.py +83 -0
  157. spikezoo/archs/wgse/dwtnets.py +145 -0
  158. spikezoo/archs/wgse/eval.py +133 -0
  159. spikezoo/archs/wgse/logs/WGSE-Dwt1dNet-db8-5-ks3/log.txt +11 -0
  160. spikezoo/archs/wgse/submodules.py +68 -0
  161. spikezoo/archs/wgse/train.py +261 -0
  162. spikezoo/archs/wgse/transform.py +139 -0
  163. spikezoo/archs/wgse/utils.py +128 -0
  164. spikezoo/archs/wgse/weights/demo.png +0 -0
  165. spikezoo/data/base/test/gt/200_part1_key_id151.png +0 -0
  166. spikezoo/data/base/test/gt/200_part3_key_id151.png +0 -0
  167. spikezoo/data/base/test/gt/203_part1_key_id151.png +0 -0
  168. spikezoo/data/base/test/spike/200_part1_key_id151.dat +0 -0
  169. spikezoo/data/base/test/spike/200_part3_key_id151.dat +0 -0
  170. spikezoo/data/base/test/spike/203_part1_key_id151.dat +0 -0
  171. spikezoo/data/base/train/gt/203_part2_key_id151.png +0 -0
  172. spikezoo/data/base/train/gt/203_part3_key_id151.png +0 -0
  173. spikezoo/data/base/train/gt/203_part4_key_id151.png +0 -0
  174. spikezoo/data/base/train/spike/203_part2_key_id151.dat +0 -0
  175. spikezoo/data/base/train/spike/203_part3_key_id151.dat +0 -0
  176. spikezoo/data/base/train/spike/203_part4_key_id151.dat +0 -0
  177. spikezoo/datasets/base_dataset.py +2 -3
  178. spikezoo/metrics/__init__.py +1 -1
  179. spikezoo/models/base_model.py +1 -3
  180. spikezoo/pipeline/base_pipeline.py +7 -5
  181. spikezoo/pipeline/train_pipeline.py +1 -1
  182. spikezoo/utils/other_utils.py +16 -6
  183. spikezoo/utils/spike_utils.py +33 -29
  184. spikezoo/utils/vidar_loader.cpython-39-x86_64-linux-gnu.so +0 -0
  185. spikezoo-0.2.1.dist-info/METADATA +167 -0
  186. spikezoo-0.2.1.dist-info/RECORD +211 -0
  187. spikezoo/models/spcsnet_model.py +0 -19
  188. spikezoo-0.1.2.dist-info/METADATA +0 -39
  189. spikezoo-0.1.2.dist-info/RECORD +0 -36
  190. {spikezoo-0.1.2.dist-info → spikezoo-0.2.1.dist-info}/LICENSE.txt +0 -0
  191. {spikezoo-0.1.2.dist-info → spikezoo-0.2.1.dist-info}/WHEEL +0 -0
  192. {spikezoo-0.1.2.dist-info → spikezoo-0.2.1.dist-info}/top_level.txt +0 -0
@@ -35,25 +35,35 @@ def save_config(cfg, filename, mode="w"):
35
35
 
36
36
  def download_file(url, output_path):
37
37
  headers = {
38
- "sec-ch-ua": '" Not A;Brand";v="99", "Chromium";v="90", "Google Chrome";v="90"',
39
- "sec-ch-ua-mobile": "?0",
40
- "User-Agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/90.0.4430.212 Safari/537.36"
38
+ "User-Agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/90.0.4430.212 Safari/537.36",
39
+ "Accept": "*/*", # 表示接受所有类型的响应
40
+ "Accept-Encoding": "gzip, deflate, br", # 支持压缩格式
41
+ "Connection": "keep-alive" # 保持连接
41
42
  }
43
+
42
44
  try:
43
- print(f"Ready to download the file from the {url} 😊😊😊.")
44
- response = requests.head(url,headers=headers)
45
+ print(f"Ready to download the file from {url} 😊😊😊.")
46
+ response = requests.head(url, headers=headers)
45
47
  response.raise_for_status()
46
48
  file_size = int(response.headers.get("Content-Length", 0))
49
+
47
50
  response = requests.get(url, stream=True)
48
51
  response.raise_for_status()
52
+
49
53
  if response.status_code == 200:
50
54
  with open(output_path, "wb") as file, tqdm(desc="Downloading", total=file_size, unit="B", unit_scale=True) as bar:
51
55
  for chunk in response.iter_content(chunk_size=1024):
52
56
  if chunk:
53
57
  file.write(chunk)
54
58
  bar.update(len(chunk))
59
+
55
60
  print(f"Files downloaded successfully 🎉🎉🎉 and saved on {output_path}!")
56
61
  else:
57
62
  raise RuntimeError(f"Files fail to download 😔😔😔. Try downloading it from {url} and move it to {output_path}.")
63
+
58
64
  except requests.exceptions.RequestException as e:
59
- raise RuntimeError(f"Files fail to download 😔😔😔. Try downloading it from {url} and move it to {output_path}.")
65
+ # If an error occurs, remove the partially downloaded file
66
+ if os.path.exists(output_path):
67
+ os.remove(output_path)
68
+ print(f"Partial download failed. The incomplete file has been removed. 😔😔😔")
69
+ raise RuntimeError(f"Files fail to download 😔😔😔. Try downloading it from {url} and move it to {output_path}.")
@@ -2,44 +2,48 @@ import numpy as np
2
2
  import torch
3
3
  import torch.nn as nn
4
4
  import os
5
+ from .vidar_loader import load_vidar_dat_cpp
6
+ from typing import Literal
5
7
 
6
- def load_vidar_dat(filename, height, width, remove_head=False, out_type="float", out_format="array"):
8
+ def load_vidar_dat(filename, height, width,remove_head=False, version:Literal['python','cpp'] = "cpp", out_format : Literal['array','tensor']="array",):
7
9
  """Load the spike stream from the .dat file."""
8
10
  # Spike decode
9
- if isinstance(filename, str):
10
- array = np.fromfile(filename, dtype=np.uint8)
11
- elif isinstance(filename, (list, tuple)):
12
- l = []
13
- for name in filename:
14
- a = np.fromfile(name, dtype=np.uint8)
15
- l.append(a)
16
- array = np.concatenate(l)
11
+ if version == "python":
12
+ if isinstance(filename, str):
13
+ array = np.fromfile(filename, dtype=np.uint8)
14
+ elif isinstance(filename, (list, tuple)):
15
+ l = []
16
+ for name in filename:
17
+ a = np.fromfile(name, dtype=np.uint8)
18
+ l.append(a)
19
+ array = np.concatenate(l)
20
+ else:
21
+ raise NotImplementedError
22
+ len_per_frame = height * width // 8
23
+ framecnt = len(array) // len_per_frame
24
+ spikes = []
25
+ for i in range(framecnt):
26
+ compr_frame = array[i * len_per_frame : (i + 1) * len_per_frame]
27
+ blist = []
28
+ for b in range(8):
29
+ blist.append(np.right_shift(np.bitwise_and(compr_frame, np.left_shift(1, b)), b))
30
+ frame_ = np.stack(blist).transpose()
31
+ frame_ = np.flipud(frame_.reshape((height, width), order="C"))
32
+ spk = frame_.copy()[None]
33
+ spk = spk[:, :, :-16] if remove_head == True else spk
34
+ spikes.append(spk)
35
+ spikes = np.concatenate(spikes).astype(np.float32)
36
+ elif version == "cpp":
37
+ spikes = load_vidar_dat_cpp(filename, height, width)
17
38
  else:
18
- raise NotImplementedError
19
- len_per_frame = height * width // 8
20
- framecnt = len(array) // len_per_frame
21
- spikes = []
22
- for i in range(framecnt):
23
- compr_frame = array[i * len_per_frame : (i + 1) * len_per_frame]
24
- blist = []
25
- for b in range(8):
26
- blist.append(np.right_shift(np.bitwise_and(compr_frame, np.left_shift(1, b)), b))
27
- frame_ = np.stack(blist).transpose()
28
- frame_ = np.flipud(frame_.reshape((height, width), order="C"))
29
- spk = frame_.copy()[None]
30
- spk = spk[:, :, :-16] if remove_head == True else spk
31
- spikes.append(spk)
32
- spikes = np.concatenate(spikes)
33
-
34
- # Data type conversion
35
- type_dict = {"float": np.float32, "int": np.uint8}
36
- spikes = spikes.astype(type_dict[out_type])
39
+ raise RuntimeError("Not recognized version.")
37
40
 
38
- # Output format conversion
41
+ # # Output format conversion
39
42
  format_dict = {"array": lambda x: x, "tensor": torch.from_numpy}
40
43
  spikes = format_dict[out_format](spikes)
41
44
  return spikes
42
45
 
46
+
43
47
  def SpikeToRaw(save_path, SpikeSeq, filpud=True, delete_if_exists=True):
44
48
  """Save the spike sequence to the .dat file."""
45
49
  if delete_if_exists:
@@ -0,0 +1,167 @@
1
+ Metadata-Version: 2.2
2
+ Name: spikezoo
3
+ Version: 0.2.1
4
+ Summary: A deep learning toolbox for spike-to-image models.
5
+ Home-page: https://github.com/chenkang455/Spike-Zoo
6
+ Author: Kang Chen
7
+ Author-email: mrchenkang@stu.pku.edu.cn
8
+ Requires-Python: >=3.7
9
+ Description-Content-Type: text/markdown
10
+ License-File: LICENSE.txt
11
+ Requires-Dist: torch
12
+ Requires-Dist: requests
13
+ Requires-Dist: numpy
14
+ Requires-Dist: tqdm
15
+ Requires-Dist: scikit-image
16
+ Requires-Dist: lpips
17
+ Requires-Dist: pyiqa
18
+ Requires-Dist: opencv-python
19
+ Requires-Dist: thop
20
+ Requires-Dist: pytorch-wavelets
21
+ Requires-Dist: pytz
22
+ Requires-Dist: PyWavelets
23
+ Requires-Dist: pandas
24
+ Requires-Dist: pillow
25
+ Requires-Dist: scikit-learn
26
+ Requires-Dist: scipy
27
+ Requires-Dist: spikingjelly
28
+ Requires-Dist: setuptools
29
+ Dynamic: author
30
+ Dynamic: author-email
31
+ Dynamic: description
32
+ Dynamic: description-content-type
33
+ Dynamic: home-page
34
+ Dynamic: requires-dist
35
+ Dynamic: requires-python
36
+ Dynamic: summary
37
+
38
+ <h2 align="center">
39
+ <a href="">Spike-Zoo: A Toolbox for Spike-to-Image Reconstruction
40
+ </a>
41
+ </h2>
42
+
43
+ ## 📖 About
44
+ ⚡ Spike-Zoo is the go-to library for state-of-the-art pretrained **spike-to-image** models designed to reconstruct images from spike streams. Whether you're looking for a simple inference solution or aiming to train your own spike-to-image models, ⚡Spike-Zoo is a modular toolbox that supports both, with key features including:
45
+
46
+ - Fast inference with pre-trained models.
47
+ - Training support for custom-designed spike-to-image models.
48
+ - Specialized functions for processing spike data.
49
+
50
+
51
+
52
+ ## 🚩 Updates/Changelog
53
+ * **25-02-02:** Release the `Spike-Zoo v0.2` code, which supports more methods, provide more usages.
54
+ * **24-08-26:** Update the `SpikeFormer` and `RSIR` methods, the `UHSR` dataset and the `piqe` non-reference metric.
55
+
56
+ * **24-07-19:** Release the `Spike-Zoo v0.1` base code.
57
+
58
+ ## 🍾 Quick Start
59
+ ### 1. Installation
60
+ For users focused on **utilizing pretrained models for spike-to-image conversion**, we recommend installing SpikeZoo using one of the following methods:
61
+
62
+ * Install the last stable version from PyPI:
63
+ ```
64
+ pip install spikezoo
65
+ ```
66
+ * Install the latest developing version from the source code:
67
+ ```
68
+ git clone https://github.com/chenkang455/Spike-Zoo
69
+ cd Spike-Zoo
70
+ python setup.py install
71
+ ```
72
+
73
+ For users interested in **training their own spike-to-image model based on our framework**, we recommend cloning the repository and modifying the related code directly.
74
+
75
+ ### 2. Inference
76
+ Reconstructing images from the spike input is super easy with Spike-Zoo. Try the following code of the single model:
77
+ ``` python
78
+ from spikezoo.pipeline import Pipeline, PipelineConfig
79
+ pipeline = Pipeline(
80
+ cfg = PipelineConfig(save_folder="results"),
81
+ model_cfg="spk2imgnet",
82
+ dataset_cfg="base"
83
+ )
84
+ ```
85
+ You can also run multiple models at once by changing the pipeline:
86
+ ``` python
87
+ from spikezoo.pipeline import EnsemblePipeline, EnsemblePipelineConfig
88
+ pipeline = EnsemblePipeline(
89
+ cfg = EnsemblePipelineConfig(save_folder="results"),
90
+ model_cfg_list=['tfp','tfi', 'spk2imgnet', 'wgse', 'ssml', 'bsf', 'stir', 'spikeclip','spikeformer'],
91
+ dataset_cfg="base"
92
+ )
93
+ ```
94
+ * Having established the pipeline, run the following code to obtain the metric and save the reconstructed image from the given spike:
95
+ ``` python
96
+ # 1. spike-to-image from the given dataset
97
+ pipeline.spk2img_from_dataset(idx = 0)
98
+
99
+ # 2. spike-to-image from the given .dat file
100
+ pipeline.spk2img_from_file(file_path = 'data/scissor.dat',width = 400,height=250)
101
+
102
+ # 3. spike-to-image from the given spike
103
+ import spikezoo as sz
104
+ spike = sz.load_vidar_dat("data/scissor.dat",width = 400,height = 250,version='cpp')
105
+ pipeline.spk2img_from_spk(spike)
106
+ ```
107
+ For detailed usage, welcome check [test_single.ipynb](examples/test_single.ipynb) and [test_multi.ipynb](examples/test_multi.ipynb) 😊😊😊.
108
+
109
+ * Save all images of the given dataset.
110
+ ``` python
111
+ pipeline.save_imgs_from_dataset()
112
+ ```
113
+
114
+ * Calculate the metrics for the specified dataset.
115
+ ``` python
116
+ pipeline.cal_metrics()
117
+ ```
118
+
119
+ * Calculate the parameters (params,flops,latency) based on the established pipeline.
120
+ ``` python
121
+ pipeline.cal_params()
122
+ ```
123
+
124
+ ### 3. Training
125
+ We provide a user-friendly code for training our provided base model (modified from the `SpikeCLIP`) for the classic `REDS` dataset introduced in `Spk2ImgNet`:
126
+ ``` python
127
+ from spikezoo.pipeline import TrainPipelineConfig, TrainPipeline
128
+ from spikezoo.datasets.reds_small_dataset import REDS_Small_Config
129
+ pipeline = TrainPipeline(
130
+ cfg=TrainPipelineConfig(save_folder="results", epochs = 10),
131
+ dataset_cfg=REDS_Small_Config(root_dir = "path/REDS_Small"),
132
+ model_cfg="base",
133
+ )
134
+ pipeline.train()
135
+ ```
136
+ We finish the training with one 4090 GPU in `2 minutes`, achieving `34.7dB` in PSNR and `0.94` in SSIM.
137
+
138
+ > 🌟 We encourage users to develop their models using our framework, with the tutorial being released soon.
139
+
140
+ ### 4. Others
141
+ We provide a faster `load_vidar_dat` function implemented with `cpp` (by @zeal-ye):
142
+ ``` python
143
+ import spikezoo as sz
144
+ spike = sz.load_vidar_dat("data/scissor.dat",width = 400,height = 250,version='cpp')
145
+ ```
146
+ 🚀 Results on [examples/test_load_dat.py](examples/test_load_dat.py) show that the `cpp` version is more than 10 times faster than the `python` version.
147
+
148
+
149
+ ## 📅 TODO
150
+ - [ ] Provide the tutorials.
151
+ - [ ] Support more training settings.
152
+ - [ ] Support more spike-based image reconstruction methods and datasets.
153
+ - [ ] Support the overall pipeline for spike simulation.
154
+
155
+ ## ✨‍ Acknowledgment
156
+ Our code is built on the open-source projects of [SpikeCV](https://spikecv.github.io/), [IQA-Pytorch](https://github.com/chaofengc/IQA-PyTorch), [BasicSR](https://github.com/XPixelGroup/BasicSR) and [NeRFStudio](https://github.com/nerfstudio-project/nerfstudio).We appreciate the effort of the contributors to these repositories. Thanks for @ruizhao26 and @Leozhangjiyuan for their help in building this project.
157
+
158
+ ## 📑 Citation
159
+ If you find our codes helpful to your research, please consider to use the following citation:
160
+ ```
161
+ @misc{spikezoo,
162
+ title={{Spike-Zoo}: Spike-Zoo: A Toolbox for Spike-to-Image Reconstruction},
163
+ author={Kang Chen and Zhiyuan Ye},
164
+ year={2025},
165
+ howpublished = "[Online]. Available: \url{https://github.com/chenkang455/Spike-Zoo}"
166
+ }
167
+ ```
@@ -0,0 +1,211 @@
1
+ spikezoo/__init__.py,sha256=edeJMa3xcfGEZTxMO4Z7YZTh4gtBnYkCGrT7aOYse2c,283
2
+ spikezoo/archs/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
3
+ spikezoo/archs/__pycache__/__init__.cpython-39.pyc,sha256=Ba5cgbZ2jiTexmzMbqpADlFBQGzqLPtU-YbGhY9wJa0,175
4
+ spikezoo/archs/base/nets.py,sha256=j-XDEkntYl4M6xe-jqyUOhFBlhh8efNwy4qu-WVqjc4,1311
5
+ spikezoo/archs/base/__pycache__/nets.cpython-39.pyc,sha256=Ie9588wEf8hRFO7donRgviMdoSK4jeeGZn56FDa_tg8,1554
6
+ spikezoo/archs/bsf/README.md,sha256=maT9K0dZcwFPiYWmFISVogF-INTwNr6alqHSNKlk7G0,2777
7
+ spikezoo/archs/bsf/main.py,sha256=at4CWWqaoGo1k6PqRBOi_PixtDPmND6_7mU6LvWWnLI,15136
8
+ spikezoo/archs/bsf/requirements.txt,sha256=ZQCaKDZAJvmFtKPcbDSIP3gBpg_YdvwCpaDkS__kfHE,79
9
+ spikezoo/archs/bsf/test.py,sha256=J___IsAa4wQwaihif0m1Zr6yUdAVxm1mXT5CVyvR9I4,481
10
+ spikezoo/archs/bsf/utils.py,sha256=Z0uqg0OvCWIICnYuDArD1pF9HKDYLV7vuqylxPpgdVs,5111
11
+ spikezoo/archs/bsf/datasets/datasets.py,sha256=US6DIs-60P01F6GTesAI9X0btf9cDtRUCLjH4_IBMrw,13294
12
+ spikezoo/archs/bsf/datasets/ds_utils.py,sha256=6UZqLOG2SDiLIs3uBVLtZDwc5G9QspDhHe33JHZX5Vw,1907
13
+ spikezoo/archs/bsf/metrics/psnr.py,sha256=OntyhZtYIKEbdy5w-qwkl6mBt767W5pitDEjMmnqjRo,707
14
+ spikezoo/archs/bsf/metrics/ssim.py,sha256=RxVoEMJPgu370DWfDRE01UnTOorh-Xy0DldXQFhAi4o,1818
15
+ spikezoo/archs/bsf/models/get_model.py,sha256=acp0f10J81ue_lSJHlMfGeMq7IijZmPsw9Trg0nqKiQ,129
16
+ spikezoo/archs/bsf/models/bsf/align.py,sha256=X_Ud0oCZSYGFQ8DWvOG4yozUaDOJi4X44vBiXsBThiI,7979
17
+ spikezoo/archs/bsf/models/bsf/bsf.py,sha256=cgm_nEl-XMWUz7EOm8aVmfqB5PorT1YsP9GZ-57rNMY,3480
18
+ spikezoo/archs/bsf/models/bsf/dsft_convert.py,sha256=xpFwWFl1ms9LxaA96xdDOf-h_S6foScc3oh-nGjSG-o,3110
19
+ spikezoo/archs/bsf/models/bsf/rep.py,sha256=xD0nNjEQdmBwzMlUEQxYmLmZBx8ieGwbIy-v_1CeXw8,1643
20
+ spikezoo/archs/bsf/models/bsf/__pycache__/align.cpython-39.pyc,sha256=vGZzrpGGFZeVzf-FGdA3Sq8u87y7CZ-wC93vgPwkT4E,7082
21
+ spikezoo/archs/bsf/models/bsf/__pycache__/bsf.cpython-39.pyc,sha256=XreuEfY7R00NN4GXjZQpt_x44n6vIfjDher7qNOWq5c,4216
22
+ spikezoo/archs/bsf/models/bsf/__pycache__/rep.cpython-39.pyc,sha256=RtKNwwHV0-vQ3AKGyDUeaeDVFopQLsVDEIdqklJIAxw,1860
23
+ spikezoo/archs/bsf/prepare_data/DSFT.py,sha256=RDFREQc-pAGxpETgb1umjQNnLAmpqsRpEmsrjrG4hEU,2203
24
+ spikezoo/archs/bsf/prepare_data/crop_dataset_train.py,sha256=CpKIhI8kc5TzWMGSHY33IlROBTrXrY0kVGxKLvwjcvo,6050
25
+ spikezoo/archs/bsf/prepare_data/crop_dataset_val.py,sha256=Zrfe2rsnHXB-TikO9J6s0SL_u0jg492NgXSCebgvK5A,6009
26
+ spikezoo/archs/bsf/prepare_data/crop_train.sh,sha256=VoIqvQ1TWSj5uvkcp6EZIo28egILv9gjGWceUSXa8WI,193
27
+ spikezoo/archs/bsf/prepare_data/crop_val.sh,sha256=PY45EGOvdn89hAUWmeIwHrzzMWmwBbqGYgh0y5oYzP4,185
28
+ spikezoo/archs/bsf/prepare_data/io_utils.py,sha256=GUs7ocNekOKMSfMNjHAWbZFSOWXiLRtboQLl5NiY-CI,1850
29
+ spikezoo/archs/spikeclip/nets.py,sha256=j2rPD3AFWLl142XxmQL4PxWw06f4gHi8zjBUjndJ8pQ,1433
30
+ spikezoo/archs/spikeclip/__pycache__/nets.cpython-39.pyc,sha256=Bp7ISUzHzcIj5ebzUzry2pDrWdg1dlr-GH72wWTHkiM,1570
31
+ spikezoo/archs/spikeformer/LICENSE,sha256=YYnrgInlCj2xhRzxoWJZDx6DGHe85AxmozL-SyCrW8k,1069
32
+ spikezoo/archs/spikeformer/README.md,sha256=e1OWqkquJpkAG-TpzB6M0Eaob2-I0zgYgk70kanLoOk,1089
33
+ spikezoo/archs/spikeformer/evaluate.py,sha256=4orJqfDyfXNiRJpOJP_JdybdcoGHMV8ugMMhGn6WX8Q,2986
34
+ spikezoo/archs/spikeformer/recon_real_data.py,sha256=GxWPYJLJRpzAp0p1XfH1wBlgt5aY-RH6UxVRm7AV17s,3250
35
+ spikezoo/archs/spikeformer/requirements.yml,sha256=v-9a7_nyM1Rj6rZPiXPD9pxMIm0C1BO9qOyjCgdHQzI,2856
36
+ spikezoo/archs/spikeformer/train.py,sha256=t8N38QUw4NagABV5gkgWi_UssbDQOcTmOgGmZJDrToA,6655
37
+ spikezoo/archs/spikeformer/utils.py,sha256=DK0qJA01_Hs0VL9pcdyHdctJoJVr2zTPkWhk-6aT6yk,821
38
+ spikezoo/archs/spikeformer/CheckPoints/readme,sha256=h_HwP9-B0O3-fBTUVNNXcoGzVV6KBFakWSlXJGlpIRg,47
39
+ spikezoo/archs/spikeformer/DataProcess/DataExtactor.py,sha256=N_OMPASjaqXOQhGokGPehrMUwaxgh9ybF_diVycRuA8,1319
40
+ spikezoo/archs/spikeformer/DataProcess/DataLoader.py,sha256=28T5hlS1qZVhHbtn1CS22V7uqX_BXlD-GqAz3HyoFOc,2951
41
+ spikezoo/archs/spikeformer/DataProcess/LoadSpike.py,sha256=sR4tIiKLBxEyF8bzNjLJsILmBNSlYjKs8On_-c-M3yk,1580
42
+ spikezoo/archs/spikeformer/EvalResults/readme,sha256=o3kRLSB6ZZPRW4naL9EghJuMvsLOnrHKauVMfWItnBM,83
43
+ spikezoo/archs/spikeformer/Metrics/Metrics.py,sha256=YxfJSL_8UqASCy2R2G-ZCumdRFGNFCxksk8PlMwpStM,1240
44
+ spikezoo/archs/spikeformer/Metrics/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
45
+ spikezoo/archs/spikeformer/Model/Loss.py,sha256=evR1RuiU-A63p1llK_TdFJ0BCyf9l2yVraoKzzlMzEM,3183
46
+ spikezoo/archs/spikeformer/Model/SpikeFormer.py,sha256=W_nSR4m2YlE_EUaVNcz_eB8QO3Ns5dJUhOzIPL20yZc,7171
47
+ spikezoo/archs/spikeformer/Model/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
48
+ spikezoo/archs/spikeformer/Model/__pycache__/SpikeFormer.cpython-39.pyc,sha256=_ofZvPDudojhpBGitJ9BP-lHQPlVLH5hwo1fi8W2E3E,7057
49
+ spikezoo/archs/spikeformer/Model/__pycache__/__init__.cpython-39.pyc,sha256=QxKCbkXZoN8TlFT05KDorzQWZH4eXsOTYTl0y_-J-8o,193
50
+ spikezoo/archs/spk2imgnet/.gitignore,sha256=LvYh4-uHW8ZL6P5S7I6f35ZaKULvuVoBQ06ia6r1llM,2148
51
+ spikezoo/archs/spk2imgnet/DCNv2.py,sha256=KqAWzoOQFX1eEqaIP90Ahhj88qvk2K-J21WyzOMQwt4,4715
52
+ spikezoo/archs/spk2imgnet/align_arch.py,sha256=NpEDZy4YX2JD6mNrw1FOfyPpMqQ866ylYktl-kRzwIU,6264
53
+ spikezoo/archs/spk2imgnet/dataset.py,sha256=ROGBtAYtIfTLAmpCUpEkYeLWTAC8QGMpS8Tfq7jK-K8,4794
54
+ spikezoo/archs/spk2imgnet/nets.py,sha256=_VFExW8fLcxrjAf8WYgqKgZhliOa5WI_lziVFOpa7nI,8039
55
+ spikezoo/archs/spk2imgnet/readme.md,sha256=0uYl7DEeXQ7EmioqIHAkmkWXIcvxpY6n12hFTwpyjr0,2591
56
+ spikezoo/archs/spk2imgnet/test_gen_imgseq.py,sha256=UpqerSdtF5qMODIxAuVY3JnhsPy8GjNQA_4xl5X7dwk,4359
57
+ spikezoo/archs/spk2imgnet/train.py,sha256=ncNwAhFlAhDd4rMkSeAjBhRECp2En1hAZn2-RH7jVpU,6195
58
+ spikezoo/archs/spk2imgnet/utils.py,sha256=Gc-05AJDfiXqkDAaiTeLBa2oiD78l1PPrBP6frlFy30,1924
59
+ spikezoo/archs/spk2imgnet/.github/workflows/pylint.yml,sha256=lNUdbM2y3yOGPPrGUNDjnQL3fJnswtFIsuLpchBfLAc,553
60
+ spikezoo/archs/spk2imgnet/__pycache__/DCNv2.cpython-39.pyc,sha256=1mSOxJwgg-5bCAwqgIp6VOUWmHpaRv4ILnkgEiCGaI0,4095
61
+ spikezoo/archs/spk2imgnet/__pycache__/align_arch.cpython-39.pyc,sha256=WqnFKRXqXZY74-4CBp9vpebg98tzwrSsvXibOU1AzO8,4361
62
+ spikezoo/archs/spk2imgnet/__pycache__/nets.cpython-39.pyc,sha256=Kam188Xp8paFk0ZhgzHfOrNibFDExu53MQNge5aM6lI,5862
63
+ spikezoo/archs/ssir/README.md,sha256=0e56N8SqYfDRgyoyxaMctJqJ2DeWdQ4NUlLDQEUr2O0,2518
64
+ spikezoo/archs/ssir/losses.py,sha256=mUEghvJQoLM4bTs0AmDa5sZedEHWo_GksW2ojNWYYC8,623
65
+ spikezoo/archs/ssir/main.py,sha256=Pn0mmUST4hg_BJPOU6NWvorjuZ0OHncFOzVEtxDEPQQ,12054
66
+ spikezoo/archs/ssir/requirements.txt,sha256=dZYhPlQBUzNFHTogpnlmTD9IpK4qzTjsdGh5E2JRp5I,89
67
+ spikezoo/archs/ssir/test.py,sha256=JsShVNfT7sxmWsFRyWeQNcw8Mj8_bwrubMCumFtT2UA,47
68
+ spikezoo/archs/ssir/utils.py,sha256=QVekmAcOt9kH2OfeRAeEP_fj0j0Myw_NgBm-MbQntvM,5113
69
+ spikezoo/archs/ssir/configs/SSIR.yml,sha256=JXxmHrigi-deTNNA5BkD4ka9_oRhw8kLO6NfBMoAPvA,551
70
+ spikezoo/archs/ssir/configs/yml_parser.py,sha256=4IMJSoMKyBU00Qv4K7mmRBT0YFCLPHb7h7J0pIGtBLo,2078
71
+ spikezoo/archs/ssir/datasets/dataset_sreds.py,sha256=OLz-sAe1hdBroDdnC9QluFvW0zZ-nwnHOI_gUjrvbts,6259
72
+ spikezoo/archs/ssir/datasets/ds_utils.py,sha256=RfQyC_8Y50-R-xnxlGooNwAoaTvviIlSQrrRIgDMGqc,1939
73
+ spikezoo/archs/ssir/metrics/psnr.py,sha256=OntyhZtYIKEbdy5w-qwkl6mBt767W5pitDEjMmnqjRo,707
74
+ spikezoo/archs/ssir/metrics/ssim.py,sha256=RxVoEMJPgu370DWfDRE01UnTOorh-Xy0DldXQFhAi4o,1818
75
+ spikezoo/archs/ssir/models/Vgg19.py,sha256=BKYf51YqQantkuxGM5S3yD2a5Pf2nYBzEfmA0XqTjGU,1435
76
+ spikezoo/archs/ssir/models/layers.py,sha256=gYShN5cp3B1GaNmQD5_6CpYSt6k6h0cZ5IJNtrt0dCw,3450
77
+ spikezoo/archs/ssir/models/networks.py,sha256=LtcYT_SU3-X-XFCDfsyMMbe1lPe0I0eiW8oNClwsjdE,1948
78
+ spikezoo/archs/ssir/models/__pycache__/layers.cpython-39.pyc,sha256=jUAodYnILqoNJWWU2P5b1EgGItM2cSppoYWeQNKtals,3753
79
+ spikezoo/archs/ssir/models/__pycache__/networks.cpython-39.pyc,sha256=XiOQdwmw-VLLBw4LEvE8PWP8nouKX6IXu_HXOB8DGHw,2799
80
+ spikezoo/archs/ssir/shells/eval_SREDS.sh,sha256=byjDfNb_NAO8z28L7Laktlc3qYZjfeY5qn0pkMmql9E,112
81
+ spikezoo/archs/ssir/shells/train_SSIR.sh,sha256=y-LlaWNqOKwGUXKm2NodCgM3LYOIans25kN7CgDJ9z4,256
82
+ spikezoo/archs/ssml/cbam.py,sha256=hfVI1vYpboEPRBMKWqWjVlqX41XQi7A4Pwou5PJlPXo,8869
83
+ spikezoo/archs/ssml/model.py,sha256=ILN9LRoY_WZYASMSSUsCpQPabAF2kybPkW4OwCiU_nE,10220
84
+ spikezoo/archs/ssml/res.png,sha256=o8VLsy8-znCM9ZoSbsBmV3dTd8O0R48JWyNtuIekQIY,37233
85
+ spikezoo/archs/ssml/test.py,sha256=3yrMAWDBdhpyVjqNeuDtz1s2XemrR9ZXZDo8yTUAfac,2036
86
+ spikezoo/archs/ssml/__pycache__/cbam.cpython-39.pyc,sha256=wtXvwDBTBBEeRhfmXeNAmbUN6puFxomELtQHk-xHnxA,8678
87
+ spikezoo/archs/ssml/__pycache__/model.cpython-39.pyc,sha256=tgYpLrpWboUAruKwXbzvdCe78ODXryO0lGKgRt2u7to,10192
88
+ spikezoo/archs/stir/.git-credentials,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
89
+ spikezoo/archs/stir/README.md,sha256=WA0QvTGiH8kDQ4mNwj4lnZHypKSDVf_UWXckXnXltJc,3223
90
+ spikezoo/archs/stir/eval_SREDS.sh,sha256=dihTfrrr0jbkXEbpEStKgsk-t5L_4ehu9mpeMliJbME,144
91
+ spikezoo/archs/stir/main.py,sha256=fuqlFwb7K-jZVwoAzx5qCVY1R1pcTfV6gvE2SjtAs_c,14903
92
+ spikezoo/archs/stir/requirements.txt,sha256=TRxH9Ju8qvmPqM2c_8ieGoTNKm9yEzgpAshZleenSmE,114
93
+ spikezoo/archs/stir/train_STIR.sh,sha256=-h7cqdpI4hFWns16S-gTnfFUoGV-sniYPKKWrCVczSA,224
94
+ spikezoo/archs/stir/ckpt_outputs/Descriptions.txt,sha256=aVIoRpwcbwHShhsCRd2A9JfK3jRnMpNI23mWfDigaDI,47
95
+ spikezoo/archs/stir/configs/STIR.yml,sha256=gmZaraM_oob_FWKOYtReD0WdZUjJrLqWz_2RSZZRj_g,548
96
+ spikezoo/archs/stir/configs/utils.py,sha256=ebPww68XHeS2Ip8G8FD7AxRDh2-cPk1VFKBPPgUWQxk,5198
97
+ spikezoo/archs/stir/configs/yml_parser.py,sha256=4IMJSoMKyBU00Qv4K7mmRBT0YFCLPHb7h7J0pIGtBLo,2078
98
+ spikezoo/archs/stir/datasets/dataset_sreds.py,sha256=lN-woOl1RJ1Q5t7ttPIZizdkUMtYCTfXj_LAn78kqr4,6803
99
+ spikezoo/archs/stir/datasets/ds_utils.py,sha256=RfQyC_8Y50-R-xnxlGooNwAoaTvviIlSQrrRIgDMGqc,1939
100
+ spikezoo/archs/stir/metrics/losses.py,sha256=BDMKLMrAb7L_4zWzS3nXoW9RKjFrYyl9PeiZpYoW5V4,7753
101
+ spikezoo/archs/stir/metrics/psnr.py,sha256=OntyhZtYIKEbdy5w-qwkl6mBt767W5pitDEjMmnqjRo,707
102
+ spikezoo/archs/stir/metrics/ssim.py,sha256=RxVoEMJPgu370DWfDRE01UnTOorh-Xy0DldXQFhAi4o,1818
103
+ spikezoo/archs/stir/models/Vgg19.py,sha256=BKYf51YqQantkuxGM5S3yD2a5Pf2nYBzEfmA0XqTjGU,1435
104
+ spikezoo/archs/stir/models/networks_STIR.py,sha256=KA0_qYBDcD_DJ6eafmvYlSpRyfVt8i8rG88BQ-IBAI4,15290
105
+ spikezoo/archs/stir/models/submodules.py,sha256=gr0W8_ghP6pF5E5M1Ii58XYXOzR5ox8n0Xoh0vDAv6c,3360
106
+ spikezoo/archs/stir/models/transformer_new.py,sha256=INZFO156bD4A0t5agChPT87uPDJXiu9gibXMORZgzxk,6343
107
+ spikezoo/archs/stir/models/__pycache__/networks_STIR.cpython-39.pyc,sha256=GZ5MB9hqinl17F51CtdfYw0EoC75eQF4SGhrwb_WcKk,13282
108
+ spikezoo/archs/stir/models/__pycache__/submodules.cpython-39.pyc,sha256=zVNI7hYN7S2yvTKYJt8FIXSU5yUg1rfE4tyZ5zPrsWo,3640
109
+ spikezoo/archs/stir/models/__pycache__/transformer_new.cpython-39.pyc,sha256=N1ZPGfGsHXI9b1bwRAOh40eJWlmYfmK7DF9qDr248d0,5939
110
+ spikezoo/archs/stir/package_core/setup.py,sha256=l0ZAYjzpqI6IvNFm5pHOmf5jmapFXqJSdOk8SBNupc0,112
111
+ spikezoo/archs/stir/package_core/build/lib/package_core/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
112
+ spikezoo/archs/stir/package_core/build/lib/package_core/convertions.py,sha256=aKnq_wJ8sddEI2T0ITxukpDdiKsdt-zf3nzM2CHFxGs,26159
113
+ spikezoo/archs/stir/package_core/build/lib/package_core/disp_netS.py,sha256=dCjMbAwbRBGekUtYzR5xOaw5NgQ0i6QsczYTyzaSIJU,5635
114
+ spikezoo/archs/stir/package_core/build/lib/package_core/flow_utils.py,sha256=3Rs-zZsoTiZXIcHDnWi56xVdCGK0g6EkftiKgOFpIgA,4104
115
+ spikezoo/archs/stir/package_core/build/lib/package_core/generic_train_test.py,sha256=E-muWXHG9JTLtYK7LV2JxbmVaZZrbf1nMfWqTYRrBm0,2055
116
+ spikezoo/archs/stir/package_core/build/lib/package_core/geometry.py,sha256=GW-G41kdE68GGqBVciMghNu1EEBHZ3UpGGFmX4OEsMU,14855
117
+ spikezoo/archs/stir/package_core/build/lib/package_core/image_proc.py,sha256=yt1X-PudencR28TwegAxjQGpEHN4SEXg75xPg349Ae4,6404
118
+ spikezoo/archs/stir/package_core/build/lib/package_core/linalg.py,sha256=uhVfd0_w-bfrySw052z1GdoCQAkm2Tm_GQ9VghGAbPI,913
119
+ spikezoo/archs/stir/package_core/build/lib/package_core/losses.py,sha256=wnHoQguWLblFwhqSpXqRhZ-klOpGDz3LsGminC-GZGE,7037
120
+ spikezoo/archs/stir/package_core/build/lib/package_core/metrics.py,sha256=Zmt6JOFSO1K72eiEtW60IlsXej8nXyiRtXQLU80itP8,1672
121
+ spikezoo/archs/stir/package_core/build/lib/package_core/model_base.py,sha256=u_MCv3Xgajh_TxfVNF0poqbKQUcO4ip5_O44adYtELM,2074
122
+ spikezoo/archs/stir/package_core/build/lib/package_core/net_basics.py,sha256=QX_5zlC5-_ITTWwzDy_RyGw-38IxGFGCifixbGkWc2E,3374
123
+ spikezoo/archs/stir/package_core/build/lib/package_core/resnet.py,sha256=l93GwsKVBl75tUMYLZWkzZjNhO3B_Yoissb3oUdq3zE,13627
124
+ spikezoo/archs/stir/package_core/build/lib/package_core/transforms.py,sha256=_hE5Y6EWsxacwcfdI1jS-wCvwGkA32-k-4XLPVhquDY,3779
125
+ spikezoo/archs/stir/package_core/build/lib/package_core/utils.py,sha256=icSibxXKqEZyHL8GU1J0PMahCfxwVSwBtLGW_kHV25g,2316
126
+ spikezoo/archs/stir/package_core/dist/package_core-0.0.0-py3.9.egg,sha256=xopvjyDcyAvLZfdUHT-tlgT5_7D35dCwscpEifkzeCw,68292
127
+ spikezoo/archs/stir/package_core/package_core/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
128
+ spikezoo/archs/stir/package_core/package_core/convertions.py,sha256=aKnq_wJ8sddEI2T0ITxukpDdiKsdt-zf3nzM2CHFxGs,26159
129
+ spikezoo/archs/stir/package_core/package_core/disp_netS.py,sha256=dCjMbAwbRBGekUtYzR5xOaw5NgQ0i6QsczYTyzaSIJU,5635
130
+ spikezoo/archs/stir/package_core/package_core/flow_utils.py,sha256=3Rs-zZsoTiZXIcHDnWi56xVdCGK0g6EkftiKgOFpIgA,4104
131
+ spikezoo/archs/stir/package_core/package_core/generic_train_test.py,sha256=E-muWXHG9JTLtYK7LV2JxbmVaZZrbf1nMfWqTYRrBm0,2055
132
+ spikezoo/archs/stir/package_core/package_core/geometry.py,sha256=GW-G41kdE68GGqBVciMghNu1EEBHZ3UpGGFmX4OEsMU,14855
133
+ spikezoo/archs/stir/package_core/package_core/image_proc.py,sha256=yt1X-PudencR28TwegAxjQGpEHN4SEXg75xPg349Ae4,6404
134
+ spikezoo/archs/stir/package_core/package_core/linalg.py,sha256=uhVfd0_w-bfrySw052z1GdoCQAkm2Tm_GQ9VghGAbPI,913
135
+ spikezoo/archs/stir/package_core/package_core/losses.py,sha256=wnHoQguWLblFwhqSpXqRhZ-klOpGDz3LsGminC-GZGE,7037
136
+ spikezoo/archs/stir/package_core/package_core/metrics.py,sha256=Zmt6JOFSO1K72eiEtW60IlsXej8nXyiRtXQLU80itP8,1672
137
+ spikezoo/archs/stir/package_core/package_core/model_base.py,sha256=u_MCv3Xgajh_TxfVNF0poqbKQUcO4ip5_O44adYtELM,2074
138
+ spikezoo/archs/stir/package_core/package_core/net_basics.py,sha256=QX_5zlC5-_ITTWwzDy_RyGw-38IxGFGCifixbGkWc2E,3374
139
+ spikezoo/archs/stir/package_core/package_core/resnet.py,sha256=l93GwsKVBl75tUMYLZWkzZjNhO3B_Yoissb3oUdq3zE,13627
140
+ spikezoo/archs/stir/package_core/package_core/transforms.py,sha256=_hE5Y6EWsxacwcfdI1jS-wCvwGkA32-k-4XLPVhquDY,3779
141
+ spikezoo/archs/stir/package_core/package_core/utils.py,sha256=icSibxXKqEZyHL8GU1J0PMahCfxwVSwBtLGW_kHV25g,2316
142
+ spikezoo/archs/stir/package_core/package_core/__pycache__/__init__.cpython-39.pyc,sha256=kMkbhE8EH_BDTqvZladciovDa11NWsKJTLZor2XXliE,206
143
+ spikezoo/archs/stir/package_core/package_core/__pycache__/net_basics.cpython-39.pyc,sha256=ZHXtxnR4tYqkTl79drTDby1cu2c_SzgYBLQMMuyVwYQ,3622
144
+ spikezoo/archs/stir/package_core/package_core.egg-info/PKG-INFO,sha256=2njov-JTXZp2Sgwyx7KSL0fIPNyOR-lUYSSxIiDQH_Q,56
145
+ spikezoo/archs/stir/package_core/package_core.egg-info/SOURCES.txt,sha256=rTCDnAkAo4JuMaJgXeumnUODeQBpefgwT-dqrIUXoRc,541
146
+ spikezoo/archs/stir/package_core/package_core.egg-info/dependency_links.txt,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
147
+ spikezoo/archs/stir/package_core/package_core.egg-info/top_level.txt,sha256=ezjGZVvZhOw8f-HRDngFtMvGh0NfvyT3sKcG4sSOSoc,13
148
+ spikezoo/archs/tfi/nets.py,sha256=Sw4Oy92DS2DqrxzS7P02-cN1JQDq35SnngcbrM8VTEk,1377
149
+ spikezoo/archs/tfi/__pycache__/nets.cpython-39.pyc,sha256=4fzCYhy51OyXFeAtEWzNCXCRPDtsRDyS3kfGByg5zhg,1393
150
+ spikezoo/archs/tfp/nets.py,sha256=e1UzuOizrXX3AaW0fvIvvjyqZ905La2EFCWJlytk7A0,383
151
+ spikezoo/archs/tfp/__pycache__/nets.cpython-39.pyc,sha256=HsOsttvdfAoTFsq1HTsFhaDnOU9ZJ6LFZG0XSgvNMYk,847
152
+ spikezoo/archs/wgse/README.md,sha256=vUKBdCOV1MMr3ZqfXgiim99dYTERinrkzejhY-uwoiQ,3151
153
+ spikezoo/archs/wgse/dataset.py,sha256=pCvOrFRHn7tCku1bAi9vLL_tPIZQnwj57mfvSjnwFgc,1822
154
+ spikezoo/archs/wgse/demo.png,sha256=6SdZmRf6WYd6OHa1ll0F8msbnR_gsHiuqR3OLmYi1fU,64157
155
+ spikezoo/archs/wgse/demo.py,sha256=9lQDIi5at95ky0ccRdcdYYxiA9U8anX2DsXTlZTIQaQ,2416
156
+ spikezoo/archs/wgse/dwtnets.py,sha256=fyLM6O1GsPVGvaZwzq0kz8jc9kVS5bElpFbLTjbrfKY,4089
157
+ spikezoo/archs/wgse/eval.py,sha256=5Daj-1eoYRqjgp3hf6W3m54jtcVoydBPBb0bR0ZIDF0,3888
158
+ spikezoo/archs/wgse/submodules.py,sha256=qFsOnAFx7uwvIo9ymUPm3Yo6JvYnZYhrnnJyqFoPJWw,2326
159
+ spikezoo/archs/wgse/train.py,sha256=8y8rjTuTFiSnYR6wWibk_mTszsINV995BoO8nxR_u18,9361
160
+ spikezoo/archs/wgse/transform.py,sha256=bX3jPacCJdOo1FZmDgIZgS5DWrkUs3kw8njJeHh0NLQ,4532
161
+ spikezoo/archs/wgse/utils.py,sha256=UXTo8HoeB4BwSLXSbi3AyM1tokPnJ--Giz9ln2Yr0nQ,3892
162
+ spikezoo/archs/wgse/__pycache__/dwtnets.cpython-39.pyc,sha256=U0zs8m7ZzDRqrclKEeU5z5YFBBHjEbzChrOCXWMMp-I,3780
163
+ spikezoo/archs/wgse/__pycache__/submodules.cpython-39.pyc,sha256=J5JCZwJBhWRP1JQXCrxPes1tJBg_ggKVym81aTsBpYQ,2179
164
+ spikezoo/archs/wgse/logs/WGSE-Dwt1dNet-db8-5-ks3/log.txt,sha256=99XvHRXAhKc8E6JwP8fFBBjjFFEvrtOL2y17ibYXiZc,990
165
+ spikezoo/archs/wgse/weights/demo.png,sha256=jy3xM3Fe_A4b79wbxHoiPGHzMmfK_LMcMs3Y9nT1i3o,69728
166
+ spikezoo/data/base/test/gt/200_part1_key_id151.png,sha256=hkKTqpvv1Ms_xjcP3lQ2pyTswiCM1I7YLKR_ANqD5Bk,52637
167
+ spikezoo/data/base/test/gt/200_part3_key_id151.png,sha256=e4IpWztmpQ6WDDpfzorLo3Uo17ZECQX7fImLXRJHwcc,58482
168
+ spikezoo/data/base/test/gt/203_part1_key_id151.png,sha256=x5hItKgaMmpk3yejx5gy2DE2ewQVCphs7gZeqRHhsWQ,65368
169
+ spikezoo/data/base/test/spike/200_part1_key_id151.dat,sha256=wD4QG8qZgD4aaIyAZSyoeSK7casuMdcDK8g5y2vVDuU,3762500
170
+ spikezoo/data/base/test/spike/200_part3_key_id151.dat,sha256=1vwYTf1-8ybKekmLXwGnrepQzIsUU0ff0ALyciTIT4g,3762500
171
+ spikezoo/data/base/test/spike/203_part1_key_id151.dat,sha256=kuj9sy0jn9b-rWWu5MnbrdQFF1gsT8RCzEClyApcSTk,3762500
172
+ spikezoo/data/base/train/gt/203_part2_key_id151.png,sha256=OqBlaDoUIIpCvLLpVds2qIR7GdbYwbugVPLFfbYstVo,60283
173
+ spikezoo/data/base/train/gt/203_part3_key_id151.png,sha256=HqUeySlLeuJyjRj3NKjy0c1W06v3rErdFaLGncJfRew,72214
174
+ spikezoo/data/base/train/gt/203_part4_key_id151.png,sha256=xUfdlXNWdPlRshLOaEF6ug1lIbx_gGphTxzDOnf6f5Y,56719
175
+ spikezoo/data/base/train/spike/203_part2_key_id151.dat,sha256=YEenLmbPvcxnKkVn3O7yDVYb-UwpM5OPlRhxVWLYy3Q,3762500
176
+ spikezoo/data/base/train/spike/203_part3_key_id151.dat,sha256=MY9nM6XzKj-P-tRQ33WZ3G5xulNTpAXKP0y8ZQo7AIQ,3762500
177
+ spikezoo/data/base/train/spike/203_part4_key_id151.dat,sha256=IVi2jics66YzpIF-WTkw47te4qOj9cjdgz56GmHpJKg,3762500
178
+ spikezoo/datasets/__init__.py,sha256=bnskj7Bo0cWEk8XEBjOB2etTOjtsoReOxjezeZmYyTs,3274
179
+ spikezoo/datasets/base_dataset.py,sha256=gm3jLmnkbiWoWZuyjnGySU2WVgkv-tdMWLRuG5lGYvI,5787
180
+ spikezoo/datasets/realworld_dataset.py,sha256=iqn9oYu56Ph2nsA2U5aesTjflNmfr3-6krHVXC49ByM,715
181
+ spikezoo/datasets/reds_small_dataset.py,sha256=nhX95iWAADlu65XCGsU8CosjLHvxzSgAumA4MDwaTUc,883
182
+ spikezoo/datasets/szdata_dataset.py,sha256=2zLLKj6PC2XUcmgUMUqKKiA_zQZdZx7hCsI8Ft-i4o8,1230
183
+ spikezoo/datasets/uhsr_dataset.py,sha256=cuQBEuOw0AFaxV-reLWqsS2KevKTYF2Pr8ak5dSlkbc,1166
184
+ spikezoo/metrics/__init__.py,sha256=LIKeWNeEMZLANITQD68XJBOhDq7iHiKC7ExtdrXMyQs,3273
185
+ spikezoo/models/__init__.py,sha256=ePC94bz7Yjvg_CQQSWCkb7J5E_Tc370408a9tkzDFDQ,1679
186
+ spikezoo/models/base_model.py,sha256=BQmiJSRHfuzeCDQOc1eoAaso3E0SWqKAtwev7-cSF18,6695
187
+ spikezoo/models/bsf_model.py,sha256=KVKmukMfhr8hOIwIlWAKnCvDSHlXK2HJwjzP9bEGpNw,3523
188
+ spikezoo/models/spikeclip_model.py,sha256=ltEbEb-TNyzHGWHgiOc6-701LSGvSHYiD1WXglwgOJM,1010
189
+ spikezoo/models/spikeformer_model.py,sha256=oNwmzWNdemJhjDWnPoJwlxUzKLAAu9YpE2J9iwEkEoA,1586
190
+ spikezoo/models/spk2imgnet_model.py,sha256=QknQeqLfZdU2MoToUsNqFvAGN0rN2-6wzOw6I4Xtqwc,1549
191
+ spikezoo/models/ssir_model.py,sha256=sVLZ_7BwBqloB3H94BuQfyhbepwS1FdoEtbIkrPqQkk,586
192
+ spikezoo/models/ssml_model.py,sha256=u_UohCL2Q_iRZ-I_udJiQ5_30_ZASKxhttDgJKLtZ6E,493
193
+ spikezoo/models/stir_model.py,sha256=fXFajxWs2P4OoAPGXGkwOGJCDslWqJ4wus4mfbvvH7w,1176
194
+ spikezoo/models/tfi_model.py,sha256=fLaBOz1f3c-wvA8bEYVfUNk9vgtbMhbS02-Xie294mg,536
195
+ spikezoo/models/tfp_model.py,sha256=khG15qA_32PEb3BaYHFo5BzXUaxh-Napl_0bwXIcGz8,535
196
+ spikezoo/models/wgse_model.py,sha256=9N1O_ucbdQ_lndLpWNjuhidyWKs8Ct8Wr-OWTYCVc44,860
197
+ spikezoo/pipeline/__init__.py,sha256=WPsukNR4cannwsghiukqNsWbWGH5DVPapR_Ly-WOU4Q,188
198
+ spikezoo/pipeline/base_pipeline.py,sha256=n6EXiNDLeEdej1fuDfPbrbGxwCseVyJeR_Tq0ipDmBo,12265
199
+ spikezoo/pipeline/ensemble_pipeline.py,sha256=Aoi0lLcSDi9aJGIyHsjs45OYBqjDBDu97353IHLoUmw,2467
200
+ spikezoo/pipeline/train_pipeline.py,sha256=xMOX-ulb1fPLOH7EIK4ww8q1SY8x_lJvnGov_kXZl-k,4060
201
+ spikezoo/utils/__init__.py,sha256=bYLlusAXwLCoY4s6nhVgviax9ioRA9aea8qgRmj2HpI,152
202
+ spikezoo/utils/data_utils.py,sha256=mk1xeyIb7o_E1J7Z6-gtPq-rpKiMTxAWSTcvvPvVku8,2033
203
+ spikezoo/utils/img_utils.py,sha256=0O9z58VzLxQEAuz-GGWCbpeHuHPOCpgBVjCBV9kf6sI,2257
204
+ spikezoo/utils/other_utils.py,sha256=fKqs4zRxzQsIfmYZv02PZlVaGrmVEjq2KHTMrk_tBKY,2845
205
+ spikezoo/utils/spike_utils.py,sha256=5pdmQUkr3Qup0kFpNGd5rPTwyut1XlJxgUnhrBx66mk,3120
206
+ spikezoo/utils/vidar_loader.cpython-39-x86_64-linux-gnu.so,sha256=uXqu7ME---cZRRU5LUcLiNrjjtlOjxNwWHyTIQ10BGg,199088
207
+ spikezoo-0.2.1.dist-info/LICENSE.txt,sha256=ukEi8E0PKq1dQGTXHUflg3rppLymwAhr7il9x-0nPgg,1062
208
+ spikezoo-0.2.1.dist-info/METADATA,sha256=rL0LmpUASG8v-vnNa6QlpTG3-cCBhTeQFwd0fgM0DkE,6241
209
+ spikezoo-0.2.1.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
210
+ spikezoo-0.2.1.dist-info/top_level.txt,sha256=xF2iuOstrACJh43NW4dsTwIdgKfXPXAb_Xzl3M1ricM,9
211
+ spikezoo-0.2.1.dist-info/RECORD,,
@@ -1,19 +0,0 @@
1
- from dataclasses import dataclass, field
2
- from spikezoo.models.base_model import BaseModel, BaseModelConfig
3
- from typing import List
4
-
5
-
6
- @dataclass
7
- class SPCSNetConfig(BaseModelConfig):
8
- # default params for WGSE
9
- model_name: str = "spcsnet"
10
- model_file_name: str = "models"
11
- model_cls_name: str = "SPCS_Net"
12
- model_win_length: int = 41
13
- require_params: bool = True
14
- ckpt_path: str = 'weights/spcsnet.pth'
15
-
16
-
17
- class SPCSNet(BaseModel):
18
- def __init__(self, cfg: BaseModelConfig):
19
- super(SPCSNet, self).__init__(cfg)
@@ -1,39 +0,0 @@
1
- Metadata-Version: 2.2
2
- Name: spikezoo
3
- Version: 0.1.2
4
- Summary: A deep learning toolbox for spike-to-image models.
5
- Home-page: https://github.com/chenkang455/Spike-Zoo
6
- Author: Kang Chen
7
- Author-email: mrchenkang@stu.pku.edu.cn
8
- Requires-Python: >=3.7
9
- Description-Content-Type: text/markdown
10
- License-File: LICENSE.txt
11
- Requires-Dist: torch
12
- Requires-Dist: requests
13
- Requires-Dist: numpy
14
- Requires-Dist: tqdm
15
- Requires-Dist: scikit-image
16
- Requires-Dist: lpips
17
- Requires-Dist: pyiqa
18
- Requires-Dist: opencv-python
19
- Requires-Dist: thop
20
- Requires-Dist: pytorch-wavelets
21
- Requires-Dist: pytz
22
- Requires-Dist: PyWavelets
23
- Requires-Dist: pandas
24
- Requires-Dist: pillow
25
- Requires-Dist: scikit-learn
26
- Requires-Dist: scipy
27
- Requires-Dist: spikingjelly
28
- Requires-Dist: setuptools
29
- Dynamic: author
30
- Dynamic: author-email
31
- Dynamic: description
32
- Dynamic: description-content-type
33
- Dynamic: home-page
34
- Dynamic: requires-dist
35
- Dynamic: requires-python
36
- Dynamic: summary
37
-
38
- ⚡Spike-Zoo is the go-to library for state-of-the-art pretrained **spike-to-image** models for reconstructing the image from the given spike stream. Whether you're looking for a **simple inference** solution or **training** your own spike-to-image models, ⚡Spike-Zoo is a modular toolbox that supports both.
39
-
@@ -1,36 +0,0 @@
1
- spikezoo/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
2
- spikezoo/archs/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
3
- spikezoo/datasets/__init__.py,sha256=bnskj7Bo0cWEk8XEBjOB2etTOjtsoReOxjezeZmYyTs,3274
4
- spikezoo/datasets/base_dataset.py,sha256=oLBy8R_iZXloYZU-kcLyu680cSgs0J3KM62gAiHyL8E,5800
5
- spikezoo/datasets/realworld_dataset.py,sha256=iqn9oYu56Ph2nsA2U5aesTjflNmfr3-6krHVXC49ByM,715
6
- spikezoo/datasets/reds_small_dataset.py,sha256=nhX95iWAADlu65XCGsU8CosjLHvxzSgAumA4MDwaTUc,883
7
- spikezoo/datasets/szdata_dataset.py,sha256=2zLLKj6PC2XUcmgUMUqKKiA_zQZdZx7hCsI8Ft-i4o8,1230
8
- spikezoo/datasets/uhsr_dataset.py,sha256=cuQBEuOw0AFaxV-reLWqsS2KevKTYF2Pr8ak5dSlkbc,1166
9
- spikezoo/metrics/__init__.py,sha256=C1sX1sx7-UAvuR-LemYejWB9fsvl786IqIKA6LSY1so,3250
10
- spikezoo/models/__init__.py,sha256=ePC94bz7Yjvg_CQQSWCkb7J5E_Tc370408a9tkzDFDQ,1679
11
- spikezoo/models/base_model.py,sha256=zqlYuAHPqhLjG86V9qaY7xNUYs7fL3UnvjNdlojOlM0,6766
12
- spikezoo/models/bsf_model.py,sha256=KVKmukMfhr8hOIwIlWAKnCvDSHlXK2HJwjzP9bEGpNw,3523
13
- spikezoo/models/spcsnet_model.py,sha256=kLzv-ASXZGnqEFx0jUBONBeRCrsnQ_omkQUYEnr6uJc,540
14
- spikezoo/models/spikeclip_model.py,sha256=ltEbEb-TNyzHGWHgiOc6-701LSGvSHYiD1WXglwgOJM,1010
15
- spikezoo/models/spikeformer_model.py,sha256=oNwmzWNdemJhjDWnPoJwlxUzKLAAu9YpE2J9iwEkEoA,1586
16
- spikezoo/models/spk2imgnet_model.py,sha256=QknQeqLfZdU2MoToUsNqFvAGN0rN2-6wzOw6I4Xtqwc,1549
17
- spikezoo/models/ssir_model.py,sha256=sVLZ_7BwBqloB3H94BuQfyhbepwS1FdoEtbIkrPqQkk,586
18
- spikezoo/models/ssml_model.py,sha256=u_UohCL2Q_iRZ-I_udJiQ5_30_ZASKxhttDgJKLtZ6E,493
19
- spikezoo/models/stir_model.py,sha256=fXFajxWs2P4OoAPGXGkwOGJCDslWqJ4wus4mfbvvH7w,1176
20
- spikezoo/models/tfi_model.py,sha256=fLaBOz1f3c-wvA8bEYVfUNk9vgtbMhbS02-Xie294mg,536
21
- spikezoo/models/tfp_model.py,sha256=khG15qA_32PEb3BaYHFo5BzXUaxh-Napl_0bwXIcGz8,535
22
- spikezoo/models/wgse_model.py,sha256=9N1O_ucbdQ_lndLpWNjuhidyWKs8Ct8Wr-OWTYCVc44,860
23
- spikezoo/pipeline/__init__.py,sha256=WPsukNR4cannwsghiukqNsWbWGH5DVPapR_Ly-WOU4Q,188
24
- spikezoo/pipeline/base_pipeline.py,sha256=Lh00P5wzhPYJRCZryVYIREw5Wv9SN0xNKRGAf22rI_M,11993
25
- spikezoo/pipeline/ensemble_pipeline.py,sha256=Aoi0lLcSDi9aJGIyHsjs45OYBqjDBDu97353IHLoUmw,2467
26
- spikezoo/pipeline/train_pipeline.py,sha256=AseUJJYhZTIA83IFG4F6nRA6UBrMQ3p4gQaDalM8mA8,4060
27
- spikezoo/utils/__init__.py,sha256=bYLlusAXwLCoY4s6nhVgviax9ioRA9aea8qgRmj2HpI,152
28
- spikezoo/utils/data_utils.py,sha256=mk1xeyIb7o_E1J7Z6-gtPq-rpKiMTxAWSTcvvPvVku8,2033
29
- spikezoo/utils/img_utils.py,sha256=0O9z58VzLxQEAuz-GGWCbpeHuHPOCpgBVjCBV9kf6sI,2257
30
- spikezoo/utils/other_utils.py,sha256=Zy7_UM0AeppKh7u9EFL3hBj7IeQtMwqrSOXDqwBzT74,2539
31
- spikezoo/utils/spike_utils.py,sha256=CJao2QuQkxEzOCGaf1UfeP0xFsrit5QhIT_uAsLk4PE,2830
32
- spikezoo-0.1.2.dist-info/LICENSE.txt,sha256=ukEi8E0PKq1dQGTXHUflg3rppLymwAhr7il9x-0nPgg,1062
33
- spikezoo-0.1.2.dist-info/METADATA,sha256=7tSincRBGIzYtuWBZ2ncv_wbnMhVptVhWwpiO0tpnoo,1232
34
- spikezoo-0.1.2.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
35
- spikezoo-0.1.2.dist-info/top_level.txt,sha256=xF2iuOstrACJh43NW4dsTwIdgKfXPXAb_Xzl3M1ricM,9
36
- spikezoo-0.1.2.dist-info/RECORD,,