spikezoo 0.2.3__py3-none-any.whl → 0.2.3.3__py3-none-any.whl

spikezoo/datasets/base_dataset.py

@@ -47,7 +47,7 @@ class BaseDatasetConfig:
     "Crop size."
     crop_size: tuple = (-1, -1)
     "Rate. (-1 denotes variant)"
-    rate: float = 1
+    rate: float = 0.6
 
     # post process
     def __post_init__(self):

spikezoo/datasets/realworld_dataset.py

@@ -12,6 +12,8 @@ class RealWorldConfig(BaseDatasetConfig):
     with_img: bool = False
     spike_length_train: int = -1
     spike_length_test: int = -1
+    rate: float = 1
+    
 
 
 class RealWorld(BaseDataset):

spikezoo/datasets/szdata_dataset.py

@@ -17,6 +17,7 @@ class SZDataConfig(BaseDatasetConfig):
     spike_length_test: int = -1
     spike_dir_name: str = "spike_data"
     img_dir_name: str = "sharp_data"
+    rate: float = 1
 
 class SZData(BaseDataset):
     def __init__(self, cfg: BaseDatasetConfig):

spikezoo/datasets/uhsr_dataset.py

@@ -4,7 +4,6 @@ from dataclasses import dataclass
 import numpy as np
 import torch
 
-
 @dataclass
 class UHSRConfig(BaseDatasetConfig):
     dataset_name: str = "uhsr"
@@ -16,7 +15,7 @@ class UHSRConfig(BaseDatasetConfig):
     spike_length_test: int = 200
     spike_dir_name: str = "spike"
     img_dir_name: str = ""
-
+    rate: float = 1
 
 class UHSR(BaseDataset):
     def __init__(self, cfg: BaseDatasetConfig):

spikezoo/models/base_model.py

@@ -93,9 +93,9 @@ class BaseModel(nn.Module):
             if version != "local":
                 load_folder = os.path.dirname(os.path.abspath(__file__))
                 ckpt_name = f"{self.cfg.model_name}.{get_suffix(self.cfg.model_name,version)}"
-                ckpt_path = f"weights/{version}/{ckpt_name}"
+                ckpt_path = os.path.join("weights",version,ckpt_name)
                 ckpt_path = os.path.join(load_folder, ckpt_path)
-                ckpt_path_url = f"{self.cfg.base_url}/{get_url_version(version)}/{ckpt_name}"
+                ckpt_path_url = os.path.join(self.cfg.base_url,get_url_version(version),ckpt_name)
             elif version == "local":
                 ckpt_path = self.cfg.ckpt_path
 

spikezoo/pipeline/base_pipeline.py

@@ -56,6 +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...")
         # 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)
@@ -68,6 +69,7 @@ class Pipeline:
 
     def _setup_pipeline(self):
         """Pipeline setup."""
+        self.logger.info("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.dist-info → spikezoo-0.2.3.3.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: spikezoo
-Version: 0.2.3
+Version: 0.2.3.3
 Summary: A deep learning toolbox for spike-to-image models.
 Home-page: https://github.com/chenkang455/Spike-Zoo
 Author: Kang Chen
@@ -36,15 +36,11 @@ Dynamic: requires-python
 Dynamic: summary
 
 <p align="center">
-    <br>
-    <img src="imgs/spike-zoo.png" width="500"/>
-    <br>
+    <img src="imgs/spike-zoo.png" width="350"/>
 <p>
-
 <h5 align="center">
 
 [![GitHub repo stars](https://img.shields.io/github/stars/chenkang455/Spike-Zoo?style=flat&logo=github&logoColor=whitesmoke&label=Stars)](https://github.com/chenkang455/Spike-Zoo/stargazers) [![GitHub Issues](https://img.shields.io/github/issues/chenkang455/Spike-Zoo?style=flat&logo=github&logoColor=whitesmoke&label=Stars)](https://github.com/chenkang455/Spike-Zoo/issues) <a href="https://badge.fury.io/py/spikezoo"><img src="https://badge.fury.io/py/spikezoo.svg" alt="PyPI version"></a> [![License](https://img.shields.io/badge/License-MIT-yellow)](https://github.com/chenkang455/Spike-Zoo)
- 
 <p>
 
 <!-- <h2 align="center"> 
@@ -165,16 +161,16 @@ We retrain all supported methods except `SPIKECLIP` on this REDS dataset (traini
 
 | Method               | PSNR  | SSIM   | LPIPS   | NIQE    | BRISQUE  | PIQE  | Params (M) | FLOPs (G) | Latency (ms) |
 |----------------------|:-------:|:--------:|:---------:|:---------:|:----------:|:-------:|:------------:|:-----------:|:--------------:|
-| `TFI`                | 16.503 | 0.454  | 0.382   | 7.289   | 43.17    | 49.12 | 0.00       | 0.00      | 3.60         |
-| `TFP`                | 24.287 | 0.644  | 0.274   | 8.197   | 48.48    | 38.38 | 0.00       | 0.00      | 0.03         |
-| `SPIKECLIP`          | 21.873 | 0.578  | 0.333   | 7.802   | 42.08    | 54.01 | 0.19       | 23.69     | 1.27         |
-| `SSIR`               | 26.544 | 0.718  | 0.325   | 4.769   | 28.45    | 21.59 | 0.38       | 25.92     | 4.52         |
-| `SSML`               | 33.697 | 0.943  | 0.088   | 4.669   | 32.48    | 37.30 | 2.38       | 386.02    | 244.18       |
-| `BASE`               | 36.589 | 0.965  | 0.034   | 4.393   | 26.16    | 38.43 | 0.18       | 18.04     | 0.40         |
-| `STIR`               | 37.914 | 0.973  | 0.027   | 4.236   | 25.10    | 39.18 | 5.08       | 43.31     | 21.07        |
-| `WGSE`               | 39.036 | 0.978  | 0.023   | 4.231   | 25.76    | 44.11 | 3.81       | 415.26    | 73.62        |
-| `SPK2IMGNET`         | 39.154 | 0.978  | 0.022   | 4.243   | 25.20    | 43.09 | 3.90       | 1000.50   | 123.38       |
-| `BSF`                | 39.576 | 0.979  | 0.019   | 4.139   | 24.93    | 43.03 | 2.47       | 705.23    | 401.50       |
+| `tfi`                | 16.503 | 0.454  | 0.382   | 7.289   | 43.17    | 49.12 | 0.00       | 0.00      | 3.60         |
+| `tfp`                | 24.287 | 0.644  | 0.274   | 8.197   | 48.48    | 38.38 | 0.00       | 0.00      | 0.03         |
+| `spikeclip`          | 21.873 | 0.578  | 0.333   | 7.802   | 42.08    | 54.01 | 0.19       | 23.69     | 1.27         |
+| `ssir`               | 26.544 | 0.718  | 0.325   | 4.769   | 28.45    | 21.59 | 0.38       | 25.92     | 4.52         |
+| `ssml`               | 33.697 | 0.943  | 0.088   | 4.669   | 32.48    | 37.30 | 2.38       | 386.02    | 244.18       |
+| `base`               | 36.589 | 0.965  | 0.034   | 4.393   | 26.16    | 38.43 | 0.18       | 18.04     | 0.40         |
+| `stir`               | 37.914 | 0.973  | 0.027   | 4.236   | 25.10    | 39.18 | 5.08       | 43.31     | 21.07        |
+| `wgse`               | 39.036 | 0.978  | 0.023   | 4.231   | 25.76    | 44.11 | 3.81       | 415.26    | 73.62        |
+| `spk2imgnet`         | 39.154 | 0.978  | 0.022   | 4.243   | 25.20    | 43.09 | 3.90       | 1000.50   | 123.38       |
+| `bsf`                | 39.576 | 0.979  | 0.019   | 4.139   | 24.93    | 43.03 | 2.47       | 705.23    | 401.50       |
 
 ### 4. Model Usage
 We also provide a direct interface for users interested in taking the spike-to-image model as a part of their work:

{spikezoo-0.2.3.dist-info → spikezoo-0.2.3.3.dist-info}/RECORD

@@ -135,15 +135,14 @@ spikezoo/data/base/train/spike/203_part2_key_id151.dat,sha256=YEenLmbPvcxnKkVn3O
 spikezoo/data/base/train/spike/203_part3_key_id151.dat,sha256=MY9nM6XzKj-P-tRQ33WZ3G5xulNTpAXKP0y8ZQo7AIQ,3762500
 spikezoo/data/base/train/spike/203_part4_key_id151.dat,sha256=IVi2jics66YzpIF-WTkw47te4qOj9cjdgz56GmHpJKg,3762500
 spikezoo/datasets/__init__.py,sha256=lRJsvCfgbe3qrd9BKTlG9dsgfIJbfXqWOynnlAcBiUI,3346
-spikezoo/datasets/base_dataset.py,sha256=gaMcrgWB3JLfc5lAM5wZjgMj72lZaO5otGfXOwRRiL8,5954
-spikezoo/datasets/realworld_dataset.py,sha256=CSyXj_uo0y1a4TlLDSTkB0CyFiFdZiU1phrWdkwvqgg,701
+spikezoo/datasets/base_dataset.py,sha256=oQ_AqWuMlaKnR712_sJ4WiTbqqPqVsfcukDNpFDYXb0,5956
+spikezoo/datasets/realworld_dataset.py,sha256=VqT6zcLa72DL3Lg8f4TThhYUa1xSIifsrPwpjvk2uBE,726
 spikezoo/datasets/reds_base_dataset.py,sha256=W-IJv9H1bsKgp3RT3zsV40jw2PqY2M76jtIS4Qpif1o,859
-spikezoo/datasets/reds_ssir_dataset.py,sha256=t0hm6PUWX5hfvSXB0UEv_JuihIhc5-mufrfur5bIq_0,7076
-spikezoo/datasets/szdata_dataset.py,sha256=7lSqNzWkVUVTVufPhm7AjpLghpZvjfu1X9c0sCsrjwM,850
-spikezoo/datasets/uhsr_dataset.py,sha256=Ue0vfKMpmvR0TelQx8G4xdWB8HdSlb4HSjIVOyFe5oE,1168
+spikezoo/datasets/szdata_dataset.py,sha256=xvgkZFHNSQ-Sk_rqmgRKAqpeb2gYpt_gmstJKJ8ooqU,870
+spikezoo/datasets/uhsr_dataset.py,sha256=MKQeQsoCal10yMgHy3I7NJDgJJgkKgruH5tantP921A,1186
 spikezoo/metrics/__init__.py,sha256=LIKeWNeEMZLANITQD68XJBOhDq7iHiKC7ExtdrXMyQs,3273
 spikezoo/models/__init__.py,sha256=QZTELBoM3bUW8jZoxN4OuA2RYKeVUT1fboyeIuK8Rtk,1722
-spikezoo/models/base_model.py,sha256=RyWcK8FSdAYdP8UxKULnnqe1uNn6Qr5tBexee-DzSm8,9138
+spikezoo/models/base_model.py,sha256=v3TD4AmjttTZUg0vEy736TOFdbbBgDLZg_RL-b4-vYM,9152
 spikezoo/models/bsf_model.py,sha256=XeZcVC_ODJxyS_I6-CtzlHXSWntgsUtbuAKjczIQ_0M,3972
 spikezoo/models/spcsnet_model.py,sha256=kLzv-ASXZGnqEFx0jUBONBeRCrsnQ_omkQUYEnr6uJc,540
 spikezoo/models/spikeclip_model.py,sha256=Ej84RuYbkFRthtBMV1JtmTkUshAqINlrrJ7yiKIsC9s,1125
@@ -155,7 +154,7 @@ spikezoo/models/tfi_model.py,sha256=tgD_HsiXk9jGuh5f_Bh6c3BqJi1p5DWCVo4N1tp5fgs,
 spikezoo/models/tfp_model.py,sha256=ihl1H__bWIbE9oair_t8rNJ5qnPJPKl-r_DpaO-0Sdk,663
 spikezoo/models/wgse_model.py,sha256=Kl9uV-LeO0Lj7SuPQ9pglw1Khs2b-7miS3A_faL6WSU,805
 spikezoo/pipeline/__init__.py,sha256=WPsukNR4cannwsghiukqNsWbWGH5DVPapR_Ly-WOU4Q,188
-spikezoo/pipeline/base_pipeline.py,sha256=ns2rBRbKaq8M9yDtWOiD-t6lo_5_c5rE2ZWH9sR78P4,13361
+spikezoo/pipeline/base_pipeline.py,sha256=3laGM-cMhNTSfCXx5jY53V8GY5BhPhHK48yjqm-Gre0,13478
 spikezoo/pipeline/ensemble_pipeline.py,sha256=ljZkGiCCpxvpC04Aa-r_tvBnqcBpUVi9fl_878tJAcg,2555
 spikezoo/pipeline/train_cfgs.py,sha256=6NO7DfPc7yjJfOrcIPQPfUPbUODz6eRKurEIDjMmaxA,3836
 spikezoo/pipeline/train_pipeline.py,sha256=BgHUsdv33B_OKauOVclNt7yIPb-_O-93ZHLHIjrwWaA,8459
@@ -167,8 +166,8 @@ spikezoo/utils/other_utils.py,sha256=uWNWaII9Jv7fkWNfkAD9wD-4ID-GAzbR-gGYT-1FF_c
 spikezoo/utils/scheduler_utils.py,sha256=5RBh-hl3-2y-IomxMs47T1p3JsbicZNYLza6q1uAKHo,828
 spikezoo/utils/spike_utils.py,sha256=u4Haa6Sp5xFqs61ztvq161oXTA_aZmNW3VYUZcayNW0,4296
 spikezoo/utils/vidar_loader.cpython-39-x86_64-linux-gnu.so,sha256=uXqu7ME---cZRRU5LUcLiNrjjtlOjxNwWHyTIQ10BGg,199088
-spikezoo-0.2.3.dist-info/LICENSE.txt,sha256=ukEi8E0PKq1dQGTXHUflg3rppLymwAhr7il9x-0nPgg,1062
-spikezoo-0.2.3.dist-info/METADATA,sha256=XajepZVqBR32M502EF2u-ALRC_s6cHk2IJK6AYgUo78,11960
-spikezoo-0.2.3.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
-spikezoo-0.2.3.dist-info/top_level.txt,sha256=xF2iuOstrACJh43NW4dsTwIdgKfXPXAb_Xzl3M1ricM,9
-spikezoo-0.2.3.dist-info/RECORD,,
+spikezoo-0.2.3.3.dist-info/LICENSE.txt,sha256=ukEi8E0PKq1dQGTXHUflg3rppLymwAhr7il9x-0nPgg,1062
+spikezoo-0.2.3.3.dist-info/METADATA,sha256=3aqIRpJr6TAfjldU8ZLmWy6uuCUla400W5tGpkH-X2M,11941
+spikezoo-0.2.3.3.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
+spikezoo-0.2.3.3.dist-info/top_level.txt,sha256=xF2iuOstrACJh43NW4dsTwIdgKfXPXAb_Xzl3M1ricM,9
+spikezoo-0.2.3.3.dist-info/RECORD,,

spikezoo/datasets/reds_ssir_dataset.py

@@ -1,181 +0,0 @@
-from torch.utils.data import Dataset
-from pathlib import Path
-from spikezoo.datasets.base_dataset import BaseDataset, BaseDatasetConfig
-from dataclasses import dataclass
-import re
-
-
-@dataclass
-class REDS_SSIRConfig(BaseDatasetConfig):
-    dataset_name: str = "reds_ssir"
-    root_dir: Path = Path(__file__).parent.parent / Path("data/REDS_SSIR")
-    train_width: int = 96
-    train_height: int = 96
-    test_width: int = 1280
-    test_height: int = 720
-    width: int = -1
-    height: int = -1
-    with_img: bool = True
-    spike_length_train: int = 41
-    spike_length_test: int = 301
-
-    # post process
-    def __post_init__(self):
-        self.root_dir = Path(self.root_dir) if isinstance(self.root_dir, str) else self.root_dir
-        # todo try download
-        assert self.root_dir.exists(), f"No files found in {self.root_dir} for the specified dataset `{self.dataset_name}`."
-        # train/test split
-        if self.split == "train":
-            self.spike_length = self.spike_length_train
-            self.width = self.train_width
-            self.height = self.train_height
-        else:
-            self.spike_length = self.spike_length_test
-            self.width = self.test_width
-            self.height = self.test_height
-
-
-class REDS_SSIR(BaseDataset):
-    def __init__(self, cfg: BaseDatasetConfig):
-        super(REDS_SSIR, self).__init__(cfg)
-
-    def prepare_data(self):
-        """Specify the spike and image files to be loaded."""
-        # spike/imgs dir train/test
-        if self.cfg.split == "train":
-            self.img_dir = self.cfg.root_dir / Path("crop_mini/spike/train/interp_20_alpha_0.40")
-            self.spike_dir = self.cfg.root_dir / Path("crop_mini/image/train/train_orig")
-        else:
-            self.img_dir = self.cfg.root_dir / Path("imgs/val/val_orig")
-            self.spike_dir = self.cfg.root_dir / Path("spike/val/interp_20_alpha_0.40")
-        # get files
-        self.spike_list = self.get_spike_files(self.spike_dir)
-        self.img_list = []
-
-
-class sreds_train(torch.utils.data.Dataset):
-    def __init__(self, cfg):
-        self.cfg = cfg
-        self.pair_step = self.cfg["loader"]["pair_step"]
-        self.augmentor = Augmentor(crop_size=self.cfg["loader"]["crop_size"])
-        self.samples = self.collect_samples()
-        print("The samples num of training data: {:d}".format(len(self.samples)))
-
-    def confirm_exist(self, path_list_list):
-        for pl in path_list_list:
-            for p in pl:
-                if not osp.exists(p):
-                    return 0
-        return 1
-
-    def collect_samples(self):
-        spike_path = osp.join(
-            self.cfg["data"]["root"], "crop_mini", "spike", "train", "interp_{:d}_alpha_{:.2f}".format(self.cfg["data"]["interp"], self.cfg["data"]["alpha"])
-        )
-        image_path = osp.join(self.cfg["data"]["root"], "crop_mini", "image", "train", "train_orig")
-        scene_list = sorted(os.listdir(spike_path))
-        samples = []
-
-        for scene in scene_list:
-            spike_dir = osp.join(spike_path, scene)
-            image_dir = osp.join(image_path, scene)
-            spk_path_list = sorted(os.listdir(spike_dir))
-
-            spklen = len(spk_path_list)
-            seq_len = self.cfg["model"]["seq_len"] + 2
-            """
-            for st in range(0, spklen - ((spklen - self.pair_step) % seq_len) - seq_len, self.pair_step):
-                # 按照文件名称读取
-                spikes_path_list = [osp.join(spike_dir, spk_path_list[ii]) for ii in range(st, st+seq_len)]
-                images_path_list = [osp.join(image_dir, spk_path_list[ii][:-4]+'.png') for ii in range(st, st+seq_len)]
-                
-                if(self.confirm_exist([spikes_path_list, images_path_list])):
-                    s = {}
-                    s['spikes_paths'] = spikes_path_list
-                    s['images_paths'] = images_path_list
-                    samples.append(s)
-            """
-            # 按照文件名称读取
-            spikes_path_list = [osp.join(spike_dir, spk_path_list[ii]) for ii in range(spklen)]
-            images_path_list = [osp.join(image_dir, spk_path_list[ii][:-4] + ".png") for ii in range(spklen)]
-
-            if self.confirm_exist([spikes_path_list, images_path_list]):
-                s = {}
-                s["spikes_paths"] = spikes_path_list
-                s["images_paths"] = images_path_list
-                samples.append(s)
-
-        return samples
-
-    def _load_sample(self, s):
-        data = {}
-
-        data["spikes"] = [np.array(dat_to_spmat(p, size=(96, 96)), dtype=np.float32) for p in s["spikes_paths"]]
-        data["images"] = [read_img_gray(p) for p in s["images_paths"]]
-
-        data["spikes"], data["images"] = self.augmentor(data["spikes"], data["images"])
-        # print("data['spikes'][0].shape, data['images'][0].shape", data['spikes'][0].shape, data['images'][0].shape)
-
-        return data
-
-    def __len__(self):
-        return len(self.samples)
-
-    def __getitem__(self, index):
-        data = self._load_sample(self.samples[index])
-        return data
-
-
-class sreds_test(torch.utils.data.Dataset):
-    def __init__(self, cfg):
-        self.cfg = cfg
-        self.samples = self.collect_samples()
-        print("The samples num of testing data: {:d}".format(len(self.samples)))
-
-    def confirm_exist(self, path_list_list):
-        for pl in path_list_list:
-            for p in pl:
-                if not osp.exists(p):
-                    return 0
-        return 1
-
-    def collect_samples(self):
-        spike_path = osp.join(
-            self.cfg["data"]["root"], "spike", "val", "interp_{:d}_alpha_{:.2f}".format(self.cfg["data"]["interp"], self.cfg["data"]["alpha"])
-        )
-        image_path = osp.join(self.cfg["data"]["root"], "imgs", "val", "val_orig")
-        scene_list = sorted(os.listdir(spike_path))
-        samples = []
-
-        for scene in scene_list:
-            spike_dir = osp.join(spike_path, scene)
-            image_dir = osp.join(image_path, scene)
-            spk_path_list = sorted(os.listdir(spike_dir))
-
-            spklen = len(spk_path_list)
-            # seq_len = self.cfg['model']['seq_len']
-
-            # 按照文件名称读取
-            spikes_path_list = [osp.join(spike_dir, spk_path_list[ii]) for ii in range(spklen)]
-            images_path_list = [osp.join(image_dir, spk_path_list[ii][:-4] + ".png") for ii in range(spklen)]
-
-            if self.confirm_exist([spikes_path_list, images_path_list]):
-                s = {}
-                s["spikes_paths"] = spikes_path_list
-                s["images_paths"] = images_path_list
-                samples.append(s)
-
-        return samples
-
-    def _load_sample(self, s):
-        data = {}
-        data["spikes"] = [np.array(dat_to_spmat(p, size=(720, 1280)), dtype=np.float32) for p in s["spikes_paths"]]
-        data["images"] = [read_img_gray(p) for p in s["images_paths"]]
-        return data
-
-    def __len__(self):
-        return len(self.samples)
-
-    def __getitem__(self, index):
-        data = self._load_sample(self.samples[index])
-        return data