Spike-Zoo 文档
⚡ Spike-Zoo 是一个用于脉冲相机图像重构网络的推理和训练框架,主要功能包括:
使用预训练权重,快速进行多个重构模型的 推理。
训练 自定义设计的脉冲图像重构网络。
提供多种处理脉冲数据的 函数。
安装
从
pypi安装最新的稳定版本:
pip install spikezoo
从
github安装最新的开发版本 (推荐):
git clone https://github.com/chenkang455/Spike-Zoo
cd Spike-Zoo
python setup.py install
使用
develop模式安装,便于调试和训练自定义网络:
git clone https://github.com/chenkang455/Spike-Zoo
cd Spike-Zoo
python setup.py develop
备注
如有任何问题或需求,欢迎提出 issue。也欢迎通过 Pull Request 将自定义方法加入到 Spike-Zoo 中。
使用教程
致谢
我们的代码基于以下开源项目的构建。感谢这些项目的贡献者:
SpikeCV: https://spikecv.github.io/
IQA-PyTorch: https://github.com/chaofengc/IQA-PyTorch
NeRFStudio: https://github.com/nerfstudio-project/nerfstudio
同时,感谢 zhiwen huang rui zhao 、jiyuan zhang 以及 shiyan chen 对我们项目的帮助。
引用
@misc{spikezoo,
title={{Spike-Zoo}: Spike-Zoo: A Toolbox for Spike-to-Image Reconstruction},
author={Kang Chen and Zhiyuan Ye and Tiejun Huang and Zhaofei Yu},
year={2025},
howpublished = "[Online]. Available: \url{https://github.com/chenkang455/Spike-Zoo}"
}