Skip to content

szmazurek/snn_dvs

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

This repository contains the code for experiments presented in:
Pedestrian intention prediction in Adverse Weather Conditions with Spiking Neural Networks and Dynamic Vision Sensors

The work is this project is still in progress. All files are presented as is and may contain bugs. Comments, issues and contributions are welcome.

Overview

All experiments are meant to be configured via config.yml file. To get more information about the configuration file, please refer to the config.yml file in the root directory and the comments there.

Further explanation for non-self-explanatory parameters:

  • model.type: temporal for multiple frames, single sample for one-frame training.

  • model.name: supported models include resnet18, resnet18_spiking, slow_r50, vgg11_spiking, and sew_resnet18_spiking.

  • model.spiking_params: used only during spiking neural network training.

  • dataset.type:

    • single sample: Single-frame approach.
    • repeated: The same frame passed n_samples times.
    • temporal: n_samples next frames used.
  • prediction modes:

    • new labeling approach for expectancy of crossing.
    • looking dataset.n_frames_predictive_horizon frames back to determine if there is a crossing label in a certain video and labeling it as crossing.

Training

To begin training your model, you can use the train_lightning.py script located in the src directory. It is refering to parameters configured in config.yml

python src/train_lightning.py

Datasets

The simulation dataset that is introduced in this work is available under the DOI: 10.5281/zenodo.11409259. The dataset is not included in this repository and should be downloaded separately.
Subset of JAAD dataset is used in this work, which can be downloaded here. For full dataset, please refer to the JAAD website.

Experiments involving energy usage monitoring

Experiments measuring the number of synaptic operations were ran using the syops package available at https://github.com/iCGY96/syops-counter. The package is not included in this repository and should be installed separately. Note that in order to run the experiments, manual changes were made in the code of the package. For further details, please contact the authors.

About

No description, website, or topics provided.

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages