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.
- spikezoo/__init__.py +13 -0
- spikezoo/archs/__pycache__/__init__.cpython-39.pyc +0 -0
- spikezoo/archs/base/__pycache__/nets.cpython-39.pyc +0 -0
- spikezoo/archs/base/nets.py +34 -0
- spikezoo/archs/bsf/README.md +92 -0
- spikezoo/archs/bsf/datasets/datasets.py +328 -0
- spikezoo/archs/bsf/datasets/ds_utils.py +64 -0
- spikezoo/archs/bsf/main.py +398 -0
- spikezoo/archs/bsf/metrics/psnr.py +22 -0
- spikezoo/archs/bsf/metrics/ssim.py +54 -0
- spikezoo/archs/bsf/models/bsf/__pycache__/align.cpython-39.pyc +0 -0
- spikezoo/archs/bsf/models/bsf/__pycache__/bsf.cpython-39.pyc +0 -0
- spikezoo/archs/bsf/models/bsf/__pycache__/rep.cpython-39.pyc +0 -0
- spikezoo/archs/bsf/models/bsf/align.py +154 -0
- spikezoo/archs/bsf/models/bsf/bsf.py +105 -0
- spikezoo/archs/bsf/models/bsf/dsft_convert.py +96 -0
- spikezoo/archs/bsf/models/bsf/rep.py +44 -0
- spikezoo/archs/bsf/models/get_model.py +7 -0
- spikezoo/archs/bsf/prepare_data/DSFT.py +62 -0
- spikezoo/archs/bsf/prepare_data/crop_dataset_train.py +135 -0
- spikezoo/archs/bsf/prepare_data/crop_dataset_val.py +139 -0
- spikezoo/archs/bsf/prepare_data/crop_train.sh +4 -0
- spikezoo/archs/bsf/prepare_data/crop_val.sh +4 -0
- spikezoo/archs/bsf/prepare_data/io_utils.py +64 -0
- spikezoo/archs/bsf/requirements.txt +9 -0
- spikezoo/archs/bsf/test.py +16 -0
- spikezoo/archs/bsf/utils.py +154 -0
- spikezoo/archs/spikeclip/__pycache__/nets.cpython-39.pyc +0 -0
- spikezoo/archs/spikeclip/nets.py +40 -0
- spikezoo/archs/spikeformer/CheckPoints/readme +1 -0
- spikezoo/archs/spikeformer/DataProcess/DataExtactor.py +60 -0
- spikezoo/archs/spikeformer/DataProcess/DataLoader.py +115 -0
- spikezoo/archs/spikeformer/DataProcess/LoadSpike.py +39 -0
- spikezoo/archs/spikeformer/EvalResults/readme +1 -0
- spikezoo/archs/spikeformer/LICENSE +21 -0
- spikezoo/archs/spikeformer/Metrics/Metrics.py +50 -0
- spikezoo/archs/spikeformer/Metrics/__init__.py +0 -0
- spikezoo/archs/spikeformer/Model/Loss.py +89 -0
- spikezoo/archs/spikeformer/Model/SpikeFormer.py +230 -0
- spikezoo/archs/spikeformer/Model/__init__.py +0 -0
- spikezoo/archs/spikeformer/Model/__pycache__/SpikeFormer.cpython-39.pyc +0 -0
- spikezoo/archs/spikeformer/Model/__pycache__/__init__.cpython-39.pyc +0 -0
- spikezoo/archs/spikeformer/README.md +30 -0
- spikezoo/archs/spikeformer/evaluate.py +87 -0
- spikezoo/archs/spikeformer/recon_real_data.py +97 -0
- spikezoo/archs/spikeformer/requirements.yml +95 -0
- spikezoo/archs/spikeformer/train.py +173 -0
- spikezoo/archs/spikeformer/utils.py +22 -0
- spikezoo/archs/spk2imgnet/.github/workflows/pylint.yml +23 -0
- spikezoo/archs/spk2imgnet/.gitignore +150 -0
- spikezoo/archs/spk2imgnet/DCNv2.py +135 -0
- spikezoo/archs/spk2imgnet/__pycache__/DCNv2.cpython-39.pyc +0 -0
- spikezoo/archs/spk2imgnet/__pycache__/align_arch.cpython-39.pyc +0 -0
- spikezoo/archs/spk2imgnet/__pycache__/nets.cpython-39.pyc +0 -0
- spikezoo/archs/spk2imgnet/align_arch.py +159 -0
- spikezoo/archs/spk2imgnet/dataset.py +144 -0
- spikezoo/archs/spk2imgnet/nets.py +230 -0
- spikezoo/archs/spk2imgnet/readme.md +86 -0
- spikezoo/archs/spk2imgnet/test_gen_imgseq.py +118 -0
- spikezoo/archs/spk2imgnet/train.py +189 -0
- spikezoo/archs/spk2imgnet/utils.py +64 -0
- spikezoo/archs/ssir/README.md +87 -0
- spikezoo/archs/ssir/configs/SSIR.yml +37 -0
- spikezoo/archs/ssir/configs/yml_parser.py +78 -0
- spikezoo/archs/ssir/datasets/dataset_sreds.py +170 -0
- spikezoo/archs/ssir/datasets/ds_utils.py +66 -0
- spikezoo/archs/ssir/losses.py +21 -0
- spikezoo/archs/ssir/main.py +326 -0
- spikezoo/archs/ssir/metrics/psnr.py +22 -0
- spikezoo/archs/ssir/metrics/ssim.py +54 -0
- spikezoo/archs/ssir/models/Vgg19.py +42 -0
- spikezoo/archs/ssir/models/__pycache__/layers.cpython-39.pyc +0 -0
- spikezoo/archs/ssir/models/__pycache__/networks.cpython-39.pyc +0 -0
- spikezoo/archs/ssir/models/layers.py +110 -0
- spikezoo/archs/ssir/models/networks.py +61 -0
- spikezoo/archs/ssir/requirements.txt +8 -0
- spikezoo/archs/ssir/shells/eval_SREDS.sh +6 -0
- spikezoo/archs/ssir/shells/train_SSIR.sh +12 -0
- spikezoo/archs/ssir/test.py +3 -0
- spikezoo/archs/ssir/utils.py +154 -0
- spikezoo/archs/ssml/__pycache__/cbam.cpython-39.pyc +0 -0
- spikezoo/archs/ssml/__pycache__/model.cpython-39.pyc +0 -0
- spikezoo/archs/ssml/cbam.py +224 -0
- spikezoo/archs/ssml/model.py +290 -0
- spikezoo/archs/ssml/res.png +0 -0
- spikezoo/archs/ssml/test.py +67 -0
- spikezoo/archs/stir/.git-credentials +0 -0
- spikezoo/archs/stir/README.md +65 -0
- spikezoo/archs/stir/ckpt_outputs/Descriptions.txt +1 -0
- spikezoo/archs/stir/configs/STIR.yml +37 -0
- spikezoo/archs/stir/configs/utils.py +155 -0
- spikezoo/archs/stir/configs/yml_parser.py +78 -0
- spikezoo/archs/stir/datasets/dataset_sreds.py +180 -0
- spikezoo/archs/stir/datasets/ds_utils.py +66 -0
- spikezoo/archs/stir/eval_SREDS.sh +5 -0
- spikezoo/archs/stir/main.py +397 -0
- spikezoo/archs/stir/metrics/losses.py +219 -0
- spikezoo/archs/stir/metrics/psnr.py +22 -0
- spikezoo/archs/stir/metrics/ssim.py +54 -0
- spikezoo/archs/stir/models/Vgg19.py +42 -0
- spikezoo/archs/stir/models/__pycache__/networks_STIR.cpython-39.pyc +0 -0
- spikezoo/archs/stir/models/__pycache__/submodules.cpython-39.pyc +0 -0
- spikezoo/archs/stir/models/__pycache__/transformer_new.cpython-39.pyc +0 -0
- spikezoo/archs/stir/models/networks_STIR.py +361 -0
- spikezoo/archs/stir/models/submodules.py +86 -0
- spikezoo/archs/stir/models/transformer_new.py +151 -0
- spikezoo/archs/stir/package_core/build/lib/package_core/__init__.py +0 -0
- spikezoo/archs/stir/package_core/build/lib/package_core/convertions.py +721 -0
- spikezoo/archs/stir/package_core/build/lib/package_core/disp_netS.py +133 -0
- spikezoo/archs/stir/package_core/build/lib/package_core/flow_utils.py +167 -0
- spikezoo/archs/stir/package_core/build/lib/package_core/generic_train_test.py +76 -0
- spikezoo/archs/stir/package_core/build/lib/package_core/geometry.py +458 -0
- spikezoo/archs/stir/package_core/build/lib/package_core/image_proc.py +183 -0
- spikezoo/archs/stir/package_core/build/lib/package_core/linalg.py +40 -0
- spikezoo/archs/stir/package_core/build/lib/package_core/losses.py +198 -0
- spikezoo/archs/stir/package_core/build/lib/package_core/metrics.py +51 -0
- spikezoo/archs/stir/package_core/build/lib/package_core/model_base.py +53 -0
- spikezoo/archs/stir/package_core/build/lib/package_core/net_basics.py +100 -0
- spikezoo/archs/stir/package_core/build/lib/package_core/resnet.py +333 -0
- spikezoo/archs/stir/package_core/build/lib/package_core/transforms.py +123 -0
- spikezoo/archs/stir/package_core/build/lib/package_core/utils.py +72 -0
- spikezoo/archs/stir/package_core/dist/package_core-0.0.0-py3.9.egg +0 -0
- spikezoo/archs/stir/package_core/package_core/__init__.py +0 -0
- spikezoo/archs/stir/package_core/package_core/__pycache__/__init__.cpython-39.pyc +0 -0
- spikezoo/archs/stir/package_core/package_core/__pycache__/net_basics.cpython-39.pyc +0 -0
- spikezoo/archs/stir/package_core/package_core/convertions.py +721 -0
- spikezoo/archs/stir/package_core/package_core/disp_netS.py +133 -0
- spikezoo/archs/stir/package_core/package_core/flow_utils.py +167 -0
- spikezoo/archs/stir/package_core/package_core/generic_train_test.py +76 -0
- spikezoo/archs/stir/package_core/package_core/geometry.py +458 -0
- spikezoo/archs/stir/package_core/package_core/image_proc.py +183 -0
- spikezoo/archs/stir/package_core/package_core/linalg.py +40 -0
- spikezoo/archs/stir/package_core/package_core/losses.py +198 -0
- spikezoo/archs/stir/package_core/package_core/metrics.py +51 -0
- spikezoo/archs/stir/package_core/package_core/model_base.py +53 -0
- spikezoo/archs/stir/package_core/package_core/net_basics.py +100 -0
- spikezoo/archs/stir/package_core/package_core/resnet.py +333 -0
- spikezoo/archs/stir/package_core/package_core/transforms.py +123 -0
- spikezoo/archs/stir/package_core/package_core/utils.py +72 -0
- spikezoo/archs/stir/package_core/package_core.egg-info/PKG-INFO +3 -0
- spikezoo/archs/stir/package_core/package_core.egg-info/SOURCES.txt +20 -0
- spikezoo/archs/stir/package_core/package_core.egg-info/dependency_links.txt +1 -0
- spikezoo/archs/stir/package_core/package_core.egg-info/top_level.txt +1 -0
- spikezoo/archs/stir/package_core/setup.py +5 -0
- spikezoo/archs/stir/requirements.txt +12 -0
- spikezoo/archs/stir/train_STIR.sh +9 -0
- spikezoo/archs/tfi/__pycache__/nets.cpython-39.pyc +0 -0
- spikezoo/archs/tfi/nets.py +43 -0
- spikezoo/archs/tfp/__pycache__/nets.cpython-39.pyc +0 -0
- spikezoo/archs/tfp/nets.py +13 -0
- spikezoo/archs/wgse/README.md +64 -0
- spikezoo/archs/wgse/__pycache__/dwtnets.cpython-39.pyc +0 -0
- spikezoo/archs/wgse/__pycache__/submodules.cpython-39.pyc +0 -0
- spikezoo/archs/wgse/dataset.py +59 -0
- spikezoo/archs/wgse/demo.png +0 -0
- spikezoo/archs/wgse/demo.py +83 -0
- spikezoo/archs/wgse/dwtnets.py +145 -0
- spikezoo/archs/wgse/eval.py +133 -0
- spikezoo/archs/wgse/logs/WGSE-Dwt1dNet-db8-5-ks3/log.txt +11 -0
- spikezoo/archs/wgse/submodules.py +68 -0
- spikezoo/archs/wgse/train.py +261 -0
- spikezoo/archs/wgse/transform.py +139 -0
- spikezoo/archs/wgse/utils.py +128 -0
- spikezoo/archs/wgse/weights/demo.png +0 -0
- spikezoo/data/base/test/gt/200_part1_key_id151.png +0 -0
- spikezoo/data/base/test/gt/200_part3_key_id151.png +0 -0
- spikezoo/data/base/test/gt/203_part1_key_id151.png +0 -0
- spikezoo/data/base/test/spike/200_part1_key_id151.dat +0 -0
- spikezoo/data/base/test/spike/200_part3_key_id151.dat +0 -0
- spikezoo/data/base/test/spike/203_part1_key_id151.dat +0 -0
- spikezoo/data/base/train/gt/203_part2_key_id151.png +0 -0
- spikezoo/data/base/train/gt/203_part3_key_id151.png +0 -0
- spikezoo/data/base/train/gt/203_part4_key_id151.png +0 -0
- spikezoo/data/base/train/spike/203_part2_key_id151.dat +0 -0
- spikezoo/data/base/train/spike/203_part3_key_id151.dat +0 -0
- spikezoo/data/base/train/spike/203_part4_key_id151.dat +0 -0
- spikezoo/datasets/base_dataset.py +2 -3
- spikezoo/metrics/__init__.py +1 -1
- spikezoo/models/base_model.py +1 -3
- spikezoo/pipeline/base_pipeline.py +7 -5
- spikezoo/pipeline/train_pipeline.py +1 -1
- spikezoo/utils/other_utils.py +16 -6
- spikezoo/utils/spike_utils.py +33 -29
- spikezoo/utils/vidar_loader.cpython-39-x86_64-linux-gnu.so +0 -0
- spikezoo-0.2.1.dist-info/METADATA +167 -0
- spikezoo-0.2.1.dist-info/RECORD +211 -0
- spikezoo/models/spcsnet_model.py +0 -19
- spikezoo-0.1.2.dist-info/METADATA +0 -39
- spikezoo-0.1.2.dist-info/RECORD +0 -36
- {spikezoo-0.1.2.dist-info → spikezoo-0.2.1.dist-info}/LICENSE.txt +0 -0
- {spikezoo-0.1.2.dist-info → spikezoo-0.2.1.dist-info}/WHEEL +0 -0
- {spikezoo-0.1.2.dist-info → spikezoo-0.2.1.dist-info}/top_level.txt +0 -0
spikezoo/utils/other_utils.py
CHANGED
@@ -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
|
-
"
|
39
|
-
"
|
40
|
-
"
|
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
|
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
|
-
|
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}.")
|
spikezoo/utils/spike_utils.py
CHANGED
@@ -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,
|
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
|
10
|
-
|
11
|
-
|
12
|
-
|
13
|
-
|
14
|
-
|
15
|
-
|
16
|
-
|
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
|
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:
|
Binary file
|
@@ -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,,
|
spikezoo/models/spcsnet_model.py
DELETED
@@ -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
|
-
|
spikezoo-0.1.2.dist-info/RECORD
DELETED
@@ -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,,
|
File without changes
|
File without changes
|
File without changes
|