.. _version: 发行版本介绍 ======================= 发行版本参数 ``version`` 的使用方式如下所示: .. code-block:: python 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 .. note:: 当前发行版本中,所有模型的脉冲输入长度设置为 ``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 | +----------------------+---------+---------+---------+---------+----------+---------+------------+-----------+--------------+