# MLF-DSResNet

**Repository Path**: xuan-zhang0509/MLF-DSResNet

## Basic Information

- **Project Name**: MLF-DSResNet
- **Description**: Reset2 SNN
- **Primary Language**: Unknown
- **License**: MIT
- **Default Branch**: main
- **Homepage**: None
- **GVP Project**: No

## Statistics

- **Stars**: 0
- **Forks**: 0
- **Created**: 2022-12-15
- **Last Updated**: 2022-12-19

## Categories & Tags

**Categories**: Uncategorized

**Tags**: None

## README

# Multi-Level Firing with Spiking DS-ResNet: Enabling Better and Deeper Directly-Trained Spiking Neural Networks
## IJCAI 2022
This repository contains Python implementation of MLF & spiking DS-ResNet.

## 1. Publication
If you find this repository useful, please consider citing the following paper:

<pre>
@inproceedings{lang2022multi,
  title     = {Multi-Level Firing with Spiking DS-ResNet: Enabling Better and Deeper Directly-Trained Spiking Neural Networks},
  author    = {Feng, Lang and Liu, Qianhui and Tang, Huajin and Ma, De and Pan, Gang},
  booktitle = {Proceedings of the Thirty-First International Joint Conference on
               Artificial Intelligence, {IJCAI-22}},
  pages     = {2471--2477},
  year      = {2022},
  doi       = {10.24963/ijcai.2022/343},
  url       = {https://doi.org/10.24963/ijcai.2022/343},
}
</pre>

## 2. Datasets
* Download link of DVS-gesture: https://research.ibm.com/interactive/dvsgesture/.
Put file `DvsGesture.tar.gz` in the path `./data/DVS_Gesture/source_DvsGesture/`, then unzip `DvsGesture.tar.gz`.

* Download link of CIFAR10-DVS: https://figshare.com/articles/dataset/CIFAR10-DVS_New/4724671/2.
Put file `airplane.zip`～`horse.zip` in the path `./data/DVS_CIFAR10/source_DvsCIFAR10/`, then unzip `airplane.zip`～`horse.zip`.

* CIFAR10 dataset can be downloaded online.

## 3. Dependencies:
* python 3.7.10
* numpy 1.19.5
* torch 1.9.0+cu111
* torchvision 0.10.0+cu111
* tensorboardX 2.4
* h5py 3.3.0

## 4. Preprocessing
DVS-gesture and CIFAR10-DVS need to be pre-processed. The syntax is as follow,
```
python DVS_CIFAR10_preprocess.py
python DVS_Gesture_preprocess.py
```

## 5. Traning
To train a new model, the basic syntax is like:
```
python train_for_cifar10.py
python train_for_gesture.py
python train_for_dvscifar10.py
```