spikezoo 0.2.3.3__tar.gz → 0.2.3.5__tar.gz

{spikezoo-0.2.3.3 → spikezoo-0.2.3.5}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: spikezoo
-Version: 0.2.3.3
+Version: 0.2.3.5
 Summary: A deep learning toolbox for spike-to-image models.
 Home-page: https://github.com/chenkang455/Spike-Zoo
 Author: Kang Chen
@@ -55,7 +55,7 @@ Dynamic: summary
 - Training support for custom-designed spike-to-image models.
 - Specialized functions for processing spike data.
 
-
+> 📚Tutorials: https://spike-zoo.readthedocs.io/zh-cn/latest/# 
 
 ## 🚩 Updates/Changelog
 * **25-02-02:** Release the `Spike-Zoo v0.2` code, which supports more methods, provide more usages like training your method from scratch.
@@ -117,7 +117,6 @@ pipeline.infer_from_dataset(idx = 0)
 pipeline.infer_from_file(file_path = 'data/scissor.dat',width = 400,height=250)
 
 # 3. spike-to-image from the given spike
-import spikezoo as sz
 spike = sz.load_vidar_dat("data/scissor.dat",width = 400,height = 250)
 pipeline.infer_from_spk(spike)
 ```
@@ -252,7 +251,7 @@ If you find our codes helpful to your research, please consider to use the follo
 ```
 @misc{spikezoo,
   title={{Spike-Zoo}: Spike-Zoo: A Toolbox for Spike-to-Image Reconstruction},
-  author={Kang Chen and Zhiyuan Ye},
+  author={Kang Chen and Zhiyuan Ye and Tiejun Huang and Zhaofei Yu},
   year={2025},
   howpublished = "[Online]. Available: \url{https://github.com/chenkang455/Spike-Zoo}"
 }

{spikezoo-0.2.3.3 → spikezoo-0.2.3.5}/spikezoo/datasets/__init__.py

@@ -15,7 +15,7 @@ dataset_list = [file.replace("_dataset.py", "") for file in files_list if file.e
 def build_dataset_cfg(cfg: BaseDatasetConfig, split: Literal["train", "test"] = "test"):
     """Build the dataset from the given dataset config."""
     # build new cfg according to split
-    cfg = replace(cfg, split=split, spike_length=cfg.spike_length_train if split == "train" else cfg.spike_length_test)
+    cfg = replace(cfg, split=split)
     # dataset module
     module_name = cfg.dataset_name + "_dataset"
     assert cfg.dataset_name in dataset_list, f"Given dataset {cfg.dataset_name} not in our dataset list {dataset_list}."

{spikezoo-0.2.3.3 → spikezoo-0.2.3.5}/spikezoo/datasets/base_dataset.py

@@ -30,13 +30,13 @@ class BaseDatasetConfig:
     spike_length_train: int = -1
     "Dataset spike length for the test data."
     spike_length_test: int = -1
-    "Dataset spike length for the instantiation dataclass."
-    spike_length: int = -1
     "Dir name for the spike."
     spike_dir_name: str = "spike"
     "Dir name for the image."
     img_dir_name: str = "gt"
-
+    "Rate. (-1 denotes variant)"
+    rate: float = 0.6
+    
     # ------------- Config -------------
     "Dataset split: train/test. Default set as the 'test' for evaluation."
     split: Literal["train", "test"] = "test"
@@ -46,8 +46,7 @@ class BaseDatasetConfig:
     use_cache: bool = False
     "Crop size."
     crop_size: tuple = (-1, -1)
-    "Rate. (-1 denotes variant)"
-    rate: float = 0.6
+
 
     # post process
     def __post_init__(self):

{spikezoo-0.2.3.3 → spikezoo-0.2.3.5}/spikezoo/datasets/szdata_dataset.py

@@ -22,7 +22,3 @@ class SZDataConfig(BaseDatasetConfig):
 class SZData(BaseDataset):
     def __init__(self, cfg: BaseDatasetConfig):
         super(SZData, self).__init__(cfg)
-
-    def prepare_data(self):
-        super().prepare_data()
-        self.img_list = [self.img_dir / Path(str(s.name).replace('.dat','.png')) for s in self.spike_list]

{spikezoo-0.2.3.3 → spikezoo-0.2.3.5}/spikezoo/models/__init__.py

@@ -1,6 +1,17 @@
 import importlib
 import inspect
 from spikezoo.models.base_model import BaseModel,BaseModelConfig
+from spikezoo.models.tfp_model import TFPModel,TFPConfig
+from spikezoo.models.tfi_model import TFIModel,TFIConfig
+from spikezoo.models.spk2imgnet_model import Spk2ImgNet,Spk2ImgNetConfig
+from spikezoo.models.wgse_model import WGSE,WGSEConfig
+from spikezoo.models.ssml_model import SSML,SSMLConfig
+from spikezoo.models.bsf_model import BSF,BSFConfig
+from spikezoo.models.stir_model import STIR,STIRConfig
+from spikezoo.models.ssir_model import SSIR,SSIRConfig
+from spikezoo.models.spikeclip_model import SpikeCLIP,SpikeCLIPConfig
+
+
 from spikezoo.utils.other_utils import getattr_case_insensitive
 import os
 from pathlib import Path

{spikezoo-0.2.3.3 → spikezoo-0.2.3.5}/spikezoo/models/bsf_model.py

@@ -4,7 +4,6 @@ from spikezoo.models.base_model import BaseModel, BaseModelConfig
 from torch.optim import Adam
 import torch.optim.lr_scheduler as lr_scheduler
 import torch.nn as nn
-from spikezoo.pipeline import TrainPipelineConfig
 from typing import List
 from spikezoo.archs.bsf.models.bsf.bsf import BSF 
 

{spikezoo-0.2.3.3 → spikezoo-0.2.3.5}/spikezoo/models/spk2imgnet_model.py

@@ -1,7 +1,6 @@
 import torch
 from dataclasses import dataclass, field
 from spikezoo.models.base_model import BaseModel, BaseModelConfig
-from spikezoo.pipeline import TrainPipelineConfig
 import torch.nn as nn
 import torch.optim as optim
 import torch.optim.lr_scheduler as lr_scheduler

{spikezoo-0.2.3.3 → spikezoo-0.2.3.5}/spikezoo/models/stir_model.py

@@ -4,7 +4,6 @@ from spikezoo.models.base_model import BaseModel, BaseModelConfig
 from torch.optim import Adam
 import torch.optim.lr_scheduler as lr_scheduler
 import torch.nn as nn
-from spikezoo.pipeline import TrainPipelineConfig
 from typing import List
 from spikezoo.archs.stir.metrics.losses import compute_per_loss_single
 from spikezoo.archs.stir.models.Vgg19 import Vgg19

{spikezoo-0.2.3.3 → spikezoo-0.2.3.5}/spikezoo/models/wgse_model.py

@@ -1,7 +1,6 @@
 from dataclasses import dataclass, field
 from spikezoo.models.base_model import BaseModel, BaseModelConfig
 from typing import List
-from spikezoo.pipeline import TrainPipelineConfig
 import torch.nn as nn
 import torch.optim as optim
 import torch.optim.lr_scheduler as lr_scheduler

{spikezoo-0.2.3.3 → spikezoo-0.2.3.5}/spikezoo/pipeline/base_pipeline.py

@@ -34,7 +34,7 @@ class PipelineConfig:
     "Evaluate metrics or not."
     save_metric: bool = True
     "Metric names for evaluation."
-    metric_names: List[str] = field(default_factory=lambda: ["psnr", "ssim"])
+    metric_names: List[str] = field(default_factory=lambda: ["psnr", "ssim","niqe","brisque"])
     "Save recoverd images or not."
     save_img: bool = True
     "Normalizing recoverd images and gt or not."
@@ -56,7 +56,7 @@ class Pipeline:
 
     def _setup_model_data(self, model_cfg, dataset_cfg):
         """Model and Data setup."""
-        self.logger.info("Model and dataset is setting up...")
+        print("Model and dataset is setting up...")
         # model [1] build the model. [2] build the network.
         self.model: BaseModel = build_model_name(model_cfg) if isinstance(model_cfg, str) else build_model_cfg(model_cfg)
         self.model.build_network(mode="eval", version=self.cfg.version)
@@ -69,7 +69,7 @@ class Pipeline:
 
     def _setup_pipeline(self):
         """Pipeline setup."""
-        self.logger.info("Pipeline is setting up...")
+        print("Pipeline is setting up...")
         # save folder
         self.thistime = datetime.now().strftime("%Y-%m-%d_%H-%M-%S-%f")[:23]
         self.save_folder = Path(f"results") if len(self.cfg.save_folder) == 0 else self.cfg.save_folder

{spikezoo-0.2.3.3 → spikezoo-0.2.3.5}/spikezoo/pipeline/train_cfgs.py

@@ -30,28 +30,6 @@ class REDS_BASE_TrainConfig(TrainPipelineConfig):
     scheduler_cfg: Optional[SchedulerConfig] = MultiStepSchedulerConfig(milestones=[400], gamma=0.2) # from wgse
     loss_weight_dict: Dict = field(default_factory=lambda: {"l1": 1})
 
-# @dataclass
-# class REDS_BASE_TrainConfig(TrainPipelineConfig):
-#     """Training setting for methods on the REDS-BASE dataset."""
-
-#     # parameters setting
-#     epochs: int = 700
-#     steps_per_save_imgs: int = 200
-#     steps_per_save_ckpt: int = 500
-#     steps_per_cal_metrics: int = 100
-#     metric_names: List[str] = field(default_factory=lambda: ["psnr", "ssim"])
-
-#     # dataloader setting
-#     bs_train: int = 8
-#     num_workers: int = 4
-#     pin_memory: bool = False
-
-#     # train setting - optimizer & scheduler & loss_dict
-#     optimizer_cfg: OptimizerConfig = AdamOptimizerConfig(lr=1e-4)
-#     scheduler_cfg: Optional[SchedulerConfig] = MultiStepSchedulerConfig(milestones=[400, 600], gamma=0.2) # from wgse
-#     loss_weight_dict: Dict = field(default_factory=lambda: {"l1": 1})
-
-
 # ! Train Config for each method on the official setting, not recommended to utilize their default parameters owing to the dataset setting.
 @dataclass
 class BSFTrainConfig(TrainPipelineConfig):

{spikezoo-0.2.3.3 → spikezoo-0.2.3.5}/spikezoo/utils/spike_utils.py

@@ -35,9 +35,9 @@ def load_vidar_dat(filename, height, width, remove_head=False, version: Literal[
             frame_ = np.stack(blist).transpose()
             frame_ = np.flipud(frame_.reshape((height, width), order="C"))
             spk = frame_.copy()[None]
-            spk = spk[:, :, :-16] if remove_head == True else spk
             spikes.append(spk)
         spikes = np.concatenate(spikes).astype(np.float32)
+    spikes = spikes[:, :, :-16] if remove_head == True else spikes
 
     # # Output format conversion
     format_dict = {"array": lambda x: x, "tensor": torch.from_numpy}

{spikezoo-0.2.3.3 → spikezoo-0.2.3.5}/spikezoo.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: spikezoo
-Version: 0.2.3.3
+Version: 0.2.3.5
 Summary: A deep learning toolbox for spike-to-image models.
 Home-page: https://github.com/chenkang455/Spike-Zoo
 Author: Kang Chen
@@ -55,7 +55,7 @@ Dynamic: summary
 - Training support for custom-designed spike-to-image models.
 - Specialized functions for processing spike data.
 
-
+> 📚Tutorials: https://spike-zoo.readthedocs.io/zh-cn/latest/# 
 
 ## 🚩 Updates/Changelog
 * **25-02-02:** Release the `Spike-Zoo v0.2` code, which supports more methods, provide more usages like training your method from scratch.
@@ -117,7 +117,6 @@ pipeline.infer_from_dataset(idx = 0)
 pipeline.infer_from_file(file_path = 'data/scissor.dat',width = 400,height=250)
 
 # 3. spike-to-image from the given spike
-import spikezoo as sz
 spike = sz.load_vidar_dat("data/scissor.dat",width = 400,height = 250)
 pipeline.infer_from_spk(spike)
 ```
@@ -252,7 +251,7 @@ If you find our codes helpful to your research, please consider to use the follo
 ```
 @misc{spikezoo,
   title={{Spike-Zoo}: Spike-Zoo: A Toolbox for Spike-to-Image Reconstruction},
-  author={Kang Chen and Zhiyuan Ye},
+  author={Kang Chen and Zhiyuan Ye and Tiejun Huang and Zhaofei Yu},
   year={2025},
   howpublished = "[Online]. Available: \url{https://github.com/chenkang455/Spike-Zoo}"
 }

{spikezoo-0.2.3.3 → spikezoo-0.2.3.5}/spikezoo.egg-info/SOURCES.txt

@@ -6,7 +6,9 @@ spikezoo.egg-info/dependency_links.txt
 spikezoo.egg-info/requires.txt
 spikezoo.egg-info/top_level.txt
 spikezoo/archs/__init__.py
+spikezoo/archs/__pycache__/__init__.cpython-39.pyc
 spikezoo/archs/base/nets.py
+spikezoo/archs/base/__pycache__/nets.cpython-39.pyc
 spikezoo/archs/bsf/README.md
 spikezoo/archs/bsf/main.py
 spikezoo/archs/bsf/requirements.txt
@@ -21,6 +23,9 @@ spikezoo/archs/bsf/models/bsf/align.py
 spikezoo/archs/bsf/models/bsf/bsf.py
 spikezoo/archs/bsf/models/bsf/dsft_convert.py
 spikezoo/archs/bsf/models/bsf/rep.py
+spikezoo/archs/bsf/models/bsf/__pycache__/align.cpython-39.pyc
+spikezoo/archs/bsf/models/bsf/__pycache__/bsf.cpython-39.pyc
+spikezoo/archs/bsf/models/bsf/__pycache__/rep.cpython-39.pyc
 spikezoo/archs/bsf/prepare_data/DSFT.py
 spikezoo/archs/bsf/prepare_data/crop_dataset_train.py
 spikezoo/archs/bsf/prepare_data/crop_dataset_val.py
@@ -28,6 +33,7 @@ spikezoo/archs/bsf/prepare_data/crop_train.sh
 spikezoo/archs/bsf/prepare_data/crop_val.sh
 spikezoo/archs/bsf/prepare_data/io_utils.py
 spikezoo/archs/spikeclip/nets.py
+spikezoo/archs/spikeclip/__pycache__/nets.cpython-39.pyc
 spikezoo/archs/spk2imgnet/.gitignore
 spikezoo/archs/spk2imgnet/DCNv2.py
 spikezoo/archs/spk2imgnet/align_arch.py
@@ -38,6 +44,9 @@ spikezoo/archs/spk2imgnet/test_gen_imgseq.py
 spikezoo/archs/spk2imgnet/train.py
 spikezoo/archs/spk2imgnet/utils.py
 spikezoo/archs/spk2imgnet/.github/workflows/pylint.yml
+spikezoo/archs/spk2imgnet/__pycache__/DCNv2.cpython-39.pyc
+spikezoo/archs/spk2imgnet/__pycache__/align_arch.cpython-39.pyc
+spikezoo/archs/spk2imgnet/__pycache__/nets.cpython-39.pyc
 spikezoo/archs/ssir/README.md
 spikezoo/archs/ssir/losses.py
 spikezoo/archs/ssir/main.py
@@ -53,12 +62,16 @@ spikezoo/archs/ssir/metrics/ssim.py
 spikezoo/archs/ssir/models/Vgg19.py
 spikezoo/archs/ssir/models/layers.py
 spikezoo/archs/ssir/models/networks.py
+spikezoo/archs/ssir/models/__pycache__/layers.cpython-39.pyc
+spikezoo/archs/ssir/models/__pycache__/networks.cpython-39.pyc
 spikezoo/archs/ssir/shells/eval_SREDS.sh
 spikezoo/archs/ssir/shells/train_SSIR.sh
 spikezoo/archs/ssml/cbam.py
 spikezoo/archs/ssml/model.py
 spikezoo/archs/ssml/res.png
 spikezoo/archs/ssml/test.py
+spikezoo/archs/ssml/__pycache__/cbam.cpython-39.pyc
+spikezoo/archs/ssml/__pycache__/model.cpython-39.pyc
 spikezoo/archs/stir/.git-credentials
 spikezoo/archs/stir/README.md
 spikezoo/archs/stir/eval_SREDS.sh
@@ -74,10 +87,15 @@ spikezoo/archs/stir/datasets/ds_utils.py
 spikezoo/archs/stir/metrics/losses.py
 spikezoo/archs/stir/metrics/psnr.py
 spikezoo/archs/stir/metrics/ssim.py
+spikezoo/archs/stir/metrics/__pycache__/losses.cpython-39.pyc
 spikezoo/archs/stir/models/Vgg19.py
 spikezoo/archs/stir/models/networks_STIR.py
 spikezoo/archs/stir/models/submodules.py
 spikezoo/archs/stir/models/transformer_new.py
+spikezoo/archs/stir/models/__pycache__/Vgg19.cpython-39.pyc
+spikezoo/archs/stir/models/__pycache__/networks_STIR.cpython-39.pyc
+spikezoo/archs/stir/models/__pycache__/submodules.cpython-39.pyc
+spikezoo/archs/stir/models/__pycache__/transformer_new.cpython-39.pyc
 spikezoo/archs/stir/package_core/setup.py
 spikezoo/archs/stir/package_core/build/lib/package_core/__init__.py
 spikezoo/archs/stir/package_core/build/lib/package_core/convertions.py
@@ -114,8 +132,15 @@ spikezoo/archs/stir/package_core/package_core.egg-info/PKG-INFO
 spikezoo/archs/stir/package_core/package_core.egg-info/SOURCES.txt
 spikezoo/archs/stir/package_core/package_core.egg-info/dependency_links.txt
 spikezoo/archs/stir/package_core/package_core.egg-info/top_level.txt
+spikezoo/archs/stir/package_core/package_core/__pycache__/__init__.cpython-39.pyc
+spikezoo/archs/stir/package_core/package_core/__pycache__/geometry.cpython-39.pyc
+spikezoo/archs/stir/package_core/package_core/__pycache__/image_proc.cpython-39.pyc
+spikezoo/archs/stir/package_core/package_core/__pycache__/losses.cpython-39.pyc
+spikezoo/archs/stir/package_core/package_core/__pycache__/net_basics.cpython-39.pyc
 spikezoo/archs/tfi/nets.py
+spikezoo/archs/tfi/__pycache__/nets.cpython-39.pyc
 spikezoo/archs/tfp/nets.py
+spikezoo/archs/tfp/__pycache__/nets.cpython-39.pyc
 spikezoo/archs/wgse/README.md
 spikezoo/archs/wgse/dataset.py
 spikezoo/archs/wgse/demo.png
@@ -126,6 +151,8 @@ spikezoo/archs/wgse/submodules.py
 spikezoo/archs/wgse/train.py
 spikezoo/archs/wgse/transform.py
 spikezoo/archs/wgse/utils.py
+spikezoo/archs/wgse/__pycache__/dwtnets.cpython-39.pyc
+spikezoo/archs/wgse/__pycache__/submodules.cpython-39.pyc
 spikezoo/archs/wgse/logs/WGSE-Dwt1dNet-db8-5-ks3/log.txt
 spikezoo/archs/wgse/weights/demo.png
 spikezoo/data/base/test/gt/200_part1_key_id151.png