发行版本介绍

发行版本参数 version 的使用方式如下所示:

from spikezoo.pipeline import Pipeline, PipelineConfig
import spikezoo as sz
pipeline = Pipeline(
    cfg=PipelineConfig(save_folder="results",version="v023"),
    model_cfg=sz.METHOD.BASE,
    dataset_cfg=sz.DATASET.BASE
)

通过指定发行版本,模型会自动加载对应版本网络参数并下载预训练权重。

v010

v0.1.0 中包含各方法原始论文中的预训练模型和参数。

地址: https://github.com/chenkang455/Spike-Zoo/releases/tag/v0.1.0

v023

v0.2.3 中包含各方法在 REDS_BASE 数据集上重新训练得到的权重,训练脚本地址 见此

地址: https://github.com/chenkang455/Spike-Zoo/releases/tag/v0.2.3

备注

当前发行版本中,所有模型的脉冲输入长度设置为 41 帧。由于 SpikeCLIP 是专门为 UHSR 数据集设计的算法,因此并未对其进行重新训练。不过,base 网络架构是从 SpikeCLIP 中修改而来的。

Method

PSNR

SSIM

LPIPS

NIQE

BRISQUE

PIQE

Params (M)

FLOPs (G)

Latency (ms)

tfi

25.098

0.726

0.318

7.395

46.44

49.44

0.00

0.00

3.60

tfp

28.169

0.770

0.261

8.058

47.26

38.13

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