发行版本介绍
发行版本参数 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) |
|---|---|---|---|---|---|---|---|---|---|
|
25.098 |
0.726 |
0.318 |
7.395 |
46.44 |
49.44 |
0.00 |
0.00 |
3.60 |
|
28.169 |
0.770 |
0.261 |
8.058 |
47.26 |
38.13 |
0.00 |
0.00 |
0.03 |
|
21.873 |
0.578 |
0.333 |
7.802 |
42.08 |
54.01 |
0.19 |
23.69 |
1.27 |
|
26.544 |
0.718 |
0.325 |
4.769 |
28.45 |
21.59 |
0.38 |
25.92 |
4.52 |
|
33.697 |
0.943 |
0.088 |
4.669 |
32.48 |
37.30 |
2.38 |
386.02 |
244.18 |
|
36.589 |
0.965 |
0.034 |
4.393 |
26.16 |
38.43 |
0.18 |
18.04 |
0.40 |
|
37.914 |
0.973 |
0.027 |
4.236 |
25.10 |
39.18 |
5.08 |
43.31 |
21.07 |
|
39.036 |
0.978 |
0.023 |
4.231 |
25.76 |
44.11 |
3.81 |
415.26 |
73.62 |
|
39.154 |
0.978 |
0.022 |
4.243 |
25.20 |
43.09 |
3.90 |
1000.50 |
123.38 |
|
39.576 |
0.979 |
0.019 |
4.139 |
24.93 |
43.03 |
2.47 |
705.23 |
401.50 |